diff --git a/ppocr/data/det/db_process.py b/ppocr/data/det/db_process.py index 72c9acd921d4ffa3e60057c1293953e7968f7777..ee131501724123ac2d718e596022c3ee266077a8 100644 --- a/ppocr/data/det/db_process.py +++ b/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 - elif len(list(imgvalue.shape)) == 2 or imgvalue.shape[2] == 1: + 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)