workspace/notes/后端开发/paddlepaddle/ppocrlabel.md

相关网站

• github：https://github.com/PaddleCV-SIG/PaddleLabel
• 官网：

加载自定义模型

1. 修改 PPOCRLabel.py

 params = {
    "use_doc_orientation_classify": False,
    "use_doc_unwarping": False,
    "use_textline_orientation": True,
    "device": self.gpu,
    "lang": self.lang,
    "text_detection_model_name": "PP-OCRv5_mobile_det", # 修改为自己的模型名称
    "text_recognition_model_name": "PP-OCRv5_mobile_rec", # 修改为自己的模型名称
    "enable_mkldnn": False,
}

if det_model_dir is not None:
    params["text_detection_model_dir"] = det_model_dir
if rec_model_dir is not None:
    params["text_recognition_model_dir"] = rec_model_dir
if cls_model_dir is not None:
    params["text_line_orientation_model_dir"] = cls_model_dir

self.ocr = PaddleOCR(**params)
self.text_recognizer = TextRecognition(
    model_name="PP-OCRv5_mobile_rec", # 修改为自己的模型名称
    model_dir=rec_model_dir,
    device=self.gpu,
)
self.text_detector = TextDetection(
    model_name="PP-OCRv5_mobile_det", # 修改为自己的模型名称
    model_dir=det_model_dir,
    device=self.gpu,
)

修改启动命令

python PPOCRLabel.py --lang ch --kie True --det_model_dir "模型路径" --rec_model_dir "模型路径"

# 示例
python PPOCRLabel.py --lang ch --kie True --det_model_dir ".\model\det_inference\" --rec_model_dir ".\model\rec_inference\"