fix crowd boxes

bes-dev · Jun 27, 2020 · f7b2188 · f7b2188
1 parent a93653a
commit f7b2188
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mean_average_precision/utils.py b/mean_average_precision/utils.py
@@ -192,9 +192,12 @@ def check_box(iou, difficult, crowd, order, matched_ind, iou_threshold, mpolicy=
             idx = order[i]
             if iou[idx] > iou_threshold:
                 if not difficult[idx]:
-                    if not idx in matched_ind or crowd[idx]:
+                    if idx not in matched_ind:
                         result = ('tp', idx)
                         break
+                    elif crowd[idx]:
+                        result = ('ignore', -1)
+                        break
                     else:
                         continue
                 else: