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

Posted 2022-05-18 AI浩

文章目录

• 问题描述一
• 问题描述二：
• 总结

问题描述一

对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参数。