我使用PaddleOCR训练文本检测模型
数据集:icdar2015
预训练模型:MobileNetV3_large_x0_5_pretrained
det_db_mv3_v1.1.yml配置文件修改了:
epoch_num: 300
save_epoch_step: 100
机器安装的是GPU版本的paddle,在./output/det_db/下得到了
best_accuracy.pdmodel
best_accuracy.pdopt
best_accuracy.pdparams
接着使用
python tools/infer_det.py -c configs/det/det_mv3_db_v1.1.yml -o TestReader.infer_img="test_ocr.jpg" Global.checkpoints="./output/det_db/best_accuracy"
得到的结果非常不理想:
是数据集不够?还是训练模型的问题呢?
收藏
点赞
0
个赞
请登录后评论
TOP
切换版块
同问,我的是release2.1,目前训练了130个,似乎loss不收敛了,太奇怪了。用了旧版的训练,hmean也只是0.71而已