Research on segmentation of key parts for train safety based on deep learning

Qian Zhang; Xiaorong Gao; Hongna Zhu; Kai Yang; Zeyong Wang

doi:10.1117/12.2548911

20 December 2019 Research on segmentation of key parts for train safety based on deep learning

Qian Zhang, Xiaorong Gao, Hongna Zhu, Kai Yang, Zeyong Wang

Proceedings Volume 11209, Eleventh International Conference on Information Optics and Photonics (CIOP 2019); 112093N (2019) https://doi.org/10.1117/12.2548911
Event: Eleventh International Conference on Information Optics and Photonics (CIOP 2019), 2019, Xi'an, China

Abstract

Image segmentation is the most fundamental part of computer vision, which is the foundation of all other methods of image processing. The quality of image segmentation technology will affect the subsequent processing considerably. Comparing with traditional image segmentation algorithms, image segmentation algorithm based on deep learning is constantly proposed, with high performance and efficiency. But there is also a lot of room for improvement. For example, key parts such as fastening bolt are usually small in size, polluted and covered, and do not have enough characteristic information, so it is difficult to obtain satisfactory results. These factors affect the accuracy of the test, which is easy to cause serious accidents. As traditional methods sometimes cannot meet the requirement of high-accuracy result, deep learning play a particularly important role in facing those problems. To solve the problem that traditional object recognition methods are not robust enough to extract image features, parts recognition accuracy is low, and segmentation is not possible, we have made some modifications based on Mask R-CNN. In this method, convolutional neural network is used to extract features from part images. Then we use some annotated images from dataset to fine-tuned Mask R-CNN network to guarantee the accuracy. At the same time, data enhancement and k-folding cross-validation are carried out to improve the robustness of the model. Finally, the result of part recognition and segmentation by building the experimental platform proves the significance of the method.

Citation Download Citation

Qian Zhang, Xiaorong Gao, Hongna Zhu, Kai Yang, and Zeyong Wang "Research on segmentation of key parts for train safety based on deep learning", Proc. SPIE 11209, Eleventh International Conference on Information Optics and Photonics (CIOP 2019), 112093N (20 December 2019); https://doi.org/10.1117/12.2548911

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Image segmentation

Data modeling

Convolution

Feature extraction

Safety

Image processing algorithms and systems

Detection and tracking algorithms

Show All Keywords

Keywords/Phrases

Search In:

Publication Years