PaddleOCR训练手写文字识别模型

最新推荐文章于 2024-03-04 16:10:14 发布

ronshi

最新推荐文章于 2024-03-04 16:10:14 发布

阅读量1.4k

点赞数 3

分类专栏： AI 文章标签： python ai

本文链接：https://blog.csdn.net/ronshi/article/details/133750680

版权

AI 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

1、安装PaddleOCR

git clone https://github.com/PaddlePaddle/PaddleOCR.git
cd PaddleOCR
pip install -r requirements.txt

2、下载数据集

http://www.nlpr.ia.ac.cn/databases/handwriting/Download.html

3、下载预训练模型

https://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_rec_train.tar

4、修改配置文件

configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml，主要修改训练轮数和学习率相关参数，设置预训练模型路径，设置数据集路径。另外，batch_size可根据自己机器显存大小进行调整。

5、训练模型

python tools/train.py -c configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml

6、评估模型

# 评估预训练模型
python tools/eval.py -c configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml -o Global.pretrained_model="./pretrained_models/ch_PP-OCRv3_rec_train/best_accuracy"

# 评估finetune效果
python tools/eval.py -c configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml -o Global.pretrained_model="./output/ch_PP-OCR_v3_rec/best_accuracy"

7、导出模型

# 转化为推理模型
python tools/export_model.py -c configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml -o Global.pretrained_model="./output/ch_PP-OCR_v3_rec/best_accuracy" Global.save_inference_dir="./inference/rec_ppocrv3/"

8、使用模型推理预测

# 推理预测
python tools/infer/predict_rec.py --image_dir="train_data/handwrite/HWDB2.0Test_images/104-P16_4.jpg" --rec_model_dir="./inference/rec_ppocrv3/Student"