Skip to content

gengyanlei/segmentation_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

semantic segmentation pytorch 语义分割

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages