Skip to content

GitLab

Unverified Commit 3ad05dc8 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #31 from myownskyW7/master 

 bug fix in coco dataset
parents 3d790bcb 9bdb82ea
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
mmdet/datasets/coco.py
View file @ 3ad05dc8
...@@ -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