未验证 提交 778db1ae 编写于 作者: D DeepGeGe 提交者: GitHub

Fix ocr text detection module bugs.

上级 7089f7b5
......@@ -389,7 +389,7 @@ class ChineseOCRDBCRNN(hub.Module):
# rec_res.append([preds_text, score])
rec_res[indices[beg_img_no + rno]] = [preds_text, score]
return rec_res
return rec_res
def save_inference_model(self, dirname, model_filename=None, params_filename=None, combined=True):
detector_dir = os.path.join(dirname, 'text_detector')
......
......@@ -393,7 +393,7 @@ class ChineseOCRDBCRNNServer(hub.Module):
# rec_res.append([preds_text, score])
rec_res[indices[beg_img_no + rno]] = [preds_text, score]
return rec_res
return rec_res
def save_inference_model(self, dirname, model_filename=None, params_filename=None, combined=True):
detector_dir = os.path.join(dirname, 'text_detector')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册