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)