We proposed a CNN network for Mnist Persian classification. We used framework pytorch==1.8.2.
MNIST persian image dataset includes a training set of 960 samples and a test set of 240 samples. Each sample is resized to 70*70 pixels on a RGB associated with a tag of 10 classes. Please download dataset from here
To train the model please run the file train.ipy
Please download the weights from here
to test the trained model, please run the following file:
python inference.py --img_path data/2.jpeg --model_path model_mnist.pth --device GPU