Skip to content

Commit db73d55

Browse files
Fix mask dimensions mismatch in rtmdet-ins models after exporting to onnx (#2574)
1 parent 8b19586 commit db73d55

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdeploy/mmcv/ops/nms.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def _multiclass_nms_single(boxes: Tensor,
369369
if pre_top_k > 0:
370370
bbox_index = pre_topk_inds[None, box_inds]
371371
if keep_top_k > 0:
372-
bbox_index = bbox_index[:, topk_inds[:-1]]
372+
bbox_index = bbox_index[:, topk_inds]
373373
return dets, labels, bbox_index
374374
else:
375375
return dets, labels

0 commit comments

Comments
 (0)