Update base.py

JialeCao001 · Jul 5, 2020 · 8aac377 · 8aac377
1 parent 1814162
commit 8aac377
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mmdet/models/detectors/base.py b/mmdet/models/detectors/base.py
@@ -167,7 +167,10 @@ def show_result(self, data, result, dataset=None, score_thr=0.3):
             bboxes = np.vstack(bbox_result)
             # draw segmentation masks
             if segm_result is not None:
-                segms = mmcv.concat_list(segm_result)
+                if len(segm_result) > 1:
+                    segms = mmcv.concat_list(segm_result[0])
+                else:
+                    segms = mmcv.concat_list(segm_result)
                 inds = np.where(bboxes[:, -1] > score_thr)[0]
                 for i in inds:
                     color_mask = np.random.randint(