add warning when label length greater than setting length

This commit is contained in:
LDOUBLEV 2020-07-10 15:39:42 +08:00
parent 29aea6a206
commit 13e59395cc
1 changed files with 5 additions and 0 deletions

View File

@ -359,6 +359,11 @@ def process_image(img,
# char_num = char_ops.get_char_num()
text = char_ops.encode(label)
if len(text) == 0 or len(text) > max_text_length:
logger.info(
"Warning in ppocr/data/rec/img_tools.py:line362: Wrong data type."
"Excepted string with length between 1 and {}, but "
"got '{}'. Label is '{}'".format(max_text_length,
len(text), label))
return None
else:
if loss_type == "ctc":