Update shapes_to_coco.py

waspinator · Apr 13, 2018 · 6197428 · 6197428
1 parent 90df8ec
commit 6197428
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/shapes/shapes_to_coco.py b/examples/shapes/shapes_to_coco.py
@@ -98,9 +98,10 @@ def main():
                 # go through each associated annotation
                 for annotation_filename in annotation_files:
 
-                    if 'square' in image_filename:
+                    print(annotation_filename)
+                    if 'square' in annotation_filename:
                         class_id = 1
-                    elif 'circle' in image_filename:
+                    elif 'circle' in annotation_filename:
                         class_id = 2
                     else:
                         class_id = 3