基于多尺度卷积神经网络的自适应熵加权决策融合船舶图像分类方法

• 作者: 任永梅 杨杰 郭志强 曹辉
• 来源期刊: 电子与信息学报
• 关键词: 图像处理 船舶图像分类 多尺度卷积神经网络 熵 决策融合
• 年，卷（期）: 2021,（5）
• 所属期刊栏目: 图像与智能信息处理|Image and Intelligent Information Processing
• 页码范围: 1424-1431
• 页数: 8页
• 分类号: TN911.73|TP391.4
• 语种: 中文
• DOI: 10.11999/JEIT200102

摘要

针对单一尺度卷积神经网络(CNN)对船舶图像分类的局限性,该文提出一种多尺度CNN自适应熵加权决策融合方法用于船舶图像分类.首先使用多尺度CNN提取不同尺寸的船舶图像的多尺度特征,并训练得到不同子网络的最优模型;接着利用测试集船舶图像在最优模型上测试,得到多尺度CNN的Softmax函数输出的概率值,并计算得到信息熵,进而实现对不同输入船舶图像赋予自适应的融合权重;最后对不同子网络的Softmax函数输出概率值进行自适应熵加权决策融合实现船舶图像的最终分类.在VAIS数据集和自建数据集上分别进行了实验,提出的方法的分类准确率分别达到了95.07%和97.50%,实验结果表明,与单一尺度CNN分类方法以及其他较新方法相比,所提方法具有更优的分类性能.