文本检测使用icdar2015训练结果不理想
我是移动者 发布于2020-11 浏览:1863 回复:1
0
收藏

我使用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
个赞
共1条回复 最后由用户已被禁言回复于2022-04
#2三叠三ぃ梦回复于2021-07

同问,我的是release2.1,目前训练了130个,似乎loss不收敛了,太奇怪了。用了旧版的训练,hmean也只是0.71而已

0
TOP
切换版块