首页 | 官方网站   微博 | 高级检索  
     

基于双注意力机制的遥感图像目标检测
引用本文:周幸,陈立福.基于双注意力机制的遥感图像目标检测[J].计算机与现代化,2020,0(8):1-7.
作者姓名:周幸  陈立福
作者单位:长沙理工大学电气与信息工程学院,湖南 长沙 410114;长沙理工大学电气与信息工程学院,湖南 长沙 410114
摘    要:针对遥感图像在复杂背景下小目标检测精度较低的问题,提出一种基于双注意力机制模型的SSD检测算法。该算法在前端特征提取网络中引入双注意力机制模型,强化低层特征图中小目标的有效特征信息并抑制冗余的语义信息,实现自适应特征学习;并在空间注意力模型中引入空洞卷积,保证卷积核感受野的同时减少了网络参数。引入Focal loss损失函数作为改进算法的分类损失函数,改善网络在训练过程中样本失衡的问题,增加正样本与难样本在训练时的权重比例,提升算法的检测性能。对遥感图像数据集NWPU VHR-10进行检测的结果表明,本文的改进算法在保证检测速度的同时提高了检测精度。与传统SSD算法相比,改进SSD算法的mAP提高了2.25个百分点,达到79.65%。

关 键 词:深度学习    目标检测    特征提取    双注意力机制模型    空洞卷积    Focal  loss损失函数  
收稿时间:2020-08-17

Object Detection of Remote Sensing Image Based on Dual Attention Mechanism
Abstract:Aiming at the problem of low accuracy of small target detection in remote sensing image under complex background, an new SSD detection algorithm based on dual attention mechanism model is proposed. The algorithm introduces the dual attention mechanism model in the front-end feature extraction network. It strengthens the effective feature information of small targets in the low-level feature map and suppresses the redundant semantic information to achieve adaptive feature learning. In addition, dilated convolution is introduced into the spatial attention model to ensure the sensitivity of convolution kernel and reduce the parameters of the network. The Focal loss function is introduced as the classification loss function of the improved algorithm to improve the imbalance of samples and increase the weight ratio of positive and difficult samples during training, it promotes the detection performance of the algorithms. The detection results of the remote sensing image data set NWPU VHR-10 show that the improved algorithm not only ensures the detection speed, but also improves the detection accuracy. Compared with the traditional SSD algorithm, the mAP of the improved SSD algorithm is increased by 2.25 percentage points to 79.65%.
Keywords:deep learning  target detection  feature extraction  double attention mechanism model  dilated convolution  Focal loss function  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机与现代化》浏览原始摘要信息
点击此处可从《计算机与现代化》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号