发明公开
- 专利标题: 基于空频特征融合检测头的无人机目标检测方法
-
申请号: CN202410698735.X申请日: 2024-05-31
-
公开(公告)号: CN118691929A公开(公告)日: 2024-09-24
- 发明人: 石争浩 , 樊嘉旺 , 尤珍臻 , 冯亚宁 , 贾萌 , 杨秀红 , 都双丽 , 石程 , 赵明华 , 梁湘 , 马品三 , 杨鸣丽
- 申请人: 西安理工大学
- 申请人地址: 陕西省西安市碑林区金花南路5号
- 专利权人: 西安理工大学
- 当前专利权人: 西安理工大学
- 当前专利权人地址: 陕西省西安市碑林区金花南路5号
- 代理机构: 西安弘理专利事务所
- 代理商 罗笛
- 主分类号: G06V10/80
- IPC分类号: G06V10/80 ; G06V10/774 ; G06V10/764 ; G06V10/82 ; G06V10/52 ; G06V20/17 ; G06N3/0464
摘要:
本发明公开了基于空频特征融合检测头的无人机目标检测方法,具体包括如下步骤方法:步骤1,输入公开数据集,并划分为训练集和测试集;步骤2,对训练集和测试集中的图片进行预处理,得到图片X;步骤3,构建无人机场景下基于空频特征融合检测头的YOLO目标检测网络模型;步骤4,对步骤3构建的无人机场景下空频特征融合检测头的YOLO目标检测网络模型进行训练;步骤5,将步骤2预处理后的测试集图片放入步骤4训练好的模型,最终得到带有检测框的无人机图像。本发明通过级联注意力增加全局信息提高了在背景复杂、场景拥挤下的检测能力,通过空频特征融合检测头解决尺度变化大的检测难题并提高了整体的检测性能。