Skip to content

GitLab

Commit 9bdb82ea authored by myownskyW7's avatar myownskyW7
Browse files

bug fix in coco dataset

parent d52e2d88
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
mmdet/datasets/coco.py
View file @ 9bdb82ea
...@@ -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