-
题名基于注意力机制的航拍图像目标检测算法
被引量：3
- 1
-
-
作者
白宗宝
张俊举
高原
胡友成
-
机构
南京理工大学电子工程与光电技术学院
南京理工大学紫金学院电子工程与光电技术学院
-
出处
《激光与光电子学进展》
CSCD
北大核心
2023年第12期312-322,共11页
-
基金
国家自然科学基金(61971386)。
-
文摘
针对现有基于水平视角图像的目标检测网络在无人机航拍图像上误检率和漏检率高的问题,提出一种基于改进注意力机制的航拍图像目标检测算法。首先,在Faster R-CNN主干网络输出端引入一种三叉戟注意力机制,分别提取三路池化层和三路扩张卷积层的多模式信息及多尺度特征信息进行压缩,实现特征通道和空间像素区域的权重再分配。其次,针对航拍图像目标的分类和边界框回归,引入一种双头检测机制,充分利用目标的语义信息和空间位置信息。在相关数据集上对所提算法进行评测,并与其他目标检测算法进行对比。结果表明所提算法的平均精度均值得到显著提升,在不同场景下的无人机航拍图像目标检测中获得更好的效果。
-
关键词
机器视觉
无人机
目标检测
注意力机制
双头检测机制
-
Keywords
machine vision
unmanned aerial vehicle
object detection
attention mechanism
double-head mechanism
-
分类号
TP391.4
[自动化与计算机技术—计算机应用技术]
-
