moridaiki/pytorch-segmentation

various networks for semantic segmentation in pytorch

PytorchSegmentation

This repository implements general network for semantic segmentation.
You can run various networks like UNet, PSPNet, ASPP, etc., just by writing the config file.

python train.py ../config/test.yaml

Directory tree

.
├── README.md
├── config
├── data
│   ├── cityscapes
│   │   ├── gtFine
│   │   └── leftImg8bit
│   └── pascal_voc_2012
│        └── VOCdevkit
│            └── VOC2012
│                ├── JPEGImages
│                ├── SegmentationClass
│                └── SegmentationClassAug
├── logs
├── model
└── src
    ├── dataset
    ├── logger
    ├── losses
    │   ├── binary
    │   └── multi
    ├── models
    │   ├── inplace_abn
    ├── start_train.sh
    ├── stop_train.sh
    ├── train.py
    └── utils

Networks

UNet

• encoder type
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • resnext101_32x4d
  • resnext101_64x4d
  • se_resnet50
  • se_resnet101
  • se_resnet152
  • se_resnext50_32x4d
  • se_resnext101_32x4d
  • senet154
• decoder type
  • unet_scse
  • unet_seibn
  • unet_oc

PSPNet

• encoder type
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • senet154
• decoder type
  • psp

ASPP

• encoder type
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • senet154
• decoder type
  • aspp

OCNet

• encoder type
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • senet154
• decoder type
  • oc_base
  • oc_aspp

Dataset

• Cityspaces
• Pascal Voc
  • augmentation
    • http://home.bharathh.info/pubs/codes/SBD/download.html
    • TheLegendAli/DeepLab-Context#10

Reference

Encoder

• https://arxiv.org/abs/1505.04597
• https://github.com/tugstugi/pytorch-saltnet

Decoder

SCSE

• https://arxiv.org/abs/1803.02579

IBN

• https://arxiv.org/abs/1807.09441
• https://github.com/XingangPan/IBN-Net
• https://github.com/SeuTao/Kaggle_TGS2018_4th_solution

OC

• https://arxiv.org/abs/1809.00916
• https://github.com/PkuRainBow/OCNet

PSP

• https://arxiv.org/abs/1612.01105

ASPP

• https://arxiv.org/abs/1802.02611

Ohter

inplaceABN

• https://arxiv.org/abs/1712.02616
• https://github.com/mapillary/inplace_abn