lucidrains / segformer-pytorch
Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch
☆351Updated 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☆854Updated 9 months ago
- [CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆1,060Updated 5 months ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation…☆1,200Updated 2 years ago
- Official Pytorch implementations for "SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation" (NeurIPS 2022)☆808Updated 2 years ago
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆450Updated 3 years ago
- Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆194Updated 4 years ago
- ☆191Updated 2 years ago
- ICCV2021 (Oral) - Exploring Cross-Image Pixel Contrast for Semantic Segmentation☆672Updated 2 years ago
- Per-Pixel Classification is Not All You Need for Semantic Segmentation (NeurIPS 2021, spotlight)☆1,377Updated 2 years ago
- CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022☆557Updated last year
- Official Pytorch Implementation of SegViT: Semantic Segmentation with Plain Vision Transformers☆236Updated last year
- Unsupervised Semantic Segmentation by Contrasting Object Mask Proposals. [ICCV 2021]☆409Updated 2 years ago
- [CVPR 2021] CPS: Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision☆512Updated last year
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆482Updated last year
- This Repository is implementation of majority of Semantic Segmentation Loss Functions☆565Updated 2 years ago
- TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022☆389Updated 2 years ago
- PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" [CVPR 2022].☆189Updated 2 years ago
- Implementation of SETR model, Original paper: Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers.☆131Updated 4 years ago
- Simple image segmentation pipeline in pytorch, using HRNet and SegFormer models☆84Updated 2 years ago
- This is an official implementation for "ResT: An Efficient Transformer for Visual Recognition".☆282Updated 2 years ago
- (ICCV 2021 Oral) CoaT: Co-Scale Conv-Attentional Image Transformers☆231Updated 3 years ago
- This is an official implementation of "Polarized Self-Attention: Towards High-quality Pixel-wise Regression"☆253Updated 3 years ago
- Don't feel pain to use Deformable Convolution☆336Updated last year
- [ NeurIPS2021] This is an official implementation of our paper "HRFormer: High-Resolution Transformer for Dense Prediction".☆499Updated 2 years ago
- [ECCV-2020-oral]-Semantic Flow for Fast and Accurate Scene Parsing☆378Updated 11 months ago
- SOTR: Segmenting Objects with Transformers☆193Updated 3 years ago
- Official code for "Boundary loss for highly unbalanced segmentation", runner-up for best paper award at MIDL 2019. Extended version in Me…☆672Updated 10 months ago
- [NIVT Workshop @ ICCV 2023] SeMask: Semantically Masked Transformers for Semantic Segmentation☆251Updated last year
- PraNet: Parallel Reverse Attention Network for Polyp Segmentation, MICCAI 2020 (Oral). Code using Jittor Framework is available.☆477Updated this week
- Official Pytorch Code base for "UNeXt: MLP-based Rapid Medical Image Segmentation Network", MICCAI 2022☆495Updated last year