Skip to content

GitLab

Commit 13e59395 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

add warning when label length greater than setting length

parent 29aea6a2
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
ppocr/data/rec/img_tools.py
View file @ 13e59395
...@@ -359,6 +359,11 @@ def process_image(img, ...@@ -359,6 +359,11 @@ def process_image(img,
# char_num = char_ops.get_char_num() # char_num = char_ops.get_char_num()
text = char_ops.encode(label) text = char_ops.encode(label)
if len(text) == 0 or len(text) > max_text_length: 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 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