计算机与数字工程2024，Vol.52Issue(3)：880-885,6.DOI:10.3969/j.issn.1672-9722.2024.03.041
基于改进Mask RCNN的遥感图像小目标检测算法研究
Research on Remote Sensing Image Detection Algorithm Based on Improved Mask RCNN
张艺博 1赵加坤 1陈攀 1支杨丹 1夏星浩1
作者信息
- 1. 西安交通大学软件学院 西安 710049
- 折叠
摘要
Abstract
With the continuous development of the field of aerial remote sensing,the detection of small targets in this scenario has become an important work in the current research field.Based on aerial remote sensing images,this paper proposes an optimiza-tion method for small target detection in the field of aerial remote sensing.In order to improve the practicability and accuracy of the algorithm in small target detection,this paper adds a spatial attention mechanism module to the Mask RCNN algorithm to reduce the noise of the image background,and uses CIOU as the bounding box regression loss function to optimize.And then it uses the Kmeans clustering algorithm instead of the original algorithm to generate a detection anchor box that more matches the small target.The improved Mask RCNN has a detection accuracy of 61.89mAP under the aerial remote sensing image data set.Compared with the current mainstream remote sensing image detection algorithm R-FCN,the detection accuracy has increased by 17%,and compared with Mask RCNN by 2.4%,reaching the best detection effect under the current conditions.关键词
改进Mask RCNN/航空遥感图像/注意力机制/CIOUKey words
improved Mask RCNN/aerial remote sensing image/attention mechanism/CIOU分类
信息技术与安全科学引用本文复制引用
张艺博,赵加坤,陈攀,支杨丹,夏星浩..基于改进Mask RCNN的遥感图像小目标检测算法研究[J].计算机与数字工程,2024,52(3):880-885,6.
