Skip to content

gupta-abhay/setr-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Segmentation Transformer

Implementation of Segmentation Transformer in PyTorch, a new model to achieve SOTA in semantic segmentation while using transformer style encoders.

SETR

Features

  • SETR
    • SETR-Naive
    • SETR-PUP
    • SETR-MLA
  • SETR-Hybrid

To Do:

  • Training Scripts

Installation

Create the environment:

conda env create -f environment.yml

About

Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers

Topics

Resources

License

Stars

Watchers

Forks

Languages