revert change in db_process.py

5 years ago · 98c80f161a
parent 72b453b9d1
commit 98c80f161a
1 changed files with 1 additions and 1 deletions
--- 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)