/pytorch-segmentation

Pytorch for Segmentation

Primary LanguagePython

Pytorch for Semantic Segmentation

This repo has been deprecated currently and I will not maintain it. Meanwhile, I strongly recommend you can refer to my new repo: TorchSeg, which offers fast, modular reference implementation and easy training of semantic segmentation algorithms in PyTorch.

A repository contains some exiting networks and some experimental networks for semantic segmentation.

  • ResNet(FCN)
    • ResNet-50
    • ResNet-101
    • Wide-ResNet
  • DUC(Understanding Convolution for Semantic Segmentation)[pdf]
  • GCN(Large Kernel Matters -- Improve Semantic Segmentation by Global Convolutional Network)[pdf]