Cnocr
CnOCR
breezedeus • Updated May 16, 2024
•
context
:预测使用的机器资源,可取值为字符串cpu
、gpu
、cuda:0
等。默认为 cpu
。此参数仅在 model_backend=='pytorch'
时有效。
•
det_model_backend
:'pytorch', or 'onnx'。表明检测时是使用 PyTorch
版本模型,还是使用 ONNX
版本模型。 同样的模型,ONNX 版本的预测速度一般是 PyTorch 版本的 2倍左右。 默认为 'onnx'。用默认配置,即吃 CPU 也吃 GPU(在并发的情况下)qps 为 7
# 识别模型
rec_model_name: str = 'densenet',
# 检测模型
det_model_name: str = 'ch_PP-OCRv3_det',
Loading...