基于改进的Yolov5算法的无人机图像目标检测技术研究

Ultralytics公司于2020年发布的YOLOv5检测算法在无人机图像目标检测领域具有很高的应用潜力。针对无人机城市巡检场景中的需求，本研究对YOLOv5n算法进行了优化与改进。通过在输入端引入Mosaic数据增强和在主干网络中引入BoT模块，我们既提高了检测速度又保持了检测精度。此外，特征融合网络中引入的CBAM注意力机制进一步增强了小目标的特征提取效果。实验结果表明，改进后的算法在无人机图像目标检测方面取得了显著的性能提升。