weitunglin/chinese_mnist

This project is to train a CNN model for Chinese MNIST

Chinese MNIST

This project is to training a CNN model for Chinese MNIST Dataset
made by Allen Lin weitung8@gmail.com

Usage

Install Dependencies

pip install -r requirements.txt

Train the model

Download dataset

• go to kaggle dataset
• download dataset
• unzip

Network Architecture

Accuracy

Testing accurarcy: 99.8%

Tensorboard

tensorboard --logdir runs

open Tensorboard to see complete result

Confusion Matrix

References

• https://www.kaggle.com/ricafernandes/chinese-mnist-pytorch-cnn-92-training
• https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html
• https://github.com/pytorch/examples/blob/master/mnist/main.py
• https://github.com/pytorch/examples/blob/master/imagenet/main.py