关于Paddle OCR检测器检测框偏小的解决方法

问题描述一

对Paddle OCR的检测器做了训练,训练完成后测试检测效果,发现检测框偏小。如下图:

image-20220513175158170

然后,百度一番,发现没有找到解决的方法,询问官方的技术大牛,得到了答案。

将yaml文件里面字段”unclip_ratio”的值调大。例如:configs/det_mv3_db.yml文件:

将默认值1.5改为3.0。

image-20220513175644145

然后重新训练(我重新训练,可以试试不训练),再测试,就可以得到一个比较好的检测效果。

image-20220513175906945

问题描述二:

在测试评估没有问题,转为推理模型,再做推理测试,发现又变小了,然后追踪推理的代码。发现推理的配置信息在tools/infer/utillity.py文件中。

image-20220514141653765

这个文件里面也有一个det_db_unclip_ratio,修改这里的unclip_ratio,检测框正常了。

总结

如果是验证评估时,检测框大小有偏差则修改yml文件中的unclip_ratio参数,如果是在推理阶段出现检测框大小有偏差,则调整utillity.py脚本中的det_db_unclip_ratio参数。

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2022年5月16日
下一篇 2022年5月16日

相关推荐

此站出售,如需请站内私信或者邮箱!