Ethiopic Digit Recognition with Resnet

Download the data from https://www.kaggle.com/c/tau-ethiopic-digit-recognition/data

This was part of a competition. Hence labels are available only for the training data. We are supposed to upload the predictied digits to Kaggle as a csv file.

Requirements:

pytorch
opencv

Steps to train the model

1. Download and extract the data into the project folder.
2. Run split_train_val.py to create the validation data.
3. Run train.py to start training the model.
4. Once the model is trained,in run_inference.py provide the path of the trained model and run it.