Skip to content

GitLab

Commit da2d01b3 authored by chenxj's avatar chenxj
Browse files

solve rec num 0 bug

parent 66e68533
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tools/infer/predict_rec.py
View file @ da2d01b3
......@@ -102,6 +102,8 @@ class TextRecognizer(object):
indices = np.argsort(np.array(width_list))
rec_res = [['', 0.0]] * img_num
if img_num <= 0:
return rec_res, 0
max_batnum = 24
min_batnum = 8
if os.environ.get("OCR_REC_MAX_BATNUM") is not None:
......
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