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

* Update centernet_head.py * Update centernet_head.py * Update centernet_head.py * Update centernet_head.py * Update centernet_head.py
4 years ago · 460934d8a6
parent 003f520cad
commit 460934d8a6
1 changed files with 2 additions and 1 deletions
--- a/mmdet/models/dense_heads/centernet_head.py
+++ b/mmdet/models/dense_heads/centernet_head.py
@ -368,7 +368,8 @@ class CenterNetHead(BaseDenseHead, BBoxTestMixin):
        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]