AI-Insects-Challenge
第二名代码分享
简介
本赛题是一个标准的目标检测任务,主要对纯色器皿中的七种虫子的位置和类别进行检测。本解决方案主要包含了两个模块,分别是YoloV3的目标检测模块,以及后置的SENet分类矫正模块。两个模块直接串联,先通过YoloV3得出检测框和分类结果,然后将检测框裁剪出来使用分类网络进一步矫正分类结果,最后通过一个改进的NMS方案得到最终结果。
注:本代码均使用paddlepaddle 1.6.1 的静态网络搭建
图像增广方案
MixUp (前期训练有效提高,后期精调建议不用)
随机旋转图像
旋转从理论上来说很有用,但是旋转之后的真实box会变大,不再紧贴目标,尤其是45度旋转的时候,暂时没有找到更优的解决方案。前两天又看到opencv中貌似有类似的处理,回头试一下再改进。
随机色彩变换
亮度、对比度、饱和度等色彩相关变换
随机扩充
本文地址:http://ww.kub2b.com/news/17252.html
企库往 http://ww.kub2b.com/ , 查看更多