lucidrains / segformer-pytorch
Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch
☆362Updated 2 years ago
Alternatives and similar repositories for segformer-pytorch:
Users that are interested in segformer-pytorch are comparing it to the libraries listed below
- [ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation☆874Updated last year
- Official Pytorch implementations for "SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation" (NeurIPS 2022)☆829Updated 2 years ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation…☆1,218Updated 2 years ago
- [CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆1,076Updated 8 months ago
- ☆194Updated 2 years ago
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆453Updated 3 years ago
- ICCV2021 (Oral) - Exploring Cross-Image Pixel Contrast for Semantic Segmentation☆677Updated 2 years ago
- CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022☆565Updated last year
- Official Pytorch Code base for "UNeXt: MLP-based Rapid Medical Image Segmentation Network", MICCAI 2022☆519Updated last year
- Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆196Updated 4 years ago
- PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" (CVPR 2022)☆193Updated 2 years ago
- Official Pytorch Implementation of SegViT: Semantic Segmentation with Plain Vision Transformers☆241Updated last year
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆494Updated last year
- [ NeurIPS2021] This is an official implementation of our paper "HRFormer: High-Resolution Transformer for Dense Prediction".☆506Updated 2 years ago
- [CVPR 2021] CPS: Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision☆519Updated last year
- [CVPR22] Official Implementation of DAFormer: Improving Network Architectures and Training Strategies for Domain-Adaptive Semantic Segmen…☆499Updated 8 months ago
- Implementation of SETR model, Original paper: Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers.☆132Updated 4 years ago
- TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022☆397Updated 2 years ago
- [ECCV-2020-oral]-Semantic Flow for Fast and Accurate Scene Parsing☆378Updated last year
- This is an official implementation of "Polarized Self-Attention: Towards High-quality Pixel-wise Regression"☆255Updated 3 years ago
- [NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation☆473Updated 3 years ago
- Per-Pixel Classification is Not All You Need for Semantic Segmentation (NeurIPS 2021, spotlight)☆1,396Updated 3 years ago
- This Repository is implementation of majority of Semantic Segmentation Loss Functions☆568Updated 2 years ago
- Simple image segmentation pipeline in pytorch, using HRNet and SegFormer models☆84Updated 2 years ago
- SOTA Semantic Segmentation Models in PyTorch☆901Updated last year
- [NIVT Workshop @ ICCV 2023] SeMask: Semantically Masked Transformers for Semantic Segmentation☆253Updated last year
- A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder☆158Updated 4 years ago
- Unsupervised Semantic Segmentation by Contrasting Object Mask Proposals. [ICCV 2021]☆411Updated 2 years ago
- HyperSeg - Official PyTorch Implementation☆213Updated 2 years ago
- [NeurIPS 2021 Spotlight] Official code for "Focal Self-attention for Local-Global Interactions in Vision Transformers"☆554Updated 3 years ago