The Image Classification Algorithm Was Implemented on The MNIST Data Set
DOI:
https://doi.org/10.61173/6xfeq893Keywords:
Convolutional neural networks, Convolution kernel, Data set, Pooling layerAbstract
In the current era of rapid development of science and technology, the recognition and classification of digital images is the key to solving many problems, such as the application of license plate recognition, document digitization, and remote sensing image surface classification. Based on the MNIST handwritten numerical data set collected by the National Institute of Standards and Technology (NIST), this report uses Python language and PyTorch programming framework to construct a convolutional neural network (CNN) structure and practice and experience the image classification of handwritten digits in the MNIST data set.