RapidAI / PaddleOCRModelConvert

Convert the model in PaddleOCR to ONNX format
Apache License 2.0
61 stars 9 forks source link

转换cls模型报错! #11

Closed situjunhao closed 1 year ago

situjunhao commented 1 year ago

代码: from paddleocr_convert import PaddleOCRModelConvert

converter = PaddleOCRModelConvert() save_dir = 'models' url = 'https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_slim_infer.tar' txt_url = 'https://raw.githubusercontent.com/PaddlePaddle/PaddleOCR/release/2.6/ppocr/utils/ppocr_keys_v1.txt'

converter(url, save_dir, txt_path=txt_url)

报错: Successfully convert model to models/ch_ppocr_mobile_v2.0_cls_slim_infer/ch_ppocr_mobile_v2.0_cls_slim_infer.onnx Traceback (most recent call last): File "/root/miniconda3/envs/rapidocr/lib/python3.11/site-packages/paddleocr_convert/main.py", line 42, in call self.change_to_dynamic(save_onnx_path) File "/root/miniconda3/envs/rapidocr/lib/python3.11/site-packages/paddleocr_convert/main.py", line 108, in change_to_dynamic onnx_model = onnx.load_model(onnx_path) ^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/miniconda3/envs/rapidocr/lib/python3.11/site-packages/onnx/init.py", line 169, in load_model s = _load_bytes(f) ^^^^^^^^^^^^^^ File "/root/miniconda3/envs/rapidocr/lib/python3.11/site-packages/onnx/init.py", line 73, in _load_bytes with open(typing.cast(str, f), "rb") as readable: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ FileNotFoundError: [Errno 2] No such file or directory: 'models/ch_ppocr_mobile_v2.0_cls_slim_infer/ch_ppocr_mobile_v2.0_cls_slim_infer.onnx'

The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "/mnt/workspace/RapidOCR/python/ppmodel2onnx.py", line 8, in converter(url, save_dir, txt_path=txt_url) File "/root/miniconda3/envs/rapidocr/lib/python3.11/site-packages/paddleocr_convert/main.py", line 44, in call raise ConvertError( paddleocr_convert.main.ConvertError: change the model to dynamic meets error.

SWHL commented 1 year ago

不支持slim的模型转换哈