Open AccessTransfer Learning for classifying front and rear views of vehicles
Author(s) -
Sara Baghdadi,
Noureddine Aboutabit
Publication year - 2021
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/1743/1/012007
Subject(s) - transfer of learning , computer science , artificial intelligence , support vector machine , scratch , classifier (uml) , similarity (geometry) , task (project management) , transfer (computing) , pattern recognition (psychology) , machine learning , computer vision , image (mathematics) , engineering , parallel computing , systems engineering , operating system
Various computer systems have been proposed to classify vehicles according to several criteria (category, brand, model). Unfortunately, there is not much research on the classification of views, especially front and rear views. Several factors make this classification very difficult including similarity in shape, size, and color. This work aims to classify front and rear views of vehicles using the Transfer Learning (TL) approach. Here, we used a pre-trained CNN (AlexNet) that has been trained on more than a million images and can classify images into 1000 object categories. Thus, we transferred its learned knowledge and applied it to our new task (Classifying vehicle views). We conducted then two experiments. The first experiment has two scenarios: the first scenario is devoted to Transfer Learning using the AlexNet model, and the second scenario aims to build a network from scratch inspired from AlexNet. Experimental results reveal that the Transfer Learning approach gives high results. On the other hand, in the second experiment, we decided to use TL-AlexNet to extract features and train them with an SVM classifier instead of fully connected layers. And also, we combined the SVM with the fully connected layers. The accuracy rates have been improved after this experiment.