From 6197428fc1d15e1fe2f2f464821058f38ddac994 Mon Sep 17 00:00:00 2001 From: waspinator Date: Thu, 12 Apr 2018 22:45:22 -0400 Subject: [PATCH] Update shapes_to_coco.py --- examples/shapes/shapes_to_coco.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/shapes/shapes_to_coco.py b/examples/shapes/shapes_to_coco.py index 3c3b8f2..fda3805 100644 --- 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