A Fine-Grained Image Classification Method Built on MobileViT

Abstract

With a rapid development of artificial intelligence technology, fine-grained image classification has gained widespread application. For mobile terminals, this paper introduces an image classification method built on MobileViT, and it can apply into fine-grained image classification. The original MobileViT model has been optimized in three ways. Initially, the h-swish activation function is used to enhance the network performance. Second, the cross-entropy loss function is used to further realize the parameter optimization and model accuracy improvement. Finally, a dropout layer is joined before the fully connected layer can effectively decrease the model recognition time and prevent over-fitting. Experimental data on public tomato disease datasets demonstrate that the improved fine-grained image classification method put forward in this paper exhibits higher classification accuracy, better stability and network generalization ability than other models.