公开(公告)号：CN116091946A

公开(公告)日：2023-05-09

申请号：CN202211559260.3

申请日：2022-12-06

Applicant: 重庆邮电大学

Inventor： 周丽芳 ,  王智峰 ,  李伟生 ,  肖明琪 ,  王婧琳

IPC: G06V20/17 ,  G06V10/24 ,  G06V10/44 ,  G06V10/762 ,  G06V10/764 ,  G06V10/82 ,  G06V10/80 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/08

Abstract: 本发明请求保护一种基于YOLOv5的无人机航拍图像目标检测方法，属于目标检测技术领域。所述方法包括以下步骤：步骤1.以YOLOv5算法为基础模型框架，为了提升城市上空航拍图像中小目标检测的精度，本发明设计了一种多种上下文特征提取的网络。步骤2.为了提升网络对于密集区域的关注度，本发明提出了一种串联交叉自注意力算法，分别加在主干网络和三个检测头之间，进一步的增强密集区域的信息。步骤3.通过迭代训练、更新参数得到最终的网络模型，再利用多尺度预测提升小目标检测性能，最终通过三个尺度的检测头预测得到最终结果。本发明有效的缓解了上下文信息丢失问题，并增强了特征提取能力，捕获了更加多样的特征空间，实现了更加清晰的锚框定位。