Skip to content

GitLab

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

Merge pull request #301 from AresGao/master 

fix a little bug in coco datasets loader
parents 893c4108 d268e93d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet/datasets/coco.py
View file @ e2227ddb
...@@ -40,7 +40,7 @@ class CocoDataset(CustomDataset): ...@@ -40,7 +40,7 @@ class CocoDataset(CustomDataset):
img_id = self.img_infos[idx]['id'] img_id = self.img_infos[idx]['id']
ann_ids = self.coco.getAnnIds(imgIds=[img_id]) ann_ids = self.coco.getAnnIds(imgIds=[img_id])
ann_info = self.coco.loadAnns(ann_ids) ann_info = self.coco.loadAnns(ann_ids)
return self._parse_ann_info(ann_info) return self._parse_ann_info(ann_info, self.with_mask)
def _filter_imgs(self, min_size=32): def _filter_imgs(self, min_size=32):
"""Filter images too small or without ground truths.""" """Filter images too small or without ground truths."""
......
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