YANG Fei, LUO Jianqiao, LI Bailin. Latent Dirichlet allocation model based on local feature coding for railway fastener detection[J]. 2020, (2): 148-152.
YANG Fei, LUO Jianqiao, LI Bailin. Latent Dirichlet allocation model based on local feature coding for railway fastener detection[J]. 2020, (2): 148-152. DOI: 10.13873/j.1000-9787(2020)02-0148-05.
基于局部特征编码的LDA铁路扣件检测模型
摘要
针对潜在狄利克雷分布(LDA)模型中
视觉单词语义分辨能力弱
忽略特征之间的空间位置关系的问题
提出一种基于局部特征编码的LDA铁路扣件检测模型LC-LDA。首先
由K均值聚类图像底层特征得到视觉词典;然后
在特征编码过程中采用局部线性约束编码(LLC)
减小特征的重构误差;同时
结合图正则化引入特征上下文信息
保证空间相邻、外观相似的特征编码结果一致;最后
利用LDA主题模型提取图像的主题分布
并在支持向量机(SVM)中学习得到扣件类别。实验结果表明
LC-LDA中各类别的扣件图像区分性增强
漏检率和误检率明显降低。
Abstract
Aiming at the weakness of visual words and the ignorance of spatial positional relationship between features in latent Dirichlet allocation( LDA) model
a LDA fastener image detection model based on local feature coding is proposed
which is named LC-LDA. Firstly
visual dictionary is obtained by bottom layer features of images of K-means clustering. Secondly
locality-constrained linear coding( LLC) which can reduce the reconstruction error of features is used in the coding process. And then
the spatial information in graph regularized is used to ensure the features which are similar and adjacent had the same coding. Finally
the topic distributions of the images are extracted by LDA topic model
and the fastener category is learned in the support vector machine( SVM). The experimental results show that the discrimination of each category is enhanced in LC-LDA model
and the rate of missed detection and false detection is significantly reduced.