Skip to content

shazraz/Semantic-Segmentation

Repository files navigation

Semantic Segmentation

Setup

Frameworks and Packages

Make sure you have the following is installed:

Dataset

Download the Kitti Road dataset from here. Extract the dataset in the data folder. This will create the folder data_road with all the training a test images.

Run

Launch your console and execute:

python main.py -d <DATA PATH> -e <EPOCHS> -b <BATCH SIZE>

The images in the runs folder were inferred using a model trained for 10 epochs on a batch size of 4. The remaining hyper-parameters are listed in the parameters text file in the runs folder.