Hi all,
We have released the TorchSeg 35.
In this repo, we will release the fast, modular reference implementation of semantic segmentation algorithms in PyTorch.
Now, this repo has implemented some state-of-art semantic segmentation algorithms: PSPNet 1, DFN 2, BiSeNet 2, PSANet 2.
More algorithms will be supported.
Highlights


Modular Design:  easily construct customized semantic segmentation models by combining different components.

Distributed Training:   >60%  faster than the multi-thread parallel method(nn.DataParallel), we use the multi-processing parallel method.

Multi-GPU training and inference:  support different manners of inference.
Provides pre-trained models and implement different semantic segmentation models.


  
      
      GitHub
  
  
    

ycszen/TorchSeg 35

Fast, modular reference implementation and easy training of Semantic Segmentation algorithms in PyTorch. - ycszen/TorchSeg