Commit 87ce916a authored by tink2123's avatar tink2123
Browse files

fix resize shape

parent 25f16f24
...@@ -97,7 +97,7 @@ def process_image(img, ...@@ -97,7 +97,7 @@ def process_image(img,
max_text_length=None, max_text_length=None,
tps=None, tps=None,
infer_mode=False): infer_mode=False):
if not infer_mode or char_ops.character_type == "en" or tps != None: if not infer_mode and (char_ops.character_type == "en" or tps != None):
norm_img = resize_norm_img(img, image_shape) norm_img = resize_norm_img(img, image_shape)
else: else:
norm_img = resize_norm_img_chinese(img, image_shape) norm_img = resize_norm_img_chinese(img, image_shape)
...@@ -109,7 +109,8 @@ def process_image(img, ...@@ -109,7 +109,8 @@ def process_image(img,
logger.info( logger.info(
"Warning in ppocr/data/rec/img_tools.py:line106: Wrong data type." "Warning in ppocr/data/rec/img_tools.py:line106: Wrong data type."
"Excepted string with length between 1 and {}, but " "Excepted string with length between 1 and {}, but "
"got '{}'. Label is '{}'".format(max_text_length, len(text),label)) "got '{}'. Label is '{}'".format(max_text_length,
len(text), label))
return None return None
else: else:
if loss_type == "ctc": if loss_type == "ctc":
......
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