https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.7/doc/doc_ch/algorithm_det_db.md
数据
1.2 公开数据集
数据集名称 图片下载地址 PaddleOCR 标注下载地址
ICDAR 2015 https://rrc.cvc.uab.es/?ch=4&com=downloads
ctw1500 https://paddleocr.bj.bcebos.com/dataset/ctw1500.zip
total text https://paddleocr.bj.bcebos.com/dataset/total_text.tar
td tr https://paddleocr.bj.bcebos.com/dataset/TD_TR.tar
配置文件
dbnet++
下载预训练模型,使用ICDAR 2015 做数据集 hmean:0.53
python .\tools\eval.py -c .\configs\det\det_r50_db++_icdar15.yml
[2024/07/29 09:15:55] ppocr INFO: metric eval ***************
[2024/07/29 09:15:55] ppocr INFO: precision:0.6973886328725039
[2024/07/29 09:15:55] ppocr INFO: recall:0.4375903614457831
[2024/07/29 09:15:55] ppocr INFO: hmean:0.5377554042049155
[2024/07/29 09:15:55] ppocr INFO: fps:23.42441488031281
训练中
python .\tools\train.py -c .\configs\det\det_r50_db++_icdar15.yml
rtx 4090训练了大概2天,训练集1000张,验证集500张,差不多达到了 heam0.86理论标准
导出模型
python tools/export_model.py -c .\configs\det\det_r50_db++_icdar15.yml -o Global.save_inference_dir=./inference/det_db++ Global.checkpoints=./output/det_r50_icdar15/best_accuracy
检测
# python tools/infer_det.py -c configs/det/det_r50_db++_icdar15.yml -o Global.checkpoints=./output/det_r50_icdar15/best_accuracy Global.infer_img=./doc/imgs_en/img_12.jpg
检测和识别
python tools/infer/predict_system.py --image_dir="./doc/imgs_en/img_12.jpg" --det_model_dir="./inference/det_db++/" --det_algorithm="DB++" --rec_model_dir="./inference/openatom_rec_svtrv2_ch_infer"
img_10.jpg
效果不错
dbnet
det_r50_vd_db.yml
预训练模型做检测,hmean:0
训练过程:
hmean 0.53
hmean 0.64
hmean 0.68
hmean 0.70
hmean0.73
hmean 0.74
hmean0.81
dbnet算法
https://github.com/PaddlePaddle/PaddleOCR/issues/8117