Author Login Editor-in-Chief Peer Review Editor Work Office Work

Computer Engineering

Previous Articles     Next Articles

Image Classification Algorithm for Deep Belief Network Based on Multi-feature Fusion

XU Qingyong a,b,JIANG Shunliang b,HUANG Wei b,LI Jing b,XU Shaoping b,YE Famao b   

  1. (a.School of Economics and Management; b.School of Information Engineering,Nanchang University,Nanchang 330031,China)
  • Received:2014-10-08 Online:2015-11-15 Published:2015-11-13

基于多特征融合的深度置信网络图像分类算法

许庆勇 a,b,江顺亮 b,黄伟 b,李菁 b,徐少平 b,叶发茂 b   

  1. (南昌大学 a.经济管理学院,南昌 330031; b.信息工程学院,南昌 330031)
  • 作者简介:许庆勇(1982-),男,讲师、博士研究生,主研方向:图像处理,机器学习,机器视觉;江顺亮,教授、博士后、博士生导师;黄伟,副教授、博士;李菁,博士;徐少平、叶发茂,副教授、博士。
  • 基金资助:
    国家自然科学基金资助项目(61463032,61363046,41261091)。

Abstract: Taking the single feature and the major classification algorithms into consideration,an image classification algorithm based on fusion of multi-feature for Deep Belief Network(DBN) is proposed to classify the nature images.The features about color,texture,shape are extracted and the characteristic weight matrix is formed.Then the characteristic matrix is normalized.The samples are trained and classified using the DBN with four levels which is constructed.The proposed method has been evaluated on the Corel dataset by train weight,and the result shows that the average classification accuracy is 85.1% by the proposed algorithm,which is higher then single feature algorithm and other mainstream algorithms.

Key words: Deep Belief Network(DBN), image classification, feature extraction, multi-feature fusion, image retrival

摘要: 针对现有单一特征描述符及浅层结构分类算法分类正确率较低的问题,基于底层图像特征提出一种针对自然界图像特点的深度置信网络(DBN)图像分类算法。提取样本图像中的颜色、纹理和形状特征,构成多特征融合的权重矩阵,并对特征矩阵进行归一化处理,利用构建的4层DBN分类器进行训练和分类。采用Corel图库,通过训练权重进行测试,结果表明,该算法的平均分类正确率达到85.1%,高于使用单一特征的分类算法和其他主流分类算法。

关键词: 深度置信网络, 图像分类, 特征提取, 多特征融合, 图像检索

CLC Number: