Merge pull request #1 from yichenj/voc2012_patch

Revise voc2012 dataset preparing script
yichenj · Aug 20, 2020 · 17bcb67 · 17bcb67
2 parents 34fd1c9 + 313df71
commit 17bcb67
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tools/voc2012.py b/tools/voc2012.py
@@ -93,10 +93,9 @@ def main(_argv):
 
     writer = tf.io.TFRecordWriter(FLAGS.output_file)
     image_list = open(os.path.join(
-        FLAGS.data_dir, 'ImageSets', 'Main', 'aeroplane_%s.txt' % FLAGS.split)).read().splitlines()
+        FLAGS.data_dir, 'ImageSets', 'Main', '%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()
+    for name in tqdm.tqdm(image_list):
         annotation_xml = os.path.join(
             FLAGS.data_dir, 'Annotations', name + '.xml')
         annotation_xml = lxml.etree.fromstring(open(annotation_xml).read())