processed_image_ids record logic changed, thanks to @mnslarcher

zylo117 · web-flow · commit 8556f8c7ee34 · 2020-05-26T09:34:29.000+08:00
diff --git a/coco_eval.py b/coco_eval.py
@@ -82,12 +82,12 @@ def evaluate_coco(img_path, set_name, image_ids, coco, model, threshold=0.05):
                             anchors, regression, classification,
                             regressBoxes, clipBoxes,
                             threshold, nms_threshold)
-
-        processed_image_ids.append(image_id)
-
+        
         if not preds:
             continue
 
+        processed_image_ids.append(image_id)
+
         preds = invert_affine(framed_metas, preds)[0]
 
         scores = preds['scores']
