摘要 返回
     
改进YOLOv5的遥感图像小目标检测算法
 

针对目前主流算法在遥感图像目标检测任务中对于图像中小目标检测效果较差、易产生漏检误检的问题,提出一种改进 YOLOv5 的小目标检测算法 YOLOv5-FRM。首先在原 YOLOv5 骨干网络的最后添加坐标注意力机制Coordinate attention(CA)模块替代原SPP模块,之后提出一种改进多尺度空间净化模块,实现了检测头的添加,并融合进原YOLOv5的颈部网络中。最后引入Copy-reduce-paste数据增强方法提高模型训练效果。实验结果表明,该改进算法有效提升了遥感图像小目标的检测精度,降低了误检率与漏检率。

 
(全文阅读请到中国期刊全文数据库)



 
   
读者园地  
   
 
 
地址:  杭州市五常港路121号西溪创智中心四号楼807室    《计算机时代》编辑部
邮编:  310012      电话:  (0571)85118010   (0571)85119435   E-mail:  Computer_era @ vip.163.com
浙ICP备10035352号