minor fix in htc tta when num det boxes is 0

open-mmlab · Jan 22, 2021 · 48258d9 · 48258d9
1 parent 3a9f5ef
commit 48258d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdet/models/roi_heads/htc_roi_head.py b/mmdet/models/roi_heads/htc_roi_head.py
@@ -536,8 +536,8 @@ def aug_test(self, img_feats, proposal_list, img_metas, rescale=False):
         if self.with_mask:
             if det_bboxes.shape[0] == 0:
                 segm_result = [[[]
-                                for _ in range(self.mask_head[-1].num_classes -
-                                               1)]]
+                                for _ in range(self.mask_head[-1].num_classes)]
+                               ]
             else:
                 aug_masks = []
                 aug_img_metas = []