摘要
在基于深度学习的高光谱图像分类算法研究中,一些网络使用二维卷积,一些网络使用三维卷积,这两种算法都各有其优势和不足。针对基于深度学习的高光谱图像分类算法不足,提出一种融合注意力机制与混合CNN模型的算法,将高光谱图像中的空间和光谱的互补信息分别以三维卷积层和二维卷积层组合到一起。本文方法相比于经典SVM算法和目前几种性能优异的算法,在分类性能OA、AA以及Kappa系数上均有提升,证明了本算法通过结合二维卷积和三维卷积,并引入注意力机制的方法,对高光谱图像分类精度提升的有效性。
- 单位
在基于深度学习的高光谱图像分类算法研究中,一些网络使用二维卷积,一些网络使用三维卷积,这两种算法都各有其优势和不足。针对基于深度学习的高光谱图像分类算法不足,提出一种融合注意力机制与混合CNN模型的算法,将高光谱图像中的空间和光谱的互补信息分别以三维卷积层和二维卷积层组合到一起。本文方法相比于经典SVM算法和目前几种性能优异的算法,在分类性能OA、AA以及Kappa系数上均有提升,证明了本算法通过结合二维卷积和三维卷积,并引入注意力机制的方法,对高光谱图像分类精度提升的有效性。