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
......@@ -90,7 +90,7 @@ class BBoxTestMixin(object):
max_size = max([proposal.size(0) for proposal in proposals])
for i, proposal in enumerate(proposals):
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)
rois = torch.stack(proposals, dim=0)
else:
......
Markdown is supported
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