Skip to content

StanlyHardy/KITTI-Road-Segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Kitti- Road Segmentation

Lane Segmentation using several architectures.

forthebadge made-with-python

PyPI pyversions PRs Welcome Open Source Love svg1

It contains the code for both training and segmentation of lane lines using Deep Learning. Currently the supported architectures are ENET, UNET, Modified VGG.

Features

  • The training code is very much scalable towards any new architecture.
  • All changes made in the config file will effect in the training process so that the training logic can be without hassle.
  • The training configuartion are easily tunable through the config file provided.

Requirements

  • The training module has been built using Pycharm 2018.1.4.
  • The System requirement’s are 2.7 GHz Intel Core i5 with atleast 8 GB of RAM.

Installation

OpenCV

You can use Anaconda to install opencv with the following command line.:

conda install -c conda-forge opencv

Image Augmentation

You can use PIP to install the module imgaug with the following command line.:

pip install imgaug

Tensorflow

You can use PIP to install tensorflow with the following command line or please go through their official installation guideline

pip install tensorflow

Keras

You can use PIP to install keras with the following command line or please go through their official installation guideline

pip install keras

Usage example

Run the following script to dispatch the trainer.

python3 train.py  --conf=./config.json

Contribute

Don't feel shy to drop a star, if you find this repo useful.I would love for you to contribute to KITT-Road Segmentation, check the LICENSE file for more info.

Meta

Stanly Moses – @Linkedinstanlimoses@gmail.com

Distributed under the MIT license. See LICENSE for more information.

https://github.com/StanlyHardy/KITTI-Road-Segmentation