kozistr / Awesome-Segmentations

Lots of semantic image segmentation implementations in Tensorflow/Keras

Awesome-Segmentation

Lots of Image Semantic Segmentation Implementations in Tensorflow/Keras

Highly inspired by HERE

Currently, under-development :(

Prerequisites

• python 3.x
• tensorflow 1.x
• keras 2.x
• numpy
• scikit-image
• opencv-python
• h5py
• tqdm

Usage

Dependency Install

$ sudo python3 -m pip install -r requirements.txt

Training Model

(Before running train.py, MAKE SURE run after downloading DataSet & changing DataSet's directory in xxx_train.py)
just after it, RUN train.py
$ python3 xxx_train.py

Implementation List

• FCNet
• SegNet
• U-Net
• FusionNet
• FC-DenseNet
• ENet
• LinkNet
• RefineNet
• PSPNet
• Mask R-CNN
• DecoupledNet
• GAN-SS
• G-FRNet

DataSets

• MS COCO 2017 DataSet will be used!

DataSet	Train	Validate	Test	Disk
MS COCO 2017	118287	5000	40670	26.3GB

Repo Tree

│
├── xxNet
│    ├──gan_img (generated images)
│    │     ├── train_xxx.png
│    │     └── train_xxx.png
│    ├── model  (model)
│    │     └── model.txt (google-drive link for pre-trained model)
│    ├── xxx_model.py (model)
│    ├── xxx_train.py (trainer)
│    ├── xxx_tb.png   (Tensor-Board result)
│    └── readme.md    (results & explains)
├── metrics.py        (metrics)
├── tfutil.py         (useful TF utils)
├── image_utils.py    (image processing)
└── datasets.py       (DataSet loader)

Pre-Trained Models

Here's a google drive link. You can download pre-trained models from here !

Papers & Codes

Name	Summary	Paper	Code
FCN	Fully Convolutional Networks for Semantic Segmentation	[arXiv]
SegNet	A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation	[arXiv]
U-Net	Convolutional Networks for Biomedical Image Segmentation	[arXiv]
FusionNet	A deep fully residual convolutional neural network for image segmentation in connectomics	[arXiv]
FC-DenseNet	Fully Convolutional DenseNets for Semantic Segmentation	[arXiv]
ENet	A Deep Neural Network Architecture for Real-Time Semantic Segmentation	[arXiv]
LinkNet	Exploiting Encoder Representations for Efficient Semantic Segmentation	[arXiv]
Mask R-CNN	Mask R-CNN	[arXiv]
PSPNet	Pyramid Scene Parsing Network	[arXiv]
RefineNet	Multi-Path Refinement Networks for High-Resolution Semantic Segmentation	[arXiv]
G-FRNet	Gated Feedback Refinement Network for Dense Image Labeling	[CVPR2017]
DeepLabv3+	Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation	[arXiv]
DecoupledNet	Decoupled Deep Neural Network for Semi-supervised Semantic Segmentation	[arXiv]
GAN-SS	Semi and Weakly Supervised Semantic Segmentation Using Generative Adversarial Network	[arXiv]

To-Do

1. Implement FCN
2. Implement Mask R-CNN
3. Upload U-Net (Tuned)
4. Upload FC-DenseNet
5. Upload DeepLabv3+

ETC

Any suggestions and PRs and issues are WELCONE :)

Author

HyeongChan Kim / @kozistr