Application Research of Image Classification Based on Pytorch and Convolutional Neural Network
Year: 2024
Type: article
Abstract: 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.
DOI: 10.70267/eta2tt72
Author: Zhaoyu Li
Institution: Guangdong University of Science and Technology
Language: English
Subfield: Computer Vision and Pattern Recognition
Field: Computer Science
Domain: Physical Sciences
Open Access status: hybrid
OpenAlex ID: W4402530753