Merge pull request #233 from LDOUBLEV/fixocr

fix img shape not match when concat
PaddlePaddle · Jun 19, 2020 · dfa00e1 · dfa00e1
2 parents 853a586 + d4b0bfd
commit dfa00e1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ppocr/data/det/dataset_traversal.py b/ppocr/data/det/dataset_traversal.py
@@ -94,6 +94,8 @@ def batch_iter_reader():
  batch_outs = []
  for img_path in img_list:
  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:
  logger.info("load image error:" + img_path)
  continue