search

kozistr / Awesome-Segmentations

Lots of semantic image segmentation implementations in Tensorflow/Keras
MIT License
65 stars 15 forks source link
decouplednet enet fc-densenet fcn g-frnet gan-ss image-segmentation linknet mask-rcnn mscoo-dataset pspnet refinenet segnet tensorflow unet

readme

Awesome-Segmentation

Lots of Image Semantic Segmentation Implementations in Tensorflow/Keras

Highly inspired by HERE

Currently, under-development :(

Prerequisites

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

DataSets

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