29 lines
1.6 KiB
Python
29 lines
1.6 KiB
Python
from paddleocr import PaddleOCR
|
|
|
|
ocr = PaddleOCR(
|
|
# 文本检测模型地址
|
|
# text_detection_model_dir = "/home/admin-root/haotian/康达瑞贝斯机器狗/ocr_model/PP-OCRv5_server_det",
|
|
text_detection_model_dir='/home/admin-root/haotian/康达瑞贝斯机器狗/PaddleOCR-3.1.0/output/PP-OCRv5_server_det_infer_20250814',
|
|
# 文本识别模型地址
|
|
# text_recognition_model_dir = "/home/admin-root/haotian/康达瑞贝斯机器狗/ocr_model/PP-OCRv5_server_rec",
|
|
text_recognition_model_dir='/home/admin-root/haotian/康达瑞贝斯机器狗/PaddleOCR-3.1.0/output/PP-OCRv5_server_rec_infer_20250815',
|
|
use_doc_orientation_classify=False,
|
|
use_doc_unwarping=False,
|
|
use_textline_orientation=False) # 文本检测+文本识别
|
|
# ocr = PaddleOCR(use_doc_orientation_classify=True, use_doc_unwarping=True) # 文本图像预处理+文本检测+方向分类+文本识别
|
|
# ocr = PaddleOCR(use_doc_orientation_classify=False, use_doc_unwarping=False) # 文本检测+文本行方向分类+文本识别
|
|
# ocr = PaddleOCR(
|
|
# text_detection_model_name="PP-OCRv5_mobile_det",
|
|
# text_recognition_model_name="PP-OCRv5_mobile_rec",
|
|
# use_doc_orientation_classify=False,
|
|
# use_doc_unwarping=False,
|
|
# use_textline_orientation=False) # 更换 PP-OCRv5_mobile 模型
|
|
result = ocr.predict("/home/admin-root/haotian/康达瑞贝斯机器狗/data_image/001读表图片/3aee64cc1f90d93a5a45979f7b17cb4b_frame_001460.jpg")
|
|
|
|
# print(result[0].shape)
|
|
# for res in result:
|
|
# res.print()
|
|
|
|
# # 输出文件夹
|
|
# res.save_to_img("output")
|
|
# res.save_to_json("output") |