Merge pull request #8 from ChaseMonsterAway/master

fix: show ignore image problem when only gt file.
Media-Smart · Dec 8, 2020 · a189a27 · a189a27
2 parents a3962ba + f5d785e
commit a189a27
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/volkscv/analyzer/visualization/utils.py b/volkscv/analyzer/visualization/utils.py
@@ -448,6 +448,7 @@ def draw_bbox(img,
     bboxes = anno.get('bboxes')
     labels = anno.get('labels')
     scores = anno.get('scores')
+    not_ignore = bboxes.shape[0]
     labels_ignore = anno.get('labels_ignore', np.zeros((0,)))
     bboxes_ignore = anno.get('bboxes_ignore', np.zeros((0, 4)))
 
@@ -508,6 +509,12 @@ def draw_bbox(img,
             if not show_fpfn:
                 cv2.rectangle(img_, left_top, right_bottom,
                               color=color, thickness=base_thickness)
+                if show_ignore and idx >= not_ignore:
+                    thickness = base_thickness
+                    color = color_ignore
+                    cv2.rectangle(
+                        img_, left_top, right_bottom, color=color,
+                        thickness=thickness)
             else:
                 if len(f) and f[idx]:
                     if show_fpfn_format == 'mask':