摘要
本发明公开了基线网络轻量化的无人机小目标检测方法,包括:对YOLOv8s模型进行改进并使用收集的无人机小目标数据集进行训练,得到YOLOv8s‑LW模型;使用YOLOv8s‑LW模型完成无人机小目标检测,改进方法包括:使用幽灵卷积和HGNetV2构建轻量化主干网络,并替换YOLOv8s基线主干结构;分别使用幽灵卷积和改进模块对C2f进行改进,并使用Drop Path正则化提升泛化能力;构造新的损失函数替换基线算法本身的CIoU损失函数,解决了无人机在执行目标检测任务中小目标像素低、背景复杂、目标密度大以及硬件资源有限等问题。