返回
Automated rail surface crack analytics using deep data-driven models and transfer learning
DOI:10.1016/j.scs.2021.102898.png)
摘要
En 中文
A deep transfer learning (DTL) framework for the rail surface crack detection using a limited volume of training images is developed in this study. In DTL, two deep learning models pre-trained by two recently popular object detection algorithms, YOLOv3 and RetinaNet, based on the COCO dataset are considered as base models for next transfer learning via using rail track images. Transferred YOLOv3 and RetinaNet models are separately applied to detect rail track cracks. Based on their outputs, an ensemble scheme is developed to further improve the detection performance. The effectiveness of the model developed by the DTL is validated by benchmarking against models developed by YOLOv3, RetinaNet, faster R-CNN (F-RCNN) and single shot multibox detector (SSD) separately as well as two conventional rail surface crack detection methods, the visual detection system (VDS) and the geometrical approach (GEA). A dataset composed of 35 rail images from the China Railway Corporation and 67 rail images from the publicly available Type-I RSDDs dataset is considered. Computational results show that the model developed by DTL attains the best recall and average precision comparing with benchmarking models. VDS and GEA almost fail in the crack detection task due to the complex background and various noises in images. In addition, in terms of the average precision, YOLOv3 model is more suitable for detecting cracks of small sizes while RetinaNet model performs better on detecting cracks of large sizes.
Keyword:
Crack detection
Rail tracks
Data analytics
Deep learning applications
Convolutional neural networks
AI总结
对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。
期刊
IF:
12
论文数:
7.9K
被引数:
5.2W
机构
引用论文
Improving rail network velocity: A machine learning approach to predictive maintenance提高铁路网络速度: 预测性维护的机器学习方法
Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning用于计算机辅助检测的深度卷积神经网络: CNN架构,数据集特征和迁移学习
Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?用于医学图像分析的卷积神经网络: 完全训练还是微调?

