theodorhusefest / CS433-project2-road-segmentation

CS433 Machine Learning at EPFL, Road Segmentation task

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Road Segmentation Project - EPFL Autumn 2019

Project Members:

  • Anna Vera Linnea Fristedt Andersson
  • Erik Agaton Sjöberg
  • Theodor Tveit Husefest

Best submission 0.902 - #29400 on AICrowd.

How to run

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.

About

CS433 Machine Learning at EPFL, Road Segmentation task


Languages

Language:Python 100.0%