A fast template matching-based algorithm for railway bolts detection
Tóm tắt
Railway bolts detection is an important task in railway maintenance and some techniques based on traditional feature extraction and classification have been used in this application. However, these techniques have two critical disadvantages, i.e., manual collection of training data set and time-consuming training process; furthermore, trained classifiers are hard to generalize from a specific railway to the others. In order to overcome these problems, we propose a fast template matching-based algorithm, named FTM, in this paper. Firstly, we use a template matching method to locate the bolts with constrains of the railway geometric structure. Then, we use a nearest neighbor classifier to determine whether a bolt is in position or not. At last, we use GPU with CUDA architecture to accelerate the most time-consuming part of FTM. The experiments demonstrate that our proposed FTM algorithm achieves the accuracy of 98.57 % in average, and the average false positive is only 0.89 %. The overall speedup of FTM by GPU is 6.11, and the most time-consuming part gets speedup of 17.73. Furthermore, FTM only need to collect several samples in a new railway without laborious training work.
Tài liệu tham khảo
Railway Ministry of China, “Twelfth Five Year” Plan of National Railway
Cybernetix Group (France) “IVOIRE: a system for rail inspection,” internal documentation. http://www.cybernetix.fr
Benntec Systemtechnik Gmbh, “RAILCHECK: image processing for rail analysis” internal documentation. http://www.benntec.com
Lin J, Luo SW, Li QY, Zhang HQ, Ren SW (2009) Real-time rail head surface defect detection: a geometrical approach. In: Proceedings of IEEE international symposium on industrial electronics, pp 769–774
Stella E, Mazzeo PL, Nitti M, Cicirelli G, Distante A, D’Orazio T (2002) Visual recognition of missing fastening elements for railroad maintenance. In: Proceedings of IEEE-ITSC international conference on intelligent transportation system, pp 94–99
Mazzeo PL, Nitti M, Stella E, Distante A (2004) Visual recognition of fastening bolts for railroad maintenance. Pattern Recogn Lett 25(6):669–677
Marino F, Distante A, Mazzeo PL, Stella E (2007) A real time visual inspection system for railway maintenance: automatic hexagonal headed bolts detection. IEEE Trans Syst Man Cybern Part C Appl 37(3):418–428
Yella S, Dougherty M, Gupta NK (2009) Condition monitoring of wooden railway sleepers. Transp Res Part C Emerg Technol 17(1):38–55
Duda RO, Hart PE (1972) Use of the Hough transformation to detect lines and curves in pictures. Commun ACM 15:11–15
von Gioi RG, Jakubowicz J, Morel JM, Randall G (2010) LSD: a fast line segment detector with a false detection control. IEEE Trans Pattern Anal Mach Intell 32(4):722–732
Ojala T, Pietikäinen M, Harwood D (1996) A comparative study of texture measures with classification based on feature distribution. Pattern Recogn 29(1):51–59
Pietikäinen M, Ojala T, Nisula J, Heikkinen J (1994) Experiments with two industrial problems using texture classification based on feature distributions. Proc SPIE 2354:197–204
Silvén O, Niskanen M, Kauppinen H (2003) Wood inspection with non-supervised clustering. Mach Vis Appl 13(5–6):275–285
Bulthoff H, Wallraven C, Graf A (2002) View-based dynamic object recognition based on human perception. In: Proceedings of international conference on pattern recognition, pp 768–776
Turtinen M, Pietikäinen M (2003) Visual training and classification of textured scene images. In: Proceedings of international workshop on texture analysis and synthesis, pp 101–106
Pietikäinen M, Nurmela T, Mäenpää T, Turtinen M (2004) View-based recognition of real-world textures. Pattern Recogn 37(2):313–323
Ahonen T, Hadid A, Pietikäinen M (2004) Face recognition with local binary patterns. In: Proceedings of European conference on computer vision, pp 469–481
Guo Z, Zhang L, Zhang D (2010) A completed modeling of local binary pattern operator for texture classification. IEEE Trans Imag Process 19(6):1657–1663
Levi K, Weiss Y (2004) Learning object detection from a small number of examples: the importance of good features. In: Proceedings of IEEE conference on computer vision and pattern recognition, pp II-53–II-60
Dalal N, Triggs B (2005) Histograms of oriented gradients for human detection. In: Proceedings of IEEE conference on computer vision and pattern recognition, pp 886–893
Lowe D (2004) Distinctive image features from scale-invariant keypoints. Int J Comput Vis 60(2):91–110
Belongie S, Malik J, Puzicha J (2002) Shape matching and object recognition using shape contexts. IEEE Trans Pattern Anal Mach Intell 24(4):509–522
Mikolajczyk K, Schmid C, Zisserman A (2004) Human detection based on a probabilistic assembly of robust part detectors. In: Proceedings of European conference on computer vision, pp 69–82
Lowe DG (1999) Object recognition from local scale-invariant features. In: International conference on computer vision, Corfu, pp 1150–1157
Comaniciu D, Ramesh V, Meer P (2003) Kernel-based object tracking. IEEE Trans Pattern Anal Mach Intell 25(5):564–575
Wei YC, Tao LT (2010) Efficient histogram-based sliding window. In: Proceedings of IEEE conference on computer vision and pattern recognition, pp 3003–3010
Nvidia, Developer Zone, CUDA Zone. https://developer.nvidia.com/category/zone/cuda-zone
Nvidia, CUDA Toolkit Documentation. http://docs.nvidia.com/cuda/index.html
Powers DMW (2007) Evaluation: from precision, recall and F-factor to ROC, informedness, markedness and correlation. J Mach Learn Technol 2(1):37–63
Baldi P, Brunak S, Chauvin Y, Andersen CAF, Nielsen H (2000) Assessing the accuracy of prediction algorithms for classification: an overview. Bioinformatics 16:412–424
Barakat M, Lefebvre D, Khalil M, Druaux F, Mustapha O (2013) Parameter selection algorithm with self-adaptive growing neural network classifier for diagnosis issues. Int J Mach Learn Cybernet 4(3):217–233
Zheng H, Wang H (2012) Improving pattern discovery and visualization with self-adaptive neural networks through data transformations. Int J Mach Learn Cybernet 3(3):173–182
Wang XZ, Dong CR, Fan TG (2007) Training T-S norm neural networks to refine weights for fuzzy if-then rules. Neurocomputing 70(13–15):2581–2587
Tsang E, Wang XZ, Yeung D (2000) Improving learning accuracy of fuzzy decision trees by hybrid neural networks. IEEE Trans Fuzzy Syst 8(5):601–614