Python download_dataset示例

示例#1

def _download_cifar_split(split, is_training):
    """Build a CIFAR-10 dataset from TFDS (as opposed to a custom TFR table)."""
    del is_training
    filenames = [
        os.path.join(FLAGS.cifar_dir, f) for f in _CIFAR_FILENAMES[split]
    ]
    dataset = (tf.data.FixedLengthRecordDataset(
        filenames, _RECORD_BYTES).map(_parse_cifar))
    return experiment_utils.download_dataset(dataset)

示例#2

def _download_alt_dataset(config):
    dataset_to_dl = tfds.load(config.alt_dataset_name,
                              split=config.split,
                              as_supervised=True)
    return experiment_utils.download_dataset(dataset_to_dl)