Car make and model recognition using convolutional neural network: fine-tune AlexNet architecture

Md. Asifur Rahman Khan, Raju Talukder, Md. Anwar Hossen, Nusrat Jahan

Abstract


Artificial intelligence (AI) has significantly contributed to car make and model recognition in this current era of intelligent technology. By using AI, it is much easier to identify car models from any picture or video. This paper introduces a new model by fine-tuning the AlexNet architecture to determine the car model from images. First of all, our car image dataset has been created. Some of these images were taken by us, and others were taken from the website of the car connection. Then we cleaned all the unwanted images for better performance. Our dataset has ten classes containing 5,000 car images split into train and test data. After that, we augmented our data with random flip, rotation, and zoom to reduce overfitting. Finally, we used a pre-trained convolutional neural network (CNN) model AlexNet architecture. We fine-tuned AlexNet (FT-AlexNet) by adding three extra layers for better classification and compared it with the original AlexNet. To measure the performance of these models, accuracy, precision, recall, and F1-score were used. The results show that fine-tune AlexNet architecture outperforms the original AlexNet architecture. The results prove that recognition accuracy has increased due to our improvement approach.

Keywords


AlexNet; Car model recognition; Convolutional neural network; Deep learning; Fine tuned-AlexNet

Full Text:

PDF


DOI: http://doi.org/10.11591/ijeecs.v33.i1.pp370-379

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).

shopify stats IJEECS visitor statistics