53 lines
1.4 KiB
Markdown
53 lines
1.4 KiB
Markdown
# 相关网站
|
||
|
||
- github：https://github.com/PaddleCV-SIG/PaddleLabel
|
||
- 官网：
|
||
|
||
|
||
|
||
# 加载自定义模型
|
||
|
||
## 1. 修改 PPOCRLabel.py
|
||
|
||
```python
|
||
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,
|
||
)
|
||
```
|
||
|
||
## 修改启动命令
|
||
|
||
```bash
|
||
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\"
|
||
```
|
||
|