Skip to content
Snippets Groups Projects
Commit 9bdb82ea authored by myownskyW7's avatar myownskyW7
Browse files

bug fix in coco dataset

parent d52e2d88
No related branches found
No related tags found
No related merge requests found
......@@ -280,6 +280,8 @@ class CocoDataset(Dataset):
raise AssertionError(
'proposals should have shapes (n, 4) or (n, 5), '
'but found {}'.format(proposal.shape))
else:
proposal = None
def prepare_single(img, scale, flip, proposal=None):
_img, img_shape, pad_shape, scale_factor = self.img_transform(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment