Research on Image Classification Algorithm Based on Pytorch

Abstract

Abstract With the development of artificial intelligence and deep learning, image classification technology has ushered in new opportunities and challenges. The so-called image classification problem is the problem that the user passes in the image, and then the computer sends out the classification and description of the content of the image. Feature description and detection as a traditional image classification method have many drawbacks, such as low accuracy and long time-consuming problems. Among the existing deep learning frameworks, Pytorch is relatively effective. Researchers can use the Pytorch framework to construct convolutional neural network models quickly and easily, and train them on massive data sets. This paper briefly introduces the image classification algorithm, neural network and Pytorch. The classification experiment is carried out on the Fashion MNIST data set, and the accuracy of the model reaches 88.39%. The future research direction is discussed.