一种基于Transformer-YOLOv5s的无人机航拍图像复杂小目标检测算法

本发明提出了一种基于Transformer‑YOLOv5s的无人机航拍图像复杂小目标检测算法，以提升在复杂背景下的检测精度。数据集部分从VisDrone2019数据集中挑选出5000张具有背景复杂、目标密集等特性的图像，并结合无人机实地航拍进行数据采集，进一步增强模型的适应性；算法改进方面，在YOLOv5s的C3模块中嵌入Transformer编码器以强化特征提取；并额外设计了160x160的极小目标检测头以提升小目标检测效果；此外，将损失函数替换为SIoU损失函数，优化模型定位精度；本发明还使用k‑means++算法构建适合无人机小目标的锚框尺度。实验结果表明，该改进算法在航拍复杂背景下的小目标检测任务中表现出更高的检测精度和鲁棒性，实现了无人机检测过程的高实时性，高精确性。