add continue in det Evalreader when img is not exist

2020-07-14 15:48:43 +08:00 · 2020-07-14 15:48:43 +08:00 · 52e7dbf773
parent eed41762b7
commit 52e7dbf773
1 changed files with 1 additions and 0 deletions
--- a/ppocr/data/det/dataset_traversal.py
+++ b/ppocr/data/det/dataset_traversal.py
@ -96,6 +96,7 @@ class EvalTestReader(object):
                img = cv2.imread(img_path)
                if img is None:
                    logger.info("{} does not exist!".format(img_path))
+                    continue
                elif len(list(img.shape)) == 2 or img.shape[2] == 1:
                    img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
                outs = process_function(img)