Fix matterport#551

Valid index 0 is not regarded, leading to possibly more prediction_matches than grountruh_matches and recall values higher than 1.
bertsky · Oct 27, 2018 · 8a09a75 · 8a09a75
1 parent a8e674c
commit 8a09a75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mrcnn/utils.py b/mrcnn/utils.py
@@ -694,7 +694,7 @@ def compute_matches(gt_boxes, gt_class_ids, gt_masks,
         # 3. Find the match
         for j in sorted_ixs:
             # If ground truth box is already matched, go to next one
-            if gt_match[j] > 0:
+            if gt_match[j] > -1:
                 continue
             # If we reach IoU smaller than the threshold, end the loop
             iou = overlaps[i, j]