首页 | 本学科首页   官方微博 | 高级检索  
     

基于多尺度卷积神经网络的自适应熵加权决策融合船舶图像分类方法
引用本文:任永梅,杨杰,郭志强,曹辉.基于多尺度卷积神经网络的自适应熵加权决策融合船舶图像分类方法[J].电子与信息学报,2021,43(5):1424-1431.
作者姓名:任永梅  杨杰  郭志强  曹辉
作者单位:1.武汉理工大学信息工程学院宽带无线通信与传感器网络湖北省重点实验室 武汉 4300702.湖南工学院电气与信息工程学院 衡阳 421002
基金项目:国家自然科学基金 (51879211),国家重点研发计划(2020YFB1710800),湖南省教育厅科学研究项目(18C0900)
摘    要:针对单一尺度卷积神经网络(CNN)对船舶图像分类的局限性,该文提出一种多尺度CNN自适应熵加权决策融合方法用于船舶图像分类。首先使用多尺度CNN提取不同尺寸的船舶图像的多尺度特征,并训练得到不同子网络的最优模型;接着利用测试集船舶图像在最优模型上测试,得到多尺度CNN的Softmax函数输出的概率值,并计算得到信息熵,进而实现对不同输入船舶图像赋予自适应的融合权重;最后对不同子网络的Softmax函数输出概率值进行自适应熵加权决策融合实现船舶图像的最终分类。在VAIS数据集和自建数据集上分别进行了实验,提出的方法的分类准确率分别达到了95.07%和97.50%,实验结果表明,与单一尺度CNN分类方法以及其他较新方法相比,所提方法具有更优的分类性能。

关 键 词:图像处理    船舶图像分类    多尺度卷积神经网络        决策融合
收稿时间:2020-02-11

Self-adaptive Entropy Weighted Decision Fusion Method for Ship Image Classification Based on Multi-scale Convolutional Neural Network
Yongmei REN,Jie YANG,Zhiqiang GUO,Hui CAO.Self-adaptive Entropy Weighted Decision Fusion Method for Ship Image Classification Based on Multi-scale Convolutional Neural Network[J].Journal of Electronics & Information Technology,2021,43(5):1424-1431.
Authors:Yongmei REN  Jie YANG  Zhiqiang GUO  Hui CAO
Affiliation:1.Hubei Key Laboratory of Broadband Wireless Communication and Sensor Networks, School of Information Engineering, Wuhan University of Technology, Wuhan 430070, China2.School of Electrical and Information Engineering, Hunan Institute of Technology, Hengyang 421002, China
Abstract: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.
Keywords:
点击此处可从《电子与信息学报》浏览原始摘要信息
点击此处可从《电子与信息学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号