首页|改进YOLOv5的遥感图像目标检测

改进YOLOv5的遥感图像目标检测

扫码查看
原文链接
  • NETL
  • NSTL
  • 万方数据
  • 维普
针对基于YOLOv5算法的小目标的检测精度低、密集目标识别效果差的问题,提出了一种改进YOLOv5目标检测 框架.YOLOv5的主干网络加入卷积块注意力模块(Convolutional block attention module,CBAM)可提高特征提取能力,增强网络对图像纹理的感知能力,使小目标获取更多关注.为了解决密集 目标检测的漏检问题,YOLOv5的颈部网络使用加权双向特征网络(Bidirectional feature pyramid network,BiFPN)替代原有的像素聚合网络(Path aggregation network,PAN),通过权值共享的方式实现多尺度特征融合.采用EIoU作为模型的边界框回归损失函数,提高了边界框回归性能,加快网络收敛速度.在DOTA数据集上,实验验证了 YOLOv5的改进结果,此方法的mAP为80.0％,能够检测更多的目标,相较于YOLOv5,改进YOLOv5的mAP提升了5.2％.
Target detection in remote sensing image based on improved YOLOv5
For the problem of low detection accuracy of small targets and poor recognition of dense targets based on YOLOv5 algorithm,the improved YOLOv5 target detection framework is proposed.The backbone network of YOLOv5 adds the Convolutional block attention module(CBAM)to enhance the network's ability to perceive the image texture so that the small targets get more attention.To solve the leakage problem of dense target detection,the neck network of YOLOv5 uses Bidirectional feature pyramid network(BiFPN)instead of the Path aggregation network(PAN)to realize multi-scale feature fusion by weight sharing.EIoU is used as the bounding box regression loss function of the model to strengthen the performance of bounding box regression and accelerate the network convergence.Experimental results on the DOTA dataset validate the improvement of YOLOv5.The mAP of the enhanced method is 80.0％,and the improved algorithm is able to detect more targets.Compared to YOLOv5,the mAP of the enhanced YOLOv5 is improved by 5.2％.

target detectionYOLOv5CBAMBiFPNEIoU

刘国新、朱福珍、巫红

展开 >

黑龙江大学电子工程学院,哈尔滨 150080

目标检测 YOLOv5算法 卷积块注意力模块 加权双向特征金字塔 EIoU

黑龙江省省属高校基本科研业务费项目黑龙江省省属高校基本科研业务费项目国家自然科学基金资助项目国家自然科学基金资助项目黑龙江省"双一流"学科协同创新成果孵化项目黑龙江大学横向课题项目

2023-KYYWF-14362022-KYYWF-10906160117462341503LJGXCG2023-0462023230101001032

2024

10.13482/j.issn1001-7011.2024.01.001

黑龙江大学自然科学学报
黑龙江大学

黑龙江大学自然科学学报

CSTPCD
影响因子：0.27
ISSN：1001-7011
年,卷(期)：2024.41(1)
  • 25