有谁能给一个最新的,从环境到,数据准备,到训练,det, rec, kie, 导出,部署到paddle lite的完整的流程吗 #14934
-
说实话,用这个ocr,我都怀疑我的智商了,老是问题不断,没个头啊! |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 3 replies
-
在 PaddleOCR 上完成从环境配置、数据准备、模型训练(包括文本检测 1. 环境配置安装 PaddlePaddle请根据你的设备来选择合适的 PaddlePaddle 版本: # CPU 版本
pip install paddlepaddle -i https://pypi.tuna.tsinghua.edu.cn/simple
# GPU 版本 (请确保安装了 CUDA 和 cuDNN)
pip install paddlepaddle-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple 安装 PaddleOCRgit clone https://github.com/PaddlePaddle/PaddleOCR.git
cd PaddleOCR
pip install -r requirements.txt 2. 数据准备PaddleOCR 默认支持
3. 模型训练(1)文本检测 (DET)使用 DBNet 进行文本检测: python tools/train.py -c configs/det/ch_ppocr_v4_det.yml 训练完成后,模型会保存在 (2)文本识别 (REC)训练文本识别模型: python tools/train.py -c configs/rec/ch_PP-OCRv4_rec.yml 输出目录: (3)关键信息提取 (KIE)训练 KIE (如 python tools/train.py -c configs/kie/xyz_kie.yml 4. 模型导出导出为推理模型: python tools/export_model.py -c configs/det/ch_ppocr_v4_det.yml \
-o Global.pretrained_model=output/ch_PP-OCRv4_det/best_accuracy \
Global.save_inference_dir=./inference/ch_ppocrv4_det 对于 python tools/export_model.py -c configs/rec/ch_PP-OCRv4_rec.yml \
-o Global.pretrained_model=output/ch_PP-OCRv4_rec/best_accuracy \
Global.save_inference_dir=./inference/ch_ppocrv4_rec 5. Paddle Lite 推理部署(1)模型转换转换为 Paddle Lite 格式: paddle_lite_opt --model_dir=inference/ch_ppocrv4_det \
--valid_targets=arm \
--optimize_out_type=naive_buffer \
--optimize_out=./lite_model/ch_ppocrv4_det_opt 对 paddle_lite_opt --model_dir=inference/ch_ppocrv4_rec \
--valid_targets=arm \
--optimize_out=./lite_model/ch_ppocrv4_rec_opt (2)部署使用 Paddle Lite API 进行推理: import paddlelite.lite as lite
config = lite.MobileConfig()
config.set_model_from_file("./lite_model/ch_ppocrv4_det_opt.nb")
predictor = lite.create_paddle_predictor(config)
# 进行推理 6. 常见问题解析在你的环境中出现错误:
可能的原因:
此外,可以参考这个 issue 来检查海光 CPU 的相关兼容性: 总结
如果你在某一步遇到了问题,可以检查对应的 log 并根据具体错误进行修改! Response generated by 🤖 feifei-bot | chatgpt-4o-latest |
Beta Was this translation helpful? Give feedback.
-
你看官网文档应该是比较新的,能跑通 |
Beta Was this translation helpful? Give feedback.
找到原因了,必须使用2.14rc的paddlelite