Semantic Segmentation

Implementation of several Semantic Segmentation architectures for common segmentation tasks.

This respository consists of the code for Semantic segmantation using several Deep Learning Architectures.

Features

• If space complexity and time complexity is taken into criteria, ENET is a viable option. It can be deployed even in mobile devices.
• If the time and space complexity can be relaxed, UNET and Modified VGG net performs better than their other variant ENET.
• 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 Semantic Segmentation, check the LICENSE file for more info.

Meta

Stanly Moses – @Linkedin – stanlimoses@gmail.com

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

https://github.com/StanlyHardy/Semantic-Segmentation