科研之友
微信
新浪微博
Facebook
分享链接
摘要
基于双分支的胶囊网络分类方法在两个通道分别提取光谱信息和空间信息，既保留了双分支卷积神经网络的特征提取方式，又提高了分类精度。但由于高光谱图像（HSI）通常由几百个通道组成，在训练胶囊网络时，动态路由过程产生了大量的训练参数。为此提出1D和2D约束窗口分别减少来自两个提取通道的胶囊数量。它以胶囊向量组为计算单位进行卷积运算，来减少胶囊网络的参数量和计算复杂度。基于该降参优化方法提出一个新的双分支胶囊神经网络（DuB-ConvCapsNet-MRF）并将其应用在高光谱图像分类任务中。此外，为进一步提高分类性能，引入马尔可夫随机场（Markov Random Field， MRF）对空间区域进行平滑后处理，获得最终输出。对两个代表性高光谱图像数据集进行消融实验并与现有的6个分类方法进行比较，结果表明，DuB-ConvCapsNet-MRF在分类精度上都优于其他方法，并且有效降低了胶囊网络的训练代价。
- 出版日期2021-6-23
- 单位辽宁工程技术大学
