Skip to content

Commit 754c080

Browse files
fix rtmdet-ins head (#1884)
1 parent dce6e64 commit 754c080

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

mmdeploy/codebase/mmdet/models/dense_heads/rtmdet_ins_head.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,9 @@ def _nms_with_mask_static(self,
170170
# pad
171171
dets = torch.cat((dets, dets.new_zeros((1, 1, 5))), 1)
172172
labels = torch.cat((labels, labels.new_zeros((1, 1))), 1)
173+
kernels = torch.cat((kernels, kernels.new_zeros(1, 1, kernels.shape[2])),
174+
1)
175+
priors = torch.cat((priors, priors.new_zeros(1, 4)), 0)
173176

174177
# topk or sort
175178
is_use_topk = keep_top_k > 0 and \

0 commit comments

Comments
 (0)