Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "335d372237389a85f4426cf1a05ccedca6923518"
Commit da2d01b3 authored by chenxj's avatar chenxj
Browse files

solve rec num 0 bug

parent 66e68533
Show 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): ...@@ -102,6 +102,8 @@ class TextRecognizer(object):
indices = np.argsort(np.array(width_list)) indices = np.argsort(np.array(width_list))
rec_res = [['', 0.0]] * img_num rec_res = [['', 0.0]] * img_num
if img_num <= 0:
return rec_res, 0
max_batnum = 24 max_batnum = 24
min_batnum = 8 min_batnum = 8
if os.environ.get("OCR_REC_MAX_BATNUM") is not None: 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