diff --git a/ppocr/data/det/dataset_traversal.py b/ppocr/data/det/dataset_traversal.py
index d7c74078fb615b0852ff4bf81bc6b72e9b72ad85..16866b0e86727480d2f6a8ee02da2148365ca45a 100644
--- a/ppocr/data/det/dataset_traversal.py
+++ b/ppocr/data/det/dataset_traversal.py
@@ -94,6 +94,9 @@ class EvalTestReader(object):
             batch_outs = []
             for img_path in img_list:
                 img = cv2.imread(img_path)
+                if img is None:
+                    logger.info("{} does not exist!".format(img_path))
+                    continue
                 if len(list(img.shape)) == 2 or img.shape[2] == 1:
                     img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
                 if img is None:
diff --git a/ppocr/data/det/db_process.py b/ppocr/data/det/db_process.py
index 0993324aa92a6765a324472f9137f266b2dabb52..ee131501724123ac2d718e596022c3ee266077a8 100644
--- a/ppocr/data/det/db_process.py
+++ b/ppocr/data/det/db_process.py
@@ -104,6 +104,8 @@ class DBProcessTrain(object):
         if imgvalue is None:
             logger.info("{} does not exist!".format(img_path))
             return None
+        if len(list(imgvalue.shape)) == 2 or imgvalue.shape[2] == 1:
+            imgvalue = cv2.cvtColor(imgvalue, cv2.COLOR_GRAY2BGR)
         data = self.make_data_dict(imgvalue, gt_label)
         data = AugmentData(data)
         data = RandomCropData(data, self.image_shape[1:])