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

Self-adaptive Entropy Weighted Decision Fusion Method for Ship Image Classification Based on Multi-scale Convolutional Neural Network

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

Considering the limitation of single scale Convolutional Neural Network(CNN)for ship image classification,a self-adaptive entropy weighted decision fusion method for ship image classification based on multi-scale CNN is proposed.Firstly,the multi-scale CNN is used to extract the multi-scale features of ship image with different sizes,and the optimum models of different sub-networks are trained.Then,the ship images of test set are tested on the optimum models,and the probability value that is output by Softmax function of multi-scale CNN is obtained,which is used to calculate the information entropy so as to realize the adaptive weight assigned to different input ship images.Finally,self-adaptive entropy weighted decision fusion is carried out for the probability value that is output by Softmax function of different sub-networks to realize the final ship image classification.Experiments perform on VAIS(Visible And Infrared Spectrums)and self-built datasets respectively,and the proposed method achieves average accuracy of 95.07%and 97.50%on these datasets respectively.The experimental results show that the proposed method has better classification performance than those of the single scale CNN classification method and other state-of-the-art methods.