Application Research of Image Classification Based on Pytorch and Convolutional Neural Network
Zhaoyu Li · Computers and artificial intelligence. · 2024
The image classification problem of convolutional neural network (CNN) on CIFAR-10 dataset is studied, and the model is implemented and experimented with PyTorch. Through the training and testing of the model, we analyze the performance of the model and explore the possibility of improvement. The experimental results show that the model achieves a certain accuracy in the image classification task, but there is space for improvement in some categories.