Project Members:
- Anna Vera Linnea Fristedt Andersson
- Erik Agaton Sjöberg
- Theodor Tveit Husefest
Best submission 0.902 - #29400 on AICrowd.
Install required packages with either pip install -r requirements.txt
or conda install requirements.txt
.
Before running the program make sure to unzip the files in the data folder.
To run the program call python3 run.py
from root directory to create submission with best pre-trained model.
To retrain the model use: python3 run.py --train
.
The model used in the final submission is trained with 4 GPU's and need 4 GPU's to predict and create a new submission. This is an error in Keras, and can be fixed with retraining.