【PPOCR】PaddleOCR微调en_PP-OCRv3_det_infer检测模型

文章来源于：【PPOCR】PaddleOCR微调en_PP-OCRv3_det_infer检测模型

1、获取微调检测模型的数据

检测模型的数据集的制作和获取具体参考PPOCRLabel进行数据标注，并划分检测数据集的步骤；得到的检测结果如下图所示：

其中trian/val/test文件夹下图片，train.txt等文件是对应的文本框的标注文件；

2、初始权重的获取

en_PP-OCRv3_det	【最新】原始超轻量模型，支持英文、数字检测	[ch_PP-OCRv3_det_cml.yml](file:/././PaddleOCR-release-2.5/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml)	3.8M	

下载地址：

 [https://paddleocr.bj.bcebos.com/PP-OCRv3/english/en_PP-OCRv3_det_distill_train.tar](https://paddleocr.bj.bcebos.com/PP-OCRv3/english/en_PP-OCRv3_det_distill_train.tar)

3、检测模型的推理：

python tools/infer_det.py -c ./configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml -o Global.infer_img=“./train_data/drivingData/” Global.pretrained_model=“./inference_model/en_PP-OCRv3_det_infer/best_accuracy”

4、检测模型的验证：

python tools/eval.py -c ./configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml -o Global.checkpoints=“{path/to/weights}/best_accuracy”

5、检测模型的训练：

有3种模型配置文件加载的方式，对应不同的yml的权重文件： 具体参考：https://aistudio.baidu.com/projectdetail/8236106
训练的细节参考：https://github.com/PaddlePaddle/PaddleOCR/blob/main/doc/doc_ch/finetune.md

4.1单机单卡训练：

python tools/train.py -c ./configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml -o Global.pretrained_model=“./inference_model/en_PP-OCRv3_det_infer/best_accuracy”

4.2混合精度训练：

python3 tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained Global.use_amp=True Global.scale_loss=1024.0 Global.use_dynamic_loss_scaling=True

5、微调结果测试：

python tools/eval.py -c ./configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml -o Global.checkpoints=“./inference_model/en_PP-OCRv3_det_infer/best_accuracy”

5.1 检测模型微调前的精度

5.2 检测模型微调后的精度

训练后的模型各方面的性能都有了提升。。。

5.训练后的模型转静态模型

python tools/export_model.py -c ./configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
-o Global.pretrained_model=./output/best_accuracy
Global.save_inference_dir=./inference/

6.静态模型转onnx支持C++的TensorRT部署

paddle2onnx --model_dir /mnt/d/Downloads/202406/paddelocr/pp/ch_PP-OCRv4_det_infer/ --model_filename inference.pdmodel --params_filename inference.pdiparams --opset_version 11 --save_file /mnt/d/Downloads/202406/paddelocr/onnx/ch_PP-OCRv4_rec_infer.onnx