Computer Engineering and Applications ›› 2017, Vol. 53 ›› Issue (21): 185-189.DOI: 10.3778/j.issn.1002-8331.1703-0454

Previous Articles     Next Articles

Inspection for railway fasteners based on entropy-weighted BOW model

LI Shuang, LI Bailin, DI Shilei, LUO Jianqiao   

  1. School of Mechanical Engineering, Southwest Jiaotong University, Chengdu 610031, China
  • Online:2017-11-01 Published:2017-11-15

基于信息熵加权词包模型的扣件图像检测

李  爽,李柏林,狄仕磊,罗建桥   

  1. 西南交通大学 机械工程学院,成都 610031

Abstract: The traditional Bag Of Words(BOW) model ignores the structure of the image while conducting the inspection of railway fasteners. To overcome this defect, this paper proposes an Entropy-Weighted BOW(EW_BOW) model for identification of the image of fasteners. On the basis of the traditional BOW model, the entropy is introduced to weight word frequency of BOW model?in the sub-image of fasteners, thus making the BOW model more distinguishable for different categories of fasteners. And then the Latent Dirichlet Allocation(LDA) is used to learn the topic distribution of the images. Finally, the Support Vector Machine(SVM) is applied to classify a new image. The experiment on four types of fasteners shows that the EW_BOW model can inspect fastener states more precisely.

Key words: railway fastener inspection, Bag Of Words(BOW) model, visual word, entropy, Latent Dirichlet Allocation(LDA) model

摘要: 针对传统“视觉词包模型”在进行铁路扣件检测时忽略图像结构而导致的区分能力不强的问题,提出一种基于信息熵加权词包模型的扣件检测模型EW_BOW。在传统“视觉词包模型”的基础上,引入信息熵对扣件图像局部区域的词包模型的词频进行加权处理,加强词包模型对不同类别扣件的区分性,并利用潜在狄利克雷分布学习扣件图像的主题分布。最后,采用支持向量机对扣件进行分类识别。对四类扣件图像的分类实验证明该模型能够有效提高扣件分类精确度。

关键词: 铁路扣件检测, 词包模型, 视觉单词, 信息熵, 潜在狄利克雷分布模型