diff --git a/dygraph/ppdet/data/source/coco.py b/dygraph/ppdet/data/source/coco.py
index 4dad8265f5e793376a58f831d56c6f428d83ca48..e631cfa9f5a5ed77b96f5af4720c3504124dc00f 100644
--- a/dygraph/ppdet/data/source/coco.py
+++ b/dygraph/ppdet/data/source/coco.py
@@ -65,7 +65,7 @@ class COCODataSet(DetDataset):
                            'and load image information only.'.format(anno_path))
 
         for img_id in img_ids:
-            img_anno = coco.loadImgs(img_id)[0]
+            img_anno = coco.loadImgs([img_id])[0]
             im_fname = img_anno['file_name']
             im_w = float(img_anno['width'])
             im_h = float(img_anno['height'])
@@ -84,7 +84,7 @@ class COCODataSet(DetDataset):
                 continue
 
             if not self.load_image_only:
-                ins_anno_ids = coco.getAnnIds(imgIds=img_id, iscrowd=False)
+                ins_anno_ids = coco.getAnnIds(imgIds=[img_id], iscrowd=False)
                 instances = coco.loadAnns(ins_anno_ids)
 
                 bboxes = []
diff --git a/ppdet/data/source/coco.py b/ppdet/data/source/coco.py
index e393a4609ab5ed4d8558fa1548cb907aeecaa8e6..388f207d75cbb595801114988c694a78e3a48ccf 100644
--- a/ppdet/data/source/coco.py
+++ b/ppdet/data/source/coco.py
@@ -101,7 +101,7 @@ class COCODataSet(DataSet):
                         'and load image information only.'.format(anno_path))
 
         for img_id in img_ids:
-            img_anno = coco.loadImgs(img_id)[0]
+            img_anno = coco.loadImgs([img_id])[0]
             im_fname = img_anno['file_name']
             im_w = float(img_anno['width'])
             im_h = float(img_anno['height'])
@@ -127,7 +127,7 @@ class COCODataSet(DataSet):
             }
 
             if not self.load_image_only:
-                ins_anno_ids = coco.getAnnIds(imgIds=img_id, iscrowd=False)
+                ins_anno_ids = coco.getAnnIds(imgIds=[img_id], iscrowd=False)
                 instances = coco.loadAnns(ins_anno_ids)
 
                 bboxes = []