Skip to content

Commit bdebcb7

Browse files
authored
Merge pull request #149 from liangxhao/fix_infer
fix bug for inference
2 parents e92a268 + 1a4e44e commit bdebcb7

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

deploy/mx_infer/processors/common/collect_process.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,8 @@ def result_handle(self, input_data):
7272
for result in input_data.infer_result:
7373
self.image_pipeline_res[input_data.image_name].append(
7474
{"transcription": result[-1], "points": result[:-1]})
75+
if not input_data.infer_result:
76+
self.image_pipeline_res[input_data.image_name] = []
7577
elif self.task_type == InferModelComb.DET:
7678
self.image_pipeline_res[input_data.image_name].extend(input_data.infer_result)
7779
elif self.task_type == InferModelComb.REC:

docs/cn/inference_cn.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ MindOCR的推理工具集成了文本检测、角度分类和文字识别模块
2222
| ------------------------------ | ------------------------------------------------------------ |
2323
| Paddle PP-OCR server 2.0 DBNet | https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_det_infer.tar |
2424
| Paddle PP-OCR server 2.0 Cls | https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar |
25-
| Paddle PP-OCR server 2.0 SVTR | https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar |
25+
| Paddle PP-OCR server 2.0 CRNN | https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar |
2626

2727
(2) Paddle PP-OCR 3.0模型:
2828

0 commit comments

Comments
 (0)