Semantic-segmentation

This project uses U-net a deep learning model and based on Apollo dataset and CamVid dataset for semantic segmentation to extract road markings

Installation

This project has the following dependencies:

-Numpy pip install numpy

-OpenCV Python apt-get install python-opencv

-TensorFlow pip install --upgrade tensorflow-gpu

Usage

The only thing you have to do to get started is set up the folders in the following structure:

├── "dataset_name"                   
|   ├── images├── test
|   |         ├── train
|   ├── labels├── test
|   |         ├── train

Results

ground truth	prediction
ground truth	prediction