Update tools/demo_api.py

tools/demo_api.py

@@ -171,16 +171,16 @@ def run_detection(predictor, path):
         img_list.append(img_entry)
 
         print(outputs[0])
-        
-        for id, output in enumerate(outputs[0]):
-            print(output)
-            ann_entry = {"id": id,
-                         "image_id": img_id,
-                         "bbox": (output[:4] / ratio).tolist(),
-                         "cls": output[6].item(),
-                         "score": (output[4] * output[5]).item() }
-            ann_list.append(ann_entry)
-
+        if len(outputs):
+            for id, output in enumerate(outputs[0]):
+                print(output)
+                ann_entry = {"id": id,
+                             "image_id": img_id,
+                             "bbox": (output[:4] / ratio).tolist(),
+                             "cls": output[6].item(),
+                             "score": (output[4] * output[5]).item() }
+                ann_list.append(ann_entry)
+    
     data_dict = { "images": img_list,
                  "annotations": ann_list
                 }