def main(_):
if not FLAGS.dataset_name:
raise ValueError(
'You must supply the dataset name with --dataset_name')
if not FLAGS.dataset_dir:
raise ValueError(
'You must supply the dataset directory with --dataset_dir')
if FLAGS.dataset_name == 'flowers':
download_and_convert_flowers.run(FLAGS.dataset_dir)
elif FLAGS.dataset_name == 'cifar10':
download_and_convert_cifar10.run(FLAGS.dataset_dir)
elif FLAGS.dataset_name == 'mnist':
download_and_convert_mnist.run(FLAGS.dataset_dir)
elif FLAGS.dataset_name == 'visualwakewords':
download_and_convert_visualwakewords.run(
FLAGS.dataset_dir, FLAGS.small_object_area_threshold,
FLAGS.foreground_class_of_interest)
elif FLAGS.dataset_name == 'generic':
download_and_convert_generic.run(FLAGS.dataset_dir,
perc_validation=FLAGS.perc_validation,
labels_filename=FLAGS.labels_filename,
num_shards=FLAGS.num_shards)
else:
raise ValueError('dataset_name [%s] was not recognized.' %
FLAGS.dataset_name)
def main(_):
if not FLAGS.dataset_name:
raise ValueError(
'You must supply the dataset name with --dataset_name')
if not FLAGS.dataset_dir:
raise ValueError(
'You must supply the dataset directory with --dataset_dir')
if FLAGS.dataset_name == 'flowers':
download_and_convert_flowers.run(FLAGS.dataset_dir)
elif FLAGS.dataset_name == 'cifar10':
download_and_convert_cifar10.run(FLAGS.dataset_dir)
elif FLAGS.dataset_name == 'mnist':
download_and_convert_mnist.run(FLAGS.dataset_dir)
elif FLAGS.dataset_name == 'visualwakewords':
download_and_convert_visualwakewords.run(
FLAGS.dataset_dir, FLAGS.small_object_area_threshold,
FLAGS.foreground_class_of_interest, FLAGS.download, FLAGS.coco_dir)
else:
raise ValueError('dataset_name [%s] was not recognized.' %
FLAGS.dataset_name)