Armored Vehicle Image Classification Based on Transfer Learning

Yi Liu,Jihuan Ren,Xiang Wu,Yuming Bo
DOI: https://doi.org/10.1109/iccsi53130.2021.9736219
2021-01-01
Abstract:Advances in science and technology are profoundly changing the battlefield environment. How to efficiently use visual inputs to quickly distinguish targets has become a hotspot of military science research. The Convolutional Neural Network (CNN) is currently one of the most popular methods in image classification tasks. However, CNNs' large demand for training samples is impractical for the classification of armored vehicles. In this context, this work applies transfer learning methods to alleviate the problem of difficult access to relevant training data. Specifically, the AlexNet (a typical CNN model) is pre-trained with the infantry fighting vehicle (IFV) samples, and transfer the learned knowledge to the tank classification to quickly reach better performance. The experimental results show that the performance of the transfer learning algorithm based on the AlexNet improved 12% compared to the same model trained from scratch. And the AlexNet also outperforms other traditional classification methods.
What problem does this paper attempt to address?