zh320 / realtime-semantic-segmentation-pytorch
PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDNet, LinkNet, PP-LiteSeg, SegNet, ShelfNet, STDC, SwiftNet, and support knowledge distillation, distributed training, Optuna etc.
☆195Updated last month
Alternatives and similar repositories for realtime-semantic-segmentation-pytorch:
Users that are interested in realtime-semantic-segmentation-pytorch are comparing it to the libraries listed below
- Official implementation of SCTNet (AAAI2024)☆265Updated last year
- [ICLR 2023 & IJCV 2025] SeaFormer: Squeeze-enhanced Axial Transformer☆323Updated 2 months ago
- This is the official repository for our recent work: PIDNet☆657Updated 8 months ago
- [CVPR 2024 Workshops] SERNet-Former: Semantic Segmentation by Efficient Residual Network with Attention-Boosting Gates and Attention-Fusi…☆66Updated 4 months ago
- ☆252Updated last month
- [CVPR2023] FastInst: A Simple Query-Based Model for Real-Time Instance Segmentation☆195Updated last year
- The official implementation of "Deep Dual-resolution Networks for Real-time and Accurate Semantic Segmentation of Road Scenes"☆431Updated last year
- (MICCAI23) This is the official code repository for "EGE-UNet: an Efficient Group Enhanced UNet for skin lesion segmentation".☆278Updated last year
- ☆144Updated last year
- SegNext Implementation in PyTorch☆55Updated last year
- Pytorch Implement of Dynamic Snake Convolution (ICCV2023)☆479Updated 5 months ago
- 这是一个segformer-pytorch的源码，可以用于训练自己的模型。☆363Updated last year
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆493Updated last year
- 这是一个hrnet-pytorch的库，可用于训练自己的语义分割数据集☆134Updated last year
- The official implementation of "Rethink Dilated Convolution for Real-time Semantic Segmentation"☆87Updated last year
- 该仓库中放置了一些工具，用于调整语义分割算法的数据集格式，使其符合训练要求。☆160Updated 2 years ago
- CEDNet: A Cascade Encoder-Decoder Network for Dense Prediction (Pattern Recognition 2024)☆122Updated 4 months ago
- This repository is an official PyTorch implementation of our paper 'LETNet_Lightweight-Real-time-Semantic-Segmentation-Network-with-Effic…☆30Updated last year
- Centralized Feature Pyramid for Object Detection☆247Updated last year
- [IMAVIS] Official implementation of "ASF-YOLO: A Novel YOLO Model with Attentional Scale Sequence Fusion for Cell Instance Segmentation".☆85Updated last month
- TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022☆397Updated 2 years ago
- This is a warehouse for MobileNetV4-Pytorch-model, can be used to train your image-datasets for vision tasks.☆173Updated 3 months ago
- ☆129Updated 7 months ago
- This is the unofficial code of Deep Dual-resolution Networks for Real-time and Accurate Semantic Segmentation of Road Scenes. which achi…☆168Updated 11 months ago
- Official implementation of the CVPR 2024 paper ViT-CoMer: Vision Transformer with Convolutional Multi-scale Feature Interaction for Dense…☆281Updated 3 months ago
- Code for ECML PKDD 2022 paper: No More Strided Convolutions or Pooling: A Novel CNN Architecture for Low-Resolution Images and Small Obje…☆248Updated 4 months ago
- Semantic segmentation models, datasets and losses implemented in PyTorch.☆105Updated 3 years ago
- ☆169Updated 2 months ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆252Updated last year
- [arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".☆273Updated last month