Kuzushiji-MNIST Classification

This repository impletments the classification for Kuzushiji-MNIST in Pytorch with model ResNet & ResMLP.

Download the dataset

wget http://codh.rois.ac.jp/kmnist/dataset/kmnist/kmnist-train-imgs.npz
wget http://codh.rois.ac.jp/kmnist/dataset/kmnist/kmnist-train-labels.npz
wget http://codh.rois.ac.jp/kmnist/dataset/kmnist/kmnist-test-imgs.npz
wget http://codh.rois.ac.jp/kmnist/dataset/kmnist/kmnist-test-labels.npz

MLP/CNN Model

You will need GPU to run the following code.

Available model names:

ResMLP-12, ResMLP-24, ResNet-18, ResNet-34

To train the model:

python classification.py --model [model name] --gpu [GPU No.] --train 1

To test the model:

python classification.py --model [model name] --gpu [GPU No.]

Results are stored in ./log