PaddleOCR
PaddleOCR copied to clipboard
使用pp_OCRv2训练的检测模型,训练效果很好,转换成推理模型后,推理结果很查
用的与训练模型是ch_PP-OCRv2_det_distill_train
基于ch_PP-OCRv2_det_student.yml训练日期检测模型。
使用训练完的模型直接测试,infer_det.py,效果很好。
然后使用export_model.py对best_accuracy模型进行转换为推理模型(基于训练时的配置表config.yml),得到inference模型,使用predict_det.py做预测。效果很不好。
如何保证predict_det.py的效果和infer_det.py一致?
后处理问题,看这里https://blog.csdn.net/YY007H/article/details/125506785
Infer_det.py https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml#L161 Default: https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/ppocr/data/imaug/operators.py#L221-L223
Predict_det.py https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/tools/infer/utility.py#L51-L52