未验证 提交 2b9807cd 编写于 作者: W will-jl944 提交者: GitHub

fix bug that pop item while traversing image list in coco dataset (#4056)

上级 421a9214
...@@ -179,7 +179,7 @@ class COCODataSet(DetDataset): ...@@ -179,7 +179,7 @@ class COCODataSet(DetDataset):
gt_poly = [None] * num_bbox gt_poly = [None] * num_bbox
has_segmentation = False has_segmentation = False
for i, box in enumerate(bboxes): for i, box in reversed(list(enumerate(bboxes))):
catid = box['category_id'] catid = box['category_id']
gt_class[i][0] = self.catid2clsid[catid] gt_class[i][0] = self.catid2clsid[catid]
gt_bbox[i, :] = box['clean_bbox'] gt_bbox[i, :] = box['clean_bbox']
...@@ -193,11 +193,10 @@ class COCODataSet(DetDataset): ...@@ -193,11 +193,10 @@ class COCODataSet(DetDataset):
elif 'segmentation' in box and box['segmentation']: elif 'segmentation' in box and box['segmentation']:
if not np.array(box['segmentation'] if not np.array(box['segmentation']
).size > 0 and not self.allow_empty: ).size > 0 and not self.allow_empty:
bboxes.pop(i)
gt_poly.pop(i) gt_poly.pop(i)
np.delete(is_crowd, i) is_crowd = np.delete(is_crowd, i)
np.delete(gt_class, i) gt_class = np.delete(gt_class, i)
np.delete(gt_bbox, i) gt_bbox = np.delete(gt_bbox, i)
else: else:
gt_poly[i] = box['segmentation'] gt_poly[i] = box['segmentation']
has_segmentation = True has_segmentation = True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册