fix a bug of pisa loss when there are no gts in some images within on…

…e batch images (#2992)
open-mmlab · Jun 12, 2020 · a9c8c94 · a9c8c94
1 parent 8951a8a
commit a9c8c94
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmdet/models/losses/pisa_loss.py b/mmdet/models/losses/pisa_loss.py
@@ -50,7 +50,8 @@ def isr_p(cls_score,
     for i in range(len(sampling_results)):
         gt_i = sampling_results[i].pos_assigned_gt_inds
         gts.append(gt_i + last_max_gt)
-        last_max_gt = gt_i.max() + 1
+        if len(gt_i) != 0:
+            last_max_gt = gt_i.max() + 1
     gts = torch.cat(gts)
     assert len(gts) == num_pos