Skip to content

GitLab

Commit 420c91e9 authored by Kai Chen's avatar Kai Chen
Browse files

Merge branch 'master' of github.com:open-mmlab/mmdetection

parents 251c4290 3ad05dc8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
mmdet/datasets/coco.py
View file @ 420c91e9
...@@ -280,6 +280,8 @@ class CocoDataset(Dataset): ...@@ -280,6 +280,8 @@ class CocoDataset(Dataset):
raise AssertionError( raise AssertionError(
'proposals should have shapes (n, 4) or (n, 5), ' 'proposals should have shapes (n, 4) or (n, 5), '
'but found {}'.format(proposal.shape)) 'but found {}'.format(proposal.shape))
else:
proposal = None
def prepare_single(img, scale, flip, proposal=None): def prepare_single(img, scale, flip, proposal=None):
_img, img_shape, pad_shape, scale_factor = self.img_transform( _img, img_shape, pad_shape, scale_factor = self.img_transform(
......
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