From 9dd4334b89bd4d3d40ed4336abcf09b9fcb1beaf Mon Sep 17 00:00:00 2001
From: Yuchen Fan <fyc0624@gmail.com>
Date: Wed, 31 Oct 2018 15:09:54 -0500
Subject: [PATCH] Add support for string ids for COCO dataset
---
mmdet/datasets/coco.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mmdet/datasets/coco.py b/mmdet/datasets/coco.py
index d583001..4058db9 100644
--- a/mmdet/datasets/coco.py
+++ b/mmdet/datasets/coco.py
@@ -16,14 +16,14 @@ class CocoDataset(CustomDataset):
self.img_ids = self.coco.getImgIds()
img_infos = []
for i in self.img_ids:
- info = self.coco.loadImgs(i)[0]
+ info = self.coco.loadImgs([i])[0]
info['filename'] = info['file_name']
img_infos.append(info)
return img_infos
def get_ann_info(self, idx):
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)
return self._parse_ann_info(ann_info)
--
GitLab