Revert "Global split for dataset"

bhardwajRahul · May 11, 2020 · c2739d3 · c2739d3
1 parent c940823
commit c2739d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/voc2012.py b/tools/voc2012.py
@@ -93,7 +93,7 @@ def main(_argv):
 
  writer = tf.io.TFRecordWriter(FLAGS.output_file)
  image_list = open(os.path.join(
- FLAGS.data_dir, 'ImageSets', 'Main', '%s.txt' % FLAGS.split)).read().splitlines()
+ FLAGS.data_dir, 'ImageSets', 'Main', 'aeroplane_%s.txt' % FLAGS.split)).read().splitlines()
  logging.info("Image list loaded: %d", len(image_list))
  for image in tqdm.tqdm(image_list):
  name, _ = image.split()