Machine learning classifiers based on HoG features extracted from locomotive neutral section images

Abstract

This paper presents a comparative study on machine learning algorithms for neutral section image classification. The classifiers are trained by employing the Histogram of Oriented Gradient features that are extracted from the neutral section dataset [1]. A neutral section is a phase break that is used on the Transnet freight rail system to separate the single-phase supply from the 25kV three-phase overhead traction supply. The 25kV is a stepped-down voltage from an 88kV three-phase supply coming from the national grid. While the main purpose of the neutral section is to separate phase voltages, electric locomotives can traverse through these phases by switching On and Off. This auto-switching is possible through induction magnets installed in between the rails and with magnet detection sensors installed underneath the locomotives. However, a computer vision model has been developed, trained, and tested with a neutral section dataset containing images having open and close markers [1]. This paper, therefore, utilises this dataset to provide performance comparison on several machine learning classification algorithms viz. Decision Tree, Discriminant Analysis, Support Vector Machine, K-Nearest Neighbors, Ensemble, Naïve Bayes, and Convolutional Neural Network. A confusion matrix, F1- measure and computation time are employed to measure the performance of each classifier. The MATLAB Classification Learner application was used to obtain the results. The results show that the Linear Support Vector Machine performs best when considering performance and prediction speed. The Linear Support Vector Machine achieved a training accuracy of 93.40% with a test accuracy reaching 94% at a prediction speed of 75 objects per second (computation time).