上海大学学报(自然科学版) ›› 2022, Vol. 28 ›› Issue (2): 314-323.doi: 10.12066/j.issn.1007-2861.2352

• 研究论文 • 上一篇    下一篇

面向遥感图像的小样本目标检测改进算法研究

李成范1,2(), 赵俊娟2   

  1. 1.东华理工大学 江西省数字国土重点实验室, 江西 南昌 330013
    2.上海大学 计算机工程与科学学院, 上海 200444
  • 收稿日期:2021-09-22 出版日期:2022-04-30 发布日期:2022-04-28
  • 通讯作者: 李成范 E-mail:lchf@shu.edu.cn
  • 作者简介:李成范(1981--), 男, 高级实验师, 博士,研究方向为智能信息处理. E-mail: lchf@shu.edu.cn
  • 基金资助:
    东华理工大学江西省数字国土重点实验室开放研究基金资助项目(DLLJ202103)

Improved approach to detect small sample target based on remote sensing image

LI Chengfan1,2(), ZHAO Junjuan2   

  1. 1. Key Laboratory for Digital Land and Resources of Jiangxi Province, East China University of Technology, Nanchang 330013, Jiangxi, China
    2. School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China
  • Received:2021-09-22 Online:2022-04-30 Published:2022-04-28
  • Contact: LI Chengfan E-mail:lchf@shu.edu.cn

摘要:

针对传统的遥感图像目标检测中面临的小样本以及目标样本分布不均衡等问题, 提出了一种基于改进的卷积神经网络(convolutional neural network, CNN)的遥感图像小样本目标检测算法. 首先, 该算法利用 $K$ 近邻($K$-nearest neighbor, kNN)回归分别对每个点和卷积层提取特征构建局部邻域; 同时, 通过最大池化聚合所有局部特征进行全局特征表示; 最后, 采用全连接层与缩放指数型线性单元(scaled expected linear unit, SELU)激活函数计算各类别对应的概率并分类. 实验结果表明, 该算法能够更有效地融合局部特征, 提高了遥感图像小样本目标识别与检测的精度, 同时保持信息的非局部扩散.

关键词: 卷积神经网络, 小样本, 深度学习

Abstract:

In this study, an improved convolutional neural network (CNN)approach is proposed to detect small sample targets from remotesensing images. The approach is designed to address the two issuesof small target samples and the unbalanced distribution of groundobject samples with respect to target detection of small samples byremote sensors. In the proposed method, first, $K$-nearest neighbor(kNN) regression is adopted to extract the features of each pointand convolution layer to construct the local neighborhood. Second,all local features are aggregated by maximum pooling layer in CNN torepresent global features. Subsequently, the full connection layerand scaled exponential linear unit (SELU) activation function areapplied to calculate the probability corresponding to each categoryfor classification. Finally, the proposed approach is tested andevaluated on hyperspectral imager remote sensing images datasets.Experimental results show that the proposed improvements to the CNNmodel fuse fully local features and result in the effectiverecognition and detection of small sample targets from remotesensing image with high accuracy while maintaining thenonlocal diffusion capabilities of information.

Key words: convolutional neural network (CNN), small sample, deep learning

中图分类号: