diff --git a/ppdet/utils/coco_eval.py b/ppdet/utils/coco_eval.py
index e7b01d673f2a116e57e7db56097ad7dd882c429b..7a8a5974535612754246b1a19915c6e12f3b5952 100644
--- a/ppdet/utils/coco_eval.py
+++ b/ppdet/utils/coco_eval.py
@@ -144,13 +144,13 @@ def mask2out(results, clsid2catid, resolution, thresh_binarize=0.5):
             continue
 
         masks = t['mask'][0]
-        im_shape = t['im_shape'][0][0]
 
         s = 0
         # for each sample
         for i in range(len(lengths)):
             num = lengths[i]
             im_id = int(im_ids[i][0])
+            im_shape = t['im_shape'][0][i]
 
             bbox = bboxes[s:s + num][:, 2:]
             clsid_scores = bboxes[s:s + num][:, 0:2]