论文中文题名: | 基于残差收缩网络与注意力机制的 遥感图像目标检测算法 |
姓名: | |
学号: | 19208207030 |
保密级别: | 公开 |
论文语种: | chi |
学科代码: | 085211 |
学科名称: | 工学 - 工程 - 计算机技术 |
学生类型: | 硕士 |
学位级别: | 工程硕士 |
学位年度: | 2022 |
培养单位: | 西安科技大学 |
院系: | |
专业: | |
研究方向: | 图像处理 |
第一导师姓名: | |
第一导师单位: | |
论文提交日期: | 2022-06-22 |
论文答辩日期: | 2022-06-07 |
论文外文题名: | Remote sensing image target detection algorithm based on residual shrinkage network and attention mechanism |
论文中文关键词: | |
论文外文关键词: | Remote sensing images ; Residual shrinkage network ; Attention mechanism ; Target detection ; Lightweight network |
论文中文摘要: |
遥感图像目标检测是对遥感图像分析应用的一项重要内容,可以应用于军事打击,城市规划和海情监测等领域。目前,基于深度学习的算法是自然图像目标检测的主流,但是由于遥感图像的特殊性,使得深度学习目标检测算法用于遥感图像时,精度下降,并且模型的参数量大,难以轻量化部署。根据这些问题,论文的主要工作如下: (1)针对遥感图像特殊性造成检测精度低的问题,提出基于残差收缩网络的遥感图像目标检测算法。该算法采用残差收缩网络作为特征提取网络,减少无用背景信息对检测效果的影响;除了常规的遥感图像增强方法比如:裁剪、旋转等,增加使用Mosaic 图像增强方法,增强对小目标的检测效果;设计结合最大值池化与均值池化的空间金字塔池化对特征进行充分融合,并结合通道注意力机制,筛选有效特征,增强算法模型对旋转目标和多尺度目标的检测效果;采用CIOU损失对目标候选区进行优化,使其定位更准确,提升对密集排列目标的检测效果。实验证明:改进的算法相比于原算法的总体 mAP 由 89.2%提升至 92.2%,获得了更好的性能表现。 (2)针对深度学习目标检测模型参数量大,模型复杂,难以轻量化部署在无人机等设备中使用这一问题,提出结合混合注意力机制的轻量化遥感图像目标检测算法。该算法构建浅层轻量化的网络模型,最大程度的减少参数量,提升检测速度。为了保持精度和速度的平衡,改进下采样模块,使用空洞卷积的特征融合模块,并结合混合注意力机制。同时,对预测目标边界框,采用Kmeans聚类进行预先的精确调整,在减少参数量的同时,降低精度的损耗。实验证明:改进算法的模型文件大小仅为3.5M,同时检测速度可以达到0.022s,mAP可以达到82.9%,在轻量化模型的同时依然可以保证精度和实时性。 |
论文外文摘要: |
Remote sensing image target detection is an important part of remote sensing image analysis and application. It can be applied to military strike, urban planning, sea monitoring and other fields. At present, the algorithm based on deep learning is the mainstream of natural image target detection,However, due to the particularity of remote sensing images, the accuracy of deep learning target detection algorithm used in remote sensing images decreases, and the parameters of the model are large, so it is difficult to deploy lightweight. According to the above problems, the main work of the paper is as follows: (1) To address the problem of poor detection accuracy caused by the specificity of remote sensing images,a target detection algorithm based on residual shrinkage network for remote sensing images is proposed.The algorithm adopts the residual shrinkage network as the feature extraction network to reduce the influence of useless background information on the detection effect; in addition to the conventional remote sensing image enhancement methods such as cropping and rotation, the Mosaic image enhancement method is added to enhance the detection effect of small targets; the spatial pyramid pooling combining the maximum pooling and mean pooling is designed to fully fuse the features and combine with the channel attention mechanism to filter the effective features and enhance the detection effect. mechanism to filter the effective features and enhance the detection effect of the algorithm model for rotating targets and multi-scale targets; the CIOU loss is used to optimize the target candidate area for more accurate localization and improve the detection effect for densely arranged targets. It is experimentally demonstrated that the overall mAP of the improved algorithm is improved from 89.2% to 92.2% compared with the original algorithm,and better performance is obtained. (2) To address the problem that deep learning target detection models have large number of parameters and complex models, which are difficult to be deployed in UAVs and other devices in a lightweight manner, we propose a lightweight remote sensing image target detection algorithm combining hybrid attention mechanism. The algorithm constructs a shallow lightweight network model to minimize the number of parameters and improve the detection speed. To maintain the balance of accuracy and speed, the downsampling module is improved and a feature fusion module with null convolution is used and combined with a hybrid attention mechanism. Meanwhile, for predicting the target bounding box, Kmeans clustering is used for pre-precise adjustment to reduce the loss of accuracy while reducing the number of parameters. Experiments prove that the model file size of the improved algorithm is only 3.5M, while the detection speed can reach 0.022s and mAP can reach 82.9%, which can still guarantee the accuracy and real-time performance while lightweighting the model. |
参考文献: |
[1]王彦情,马雷,田原. 光学遥感图像舰船目标检测与识别综述 [J].自动化学报, 2011,37(09):1029-1039. 聂光涛,黄华.光学遥感图像目标检测算法综述[J].自动化学报,2021,47(8): 1749−1768. [3] 张朝阳,程海峰,陈朝辉,等.高光谱遥感的发展及其对军事装备的威胁[J].光电技术应用,2008(01):10-12. [4] McKeown Jr D M, Denlinger J L. Cooperative methods for road tracking in aerial imagery[C]//the 1988 DARPA IUS Workshop. 1988: 327-341. [10] 李航.统计学习方法[M].北京:清华大学出版社,2012. [24] 李玉峰,顾曼璇,赵亮.采用改进 Faster R-CNN 的遥感图像目标检测方法[J]. 信号处理,2020,36(08):1363-1373. [25] 郑哲,雷琳,孙浩,匡纲要. FAGNet: 基于 MAFPN 和GVR的遥感图像多尺度目标检测算法[J].计算机辅助设计与图形学学报.2021,33(6):1-12. [26] 李红艳,李春庚,安居白,任俊丽.注意力机制改进卷积神经网络的遥感图像目标检测[J].中国图象图形学报,2019,24( 08) :1400-1408. [28] 吴湘宁,贺鹏,邓中港,李佳琪,王稳,陈苗.一种基于注 意力机制的小目标检测深度学习模型[J].计算机工程与科学,2021,43(1):95-104. [30] 董彪,熊风光,韩燮,况立群,徐清宇基于改进 Yolov3算法的遥感建筑物检测研究[J].计算机工程与应用,2020,56(18):209-213. [31] 成喆,吕京国,白颖奇,曹逸飞. 结合RPN网络与SSD算法的遥感影像目标检测算法[J]. 测绘科学,2021,46(04):75-82+99. |
中图分类号: | TP751 |
开放日期: | 2022-06-22 |