make sure all images are RGB

ppocr/data/det/dataset_traversal.py

@@ -97,11 +97,8 @@ class EvalTestReader(object):
                 if img is None:
                     logger.info("{} does not exist!".format(img_path))
                     continue
-                if len(list(img.shape)) == 2 or img.shape[2] == 1:
+                elif len(list(img.shape)) == 2 or img.shape[2] == 1:
                     img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
-                if img is None:
-                    logger.info("load image error:" + img_path)
-                    continue
                 outs = process_function(img)
                 outs.append(img_path)
                 batch_outs.append(outs)

ppocr/data/det/db_process.py

@@ -104,7 +104,7 @@ 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:
+        elif 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)