提交 87ce916a 编写于 作者: T tink2123

fix resize shape

上级 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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册