Fix mask dimensions mismatch in rtmdet-ins models after exporting to …

…onnx (#2574)
open-mmlab · Nov 30, 2023 · db73d55 · db73d55
1 parent 8b19586
commit db73d55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdeploy/mmcv/ops/nms.py b/mmdeploy/mmcv/ops/nms.py
@@ -369,7 +369,7 @@ def _multiclass_nms_single(boxes: Tensor,
         if pre_top_k > 0:
             bbox_index = pre_topk_inds[None, box_inds]
         if keep_top_k > 0:
-            bbox_index = bbox_index[:, topk_inds[:-1]]
+            bbox_index = bbox_index[:, topk_inds]
         return dets, labels, bbox_index
     else:
         return dets, labels