Implementation of Segmentation Transformer in PyTorch, a new model to achieve SOTA in semantic segmentation while using transformer style encoders.
- SETR
- SETR-Naive
- SETR-PUP
- SETR-MLA
- SETR-Hybrid
To Do:
- Training Scripts
Create the environment:
conda env create -f environment.yml