Ethiopic Digit Recognition with Resnet
Download the data from
kaggle.com
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.