WS⁃YOLO:航拍视角小目标检测方法

  • 打印
  • 收藏
收藏成功


打开文本图片集

摘  要: 航拍小目标检测中,由于图像背景复杂、目标尺度小、空间尺度变化大,导致无人机航拍出现漏检、误检等一系列问题。为提高无人机航拍检测准确性、实时性以及实现轻量化,文中基于YOLOv8n提出一种改进的无人机航拍视角小目标检测方法WS⁃YOLO。首先,重构YOLOv8n网络结构,并增加160×160特征图对应预测头,以提高模型对小尺度目标检测的准确性与鲁棒性;然后,主干网络卷积层后嵌入SPD,防止细粒度信息的丢失并学习不太有效的特征表示,以提高对低分辨率图像的识别能力;最后,将CIoU损失函数替换为WIoU(v3)损失函数,以减轻图片质量低下对检测过程的影响。(剩余10402字)

monitor