Skip to content
Snippets Groups Projects
Unverified Commit ff38f207 authored by Ye Liu's avatar Ye Liu Committed by GitHub
Browse files

Fix supplement size in simple_test_bboxes (#4860)

parent a5ee9b41
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mmdet/models/roi_heads/test_mixins.py
+ 1
− 1
View file @ ff38f207
...@@ -90,7 +90,7 @@ class BBoxTestMixin(object): ...@@ -90,7 +90,7 @@ class BBoxTestMixin(object):
max_size = max([proposal.size(0) for proposal in proposals]) max_size = max([proposal.size(0) for proposal in proposals])
for i, proposal in enumerate(proposals): for i, proposal in enumerate(proposals):
supplement = proposal.new_full( supplement = proposal.new_full(
(max_size - proposal.size(0), 5), 0) (max_size - proposal.size(0), proposal.size(1)), 0)
proposals[i] = torch.cat((supplement, proposal), dim=0) proposals[i] = torch.cat((supplement, proposal), dim=0)
rois = torch.stack(proposals, dim=0) rois = torch.stack(proposals, dim=0)
else: else:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment