Skip to content

GitLab

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

Merge pull request #89 from ychfan/master 

Add support for string ids for COCO dataset
parents 86e25e41 9dd4334b
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmdet/datasets/coco.py
View file @ 3f5df4f0
...@@ -16,14 +16,14 @@ class CocoDataset(CustomDataset): ...@@ -16,14 +16,14 @@ class CocoDataset(CustomDataset):
self.img_ids = self.coco.getImgIds() self.img_ids = self.coco.getImgIds()
img_infos = [] img_infos = []
for i in self.img_ids: for i in self.img_ids:
info = self.coco.loadImgs(i)[0] info = self.coco.loadImgs([i])[0]
info['filename'] = info['file_name'] info['filename'] = info['file_name']
img_infos.append(info) img_infos.append(info)
return img_infos return img_infos
def get_ann_info(self, idx): def get_ann_info(self, idx):
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)
......
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