diff --git a/ppdet/data/source/coco.py b/ppdet/data/source/coco.py
index d2d16f1f48625c28322e844c2126d6c696c066bc..ea97f66750db7b0c4b8c23e1261f2218027cb7fe 100644
--- a/ppdet/data/source/coco.py
+++ b/ppdet/data/source/coco.py
@@ -137,12 +137,14 @@ class COCODataSet(DataSet):
                     y1 = max(0, y)
                     x2 = min(im_w - 1, x1 + max(0, box_w - 1))
                     y2 = min(im_h - 1, y1 + max(0, box_h - 1))
-                    if 'segmentation' in inst and len(inst[
-                            'segmentation']) == 0:
-                        logger.warn(
-                            'Found an empty segmentation in annotations: im_id: {}.'.
-                            format(img_id))
-                    elif x2 >= x1 and y2 >= y1:
+                    if 'segmentation' in inst:
+                        if inst['segmentation'] is None or len(inst[
+                                'segmentation']) == 0:
+                            logger.warn(
+                                'Found an empty segmentation in annotations: im_id: {}.'.
+                                format(img_id))
+                            continue
+                    if x2 >= x1 and y2 >= y1:
                         inst['clean_bbox'] = [x1, y1, x2, y2]
                         bboxes.append(inst)
                     else: