diff --git a/mmdet/datasets/coco.py b/mmdet/datasets/coco.py
index d5830018c2a15a5a4cc5dca2e5dbff29751201aa..4058db90be6b50336af66fa14270d7dd0d16f882 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)