From 9bdb82ea53de1104a79ee8f0b145949ef2e7fa66 Mon Sep 17 00:00:00 2001
From: myownskyW7 <727032989@qq.com>
Date: Tue, 16 Oct 2018 14:32:36 +0800
Subject: [PATCH] bug fix in coco dataset
---
mmdet/datasets/coco.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/mmdet/datasets/coco.py b/mmdet/datasets/coco.py
index 73a91c5..0ee92f9 100644
--- a/mmdet/datasets/coco.py
+++ b/mmdet/datasets/coco.py
@@ -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(
--
GitLab