一种基于YOLOv7的轻量级无人机图像落水人员检测算法

本发明属于深度学习目标检测技术领域，尤其为一种基于YOLOv7的轻量级无人机图像落水人员检测算法，包括以下步骤：S1、搜集适量游泳和落水人员图片，制作数据集；S2、将YOLOv7骨干网络替换为MobileNetV4；S3、重构YOLOv7的SPPCSPC结构，并引入CA注意力机制与Wise‑IoU损失函数；S4、在自制数据集上进行消融实验；S5、在自制数据集上将改进的算法与其他目标检测算法进行对比。本发明提出的算法在仅损失少量检测精度的前提下，大幅减少了原算法的参数量和计算量，并大幅提高了检测速度，通过轻量化原有算法，使其更易于部署在无人机等算力有限的边缘设备上。