gupta-abhay / setr-pytorch

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

Home Page:https://arxiv.org/abs/2012.15840

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

https://arxiv.org/abs/2012.15840

License:MIT License


Languages

Language:Python 100.0%