问题描述一
对Paddle OCR的检测器做了训练,训练完成后测试检测效果,发现检测框偏小。如下图:
然后,百度一番,发现没有找到解决的方法,询问官方的技术大牛,得到了答案。
将yaml文件里面字段”unclip_ratio”的值调大。例如:configs/det_mv3_db.yml文件:
将默认值1.5改为3.0。
然后重新训练(我重新训练,可以试试不训练),再测试,就可以得到一个比较好的检测效果。
问题描述二:
在测试评估没有问题,转为推理模型,再做推理测试,发现又变小了,然后追踪推理的代码。发现推理的配置信息在tools/infer/utillity.py文件中。
这个文件里面也有一个det_db_unclip_ratio,修改这里的unclip_ratio,检测框正常了。
总结
如果是验证评估时,检测框大小有偏差则修改yml文件中的unclip_ratio参数,如果是在推理阶段出现检测框大小有偏差,则调整utillity.py脚本中的det_db_unclip_ratio参数。
文章出处登录后可见!
已经登录?立即刷新