Fix scores not contiguous error in centernet_head. (open-mmlab#6016)

* Update centernet_head.py * Update centernet_head.py * Update centernet_head.py * Update centernet_head.py * Update centernet_head.py
jozhang97 · Sep 7, 2021 · 460934d · 460934d
1 parent 003f520
commit 460934d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmdet/models/dense_heads/centernet_head.py b/mmdet/models/dense_heads/centernet_head.py
@@ -368,7 +368,8 @@ def _bboxes_nms(self, bboxes, labels, cfg):
         if labels.numel() == 0:
             return bboxes, labels
 
-        out_bboxes, keep = batched_nms(bboxes[:, :4], bboxes[:, -1], labels,
+        out_bboxes, keep = batched_nms(bboxes[:, :4].contiguous(),
+                                       bboxes[:, -1].contiguous(), labels,
                                        cfg.nms_cfg)
         out_labels = labels[keep]