如何保存与加载PaddleOCR模型？代码实现方法咨询

PaddleOCR的PaddleOCR实例封装了检测、识别、角度分类三个子模型，没有直接的一键保存方法，需要分别保存各子模型的参数；加载时通过指定模型路径初始化实例即可。

一、保存模型代码实现 #

你可以借助PaddlePaddle的paddle.save函数，分别保存检测、识别、分类模型的参数到指定路径：

import paddle
from paddleocr import PaddleOCR

# 初始化模型
model = PaddleOCR(use_angle_cls=True, lang='en')

save_path = '/some/custom/path'
# 分别创建三个子模型的保存目录（可根据需求调整）
det_save_dir = f"{save_path}/det"
rec_save_dir = f"{save_path}/rec"
cls_save_dir = f"{save_path}/cls"

# 保存检测模型参数
paddle.save(model.det_model.state_dict(), det_save_dir)
# 保存识别模型参数
paddle.save(model.rec_model.state_dict(), rec_save_dir)
# 保存角度分类模型参数
paddle.save(model.cls_model.state_dict(), cls_save_dir)

二、加载模型代码实现 #

初始化PaddleOCR时，通过det_model_dir、rec_model_dir、cls_model_dir参数指定已保存的模型路径即可：

from paddleocr import PaddleOCR

save_path = '/some/custom/path'
det_model_path = f"{save_path}/det"
rec_model_path = f"{save_path}/rec"
cls_model_path = f"{save_path}/cls"

# 加载保存的模型
model = PaddleOCR(
    use_angle_cls=True,
    lang='en',
    det_model_dir=det_model_path,
    rec_model_dir=rec_model_path,
    cls_model_dir=cls_model_path
)

# 测试加载后的模型（示例）
result = model.ocr('test_image.jpg', cls=True)
print(result)

注意：

• 保存时要确保目标路径有写入权限，加载时路径需与保存时完全对应
• 如果不需要角度分类功能，可去掉use_angle_cls=True和cls_model_dir相关代码

内容的提问来源于stack exchange，提问作者Alcibiades