Skip to content

GitLab

"docs/source/en/api/loaders.md" did not exist on "95ea538c7969b74f1da8971dfd3bfe3e794c96cc"
Commit d4b0bfd6 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix img shape not match when concat

parent 0322d0e3
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
ppocr/data/det/dataset_traversal.py
View file @ d4b0bfd6
...@@ -94,6 +94,8 @@ class EvalTestReader(object): ...@@ -94,6 +94,8 @@ class EvalTestReader(object):
batch_outs = [] batch_outs = []
for img_path in img_list: for img_path in img_list:
img = cv2.imread(img_path) img = cv2.imread(img_path)
if len(list(img.shape)) == 2 or img.shape[2] == 1:
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
if img is None: if img is None:
logger.info("load image error:" + img_path) logger.info("load image error:" + img_path)
continue continue
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment