关于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参数。

猜你喜欢

转载自blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/124767835