提交 1c8cddf4 编写于 作者: T tink2123

fix resize norm for ch

上级 87ce916a
...@@ -53,6 +53,7 @@ def resize_norm_img(img, image_shape): ...@@ -53,6 +53,7 @@ def resize_norm_img(img, image_shape):
def resize_norm_img_chinese(img, image_shape): def resize_norm_img_chinese(img, image_shape):
imgC, imgH, imgW = image_shape imgC, imgH, imgW = image_shape
# todo: change to 0 and modified image shape # todo: change to 0 and modified image shape
imgC = 3
max_wh_ratio = 0 max_wh_ratio = 0
h, w = img.shape[0], img.shape[1] h, w = img.shape[0], img.shape[1]
ratio = w * 1.0 / h ratio = w * 1.0 / h
...@@ -97,10 +98,11 @@ def process_image(img, ...@@ -97,10 +98,11 @@ 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 and (char_ops.character_type == "en" or tps != None): if infer_mode and char_ops.character_type == "ch" and not tps:
norm_img = resize_norm_img(img, image_shape)
else:
norm_img = resize_norm_img_chinese(img, image_shape) norm_img = resize_norm_img_chinese(img, image_shape)
else:
norm_img = resize_norm_img(img, image_shape)
norm_img = norm_img[np.newaxis, :] norm_img = norm_img[np.newaxis, :]
if label is not None: if label is not None:
# char_num = char_ops.get_char_num() # char_num = char_ops.get_char_num()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册