应用科学学报 ›› 2022, Vol. 40 ›› Issue (1): 155-166.doi: 10.3969/j.issn.0255-8297.2022.01.014

• 计算机应用专辑 • 上一篇    下一篇

基于TSCD模型的轨道板裂缝检测方法

李文举1,2, 张耀星1, 陈慧玲2, 李培刚1, 沙利业3   

  1. 1. 上海应用技术大学 轨道交通学院, 上海 201418;
    2. 上海应用技术大学 计算机科学与信息工程学院, 上海 201418;
    3. 上海普利森配料系统有限公司, 上海 201108
  • 收稿日期:2021-07-14 出版日期:2022-01-28 发布日期:2022-01-28
  • 通信作者: 李培刚,博士,讲师,研究方向为高速、重载及交通轨道结构。E-mail:lipeigang@sit.edu.cn E-mail:lipeigang@sit.edu.cn
  • 基金资助:
    上海市科技创新行动计划基金(No.21210750300)资助

Track Slab Crack Detection Method Based on TSCD Model

LI Wenju1,2, ZHANG Yaoxing1, CHEN Huiling2, LI Peigang1, SHA Liye3   

  1. 1. School of Railway Transportation, Shanghai Institute of Technology, Shanghai 201418, China;
    2. School of Computer Science and Information Engineering, Shanghai Institute of Technology, Shanghai 201418, China;
    3. Shanghai Prinsen Dosing & Weighting System Co., Ltd, Shanghai 201108, China
  • Received:2021-07-14 Online:2022-01-28 Published:2022-01-28

摘要: 为解决轨道板裂缝检测问题,提出了一种基于分支级联卷积神经网络的轨道板裂缝检测模型--TSCD。首先该模型通过注意力机制和搜索分支结构突出轨道板裂缝的位置信息,同时抑制干扰信息;然后采用检测分支结构完成裂缝的像素级检测;最后对检测结果中出现的图像细节退化问题,利用参数映射关系实现特征图的上采样。实验结果表明:所提出的方法能够准确地检测出轨道板表面图像中的裂缝,其像素准确率可达97.56%,F1-score可达86.28%,并且在跨数据集测试中表现出较强的泛化性。

关键词: 裂缝检测, 分支级联, 参数映射, 轨道板, 卷积神经网络

Abstract: In order to solve the problem of track slab crack detection, a track slab crack detection model based on branch cascaded convolutional neural network, TSCD, is proposed. First, the model highlights the position information of track slab cracks through attention mechanism and structure of search branches to suppress interference information. Second, it realizes the pixel-level detection of cracks by structure of detecting branches. Finally, in order to solve the problem of image detail degradation in detection results, parameter mapping is used to achieve up-sampling of the feature maps. Experimental results show that the proposed model in this paper can not only detect the cracks in track plate surface images accurately with pixel accuracy rate of 97.56% and F1-score of 86.28%, but also performs strong generalization in cross-dataset tests.

Key words: crack detection, branch cascade, parameter mapping, track plate, convolutional neural network

中图分类号: