/segmentation_pytorch

semantic segmentation pytorch 语义分割

Primary LanguagePython

Semantic Segmentation Pytorch

    author is leilei
    Restart this project from 2017-10-01 
    
    Now the 1.alpha.0 version has been basically completed, to be tested.
    TODO
    Add distributed and optimize code.

Environment

    python: 3.6+
    ubuntu16.04 or 18.04
    pytorch 1.6 (cuda10.2 docker)
    tensorboard 2.0
    scikit-learn 0.24.1

Note

  • If a black border is introduced, it will be regarded as one type, and the default is 0 !
  • label value is [1, N], 0 is black border class !
  • Not supporting distributed(NCCL), just support DataParallel.

Getting Started

Demo

Evaluation index

  • Just see metrics.py
  • Support acc, mean_precision, mean_recall, mean_iou

Support Network

Data Aug

support 
    random zoom-in/out, random noise,
    random blur, random color-jitter(brightness-contrast-saturation-hue)
    random affine, random rotate, random flip

Others