• Stars
    star
    105
  • Rank 328,196 (Top 7 %)
  • Language
    Python
  • Created over 6 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

semantic segmentation pytorch 语义分割

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