qubvel-org/segmentation_models.pytorch

View on GitHub

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

Stars: 11.7k
Forks: 1.8k
Open beginner issues: 0
Indexed issues: 70
Avg merge: 5h 42m
Merged PRs (30d): 11

Dominant language: Python
License: MIT
Last GitHub push: Sep 15, 2026
Latest indexed: Sep 19, 2026
Contributing guide: No contributing guide
Code of conduct: No code of conduct
Beginner labels: good first issue help wanted

Beginner friendly Everything

Open All

70 open issues indexed

DPTDecoder doesn't work properly with VIT using patch_size=14 Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

qubvel-org/segmentation_models.pytorch#1328 · Opened Apr 20, 2026
The training result is so much lower than the sample code. Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1323 · 2 comments · Opened Apr 13, 2026
ONNX Unsupported export of operator adaptive_avg_pool2d Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1266 · Opened Feb 10, 2026
Inconsistent `from_logits` parameter across loss functions Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

qubvel-org/segmentation_models.pytorch#1263 · 3 comments · Opened Dec 2, 2025
How to load Huggingface pretrained model for SMP training? Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1252 · Opened Oct 14, 2025
can I use convnext to unet model Open

Difficulty 5/5 Over a week Newbie friendliness 30/100

qubvel-org/segmentation_models.pytorch#1251 · Opened Oct 14, 2025
VIT Adapter Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1244 · 5 comments · 5 reactions · Opened Sep 21, 2025
how to print Dice,HD95 metrics? Open

Difficulty 3/5 1-2 days Newbie friendliness 20/100

qubvel-org/segmentation_models.pytorch#1241 · Opened Sep 18, 2025
SOTA Panoptic Methods Implementation Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1240 · 1 reaction · Opened Sep 18, 2025
segformer model implementation != original arch design Open

Difficulty 4/5 3-5 days Newbie friendliness 28/100

qubvel-org/segmentation_models.pytorch#1237 · 8 comments · Opened Sep 15, 2025
Possibility of optimisation in Focal Loss implementation Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

qubvel-org/segmentation_models.pytorch#1235 · 1 comment · Opened Sep 14, 2025
Example "Train multiclass segmentation on CamVid" Image normalization may be error Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1224 · 2 comments · Opened Aug 29, 2025
tu-vit_base_patch16_224 not supported Open

Difficulty 3/5 1-2 days Newbie friendliness 38/100

qubvel-org/segmentation_models.pytorch#1184 · 4 comments · Opened Jun 22, 2025
AssertionError description typo Open

Difficulty 1/5 Under an hour Newbie friendliness 55/100

qubvel-org/segmentation_models.pytorch#1167 · Opened May 28, 2025
KeyError when unet uses certain encoders Open

Difficulty 3/5 1-2 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1159 · 3 comments · Opened May 26, 2025
Changing dilation rate of backbones Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1158 · 1 comment · Opened May 23, 2025
Exporting UnetPlusPlus as TorchScript gives error Open

Difficulty 4/5 3-5 days Newbie friendliness 46/100

qubvel-org/segmentation_models.pytorch#1156 · Opened May 22, 2025
How to migrate existing models from 0.4.0 to 0.5.0? Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1144 · 4 comments · Opened May 5, 2025
Add EoMT from ViT is Secretly an Image Segmentation Model Open

good difficult issue new-model

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1132 · 2 comments · Opened Apr 19, 2025
Addition of other losses Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1114 · Opened Apr 3, 2025
Input Shape for Training and Inference with Pretrained Models Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1112 · 2 comments · Opened Apr 2, 2025
Add upsampling parameter to Segformer Open

feature good first issue

Difficulty 1/5 Under an hour Newbie friendliness 90/100

qubvel-org/segmentation_models.pytorch#1106 · 3 comments · 1 reaction · Opened Mar 26, 2025
Encoder weights from local directory Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1102 · 4 comments · 1 reaction · Opened Mar 25, 2025
Is there any other pre-trained model in this pkg besides segformer? Open

Difficulty 1/5 Under an hour Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1100 · 2 comments · Opened Mar 24, 2025
Using Vanilla Encoder Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1089 · 1 comment · Opened Mar 12, 2025
Image Size Changes from [640, 640] to [H, W] During Training? Open

Difficulty 3/5 1-2 days Newbie friendliness 30/100

qubvel-org/segmentation_models.pytorch#1088 · 1 comment · Opened Mar 10, 2025
Crash on Colab When Switching Encoder to 'mobilenet_v2' for the Binary Segmentation Tutorial Example Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1085 · Opened Mar 9, 2025
Xception is unavailable Open

Difficulty 3/5 1-2 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1076 · 1 comment · Opened Feb 24, 2025
Adding DPT to SMP Open

new-model

Difficulty 5/5 Over a week Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1073 · 6 comments · Opened Feb 23, 2025
Camvid dataset labels Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1072 · 2 comments · Opened Feb 21, 2025
Can the model be stopped early during custom training? Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1071 · 2 comments · Opened Feb 21, 2025
InceptionV4 pretrained weights link is dead Open

Difficulty 2/5 1-3 hours Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1065 · 1 comment · Opened Feb 19, 2025
How to calculate and print iou for each category in multi-classification task of camvid dataset? Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

qubvel-org/segmentation_models.pytorch#1064 · Opened Feb 19, 2025
Does custom training support setting parameters such as step and learning rate? Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

qubvel-org/segmentation_models.pytorch#1063 · 2 comments · Opened Feb 19, 2025
DeepLabV3 and DeepLabV3+ cannot use encoder “Xception” Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1062 · 2 comments · Opened Feb 18, 2025
Docs are sometimes a unclear Open

documentation

Difficulty 1/5 Under an hour Newbie friendliness 68/100

qubvel-org/segmentation_models.pytorch#1061 · 3 comments · Opened Feb 16, 2025
How can I perform data training on a custom binary segmentation dataset? Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1056 · 3 comments · Opened Feb 7, 2025
Add Mask2Former to SMP Open

good difficult issue help wanted new-model

Difficulty 5/5 Over a week Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1044 · 5 comments · 1 reaction · Opened Jan 24, 2025
Add Beit segmentation model Open

good first issue help wanted new-model

Difficulty 5/5 Over a week Newbie friendliness 28/100

qubvel-org/segmentation_models.pytorch#1024 · 10 comments · 1 reaction · Opened Jan 7, 2025
Ensure ONNX support Open

feature-request ONNX

Difficulty 5/5 Over a week Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#1015 · 2 comments · 1 reaction · Opened Dec 26, 2024
FPN Training Issue: Zero F-Score/IoU Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#1007 · Opened Dec 12, 2024
How to add a mamba backbone as an encoder for training Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

qubvel-org/segmentation_models.pytorch#1003 · 3 comments · Opened Dec 6, 2024
Add example of amp training & qunatization-aware training Open

examples feature-request good first issue help wanted

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#995 · 5 comments · Opened Dec 4, 2024
Pytorch Lightning Checkpoint saving and Earlystopping Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#979 · 4 comments · Opened Nov 15, 2024
TypeError: __init__() got an unexpected keyword argument 'output_stride' Open

Difficulty 3/5 1-2 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#978 · 1 comment · Opened Nov 14, 2024
Error when trying to parse the encoder output to the decoder Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#977 · 1 comment · Opened Nov 14, 2024
Where is the tensorRT deployment tutorial Open

Difficulty 1/5 Under an hour Newbie friendliness 20/100

qubvel-org/segmentation_models.pytorch#973 · Opened Nov 4, 2024
Encoder Compatibility Issue Across Models Using mit_b0 Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

qubvel-org/segmentation_models.pytorch#967 · 1 comment · 1 reaction · Opened Oct 28, 2024
Model preservation and transformation problems(模型保存和转换的问题) Open

Difficulty 3/5 1-2 days Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#964 · 7 comments · Opened Oct 26, 2024
add pixel-wise loss weight to unet Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

qubvel-org/segmentation_models.pytorch#963 · Opened Oct 25, 2024

qubvel-org/segmentation_models.pytorch

Get new issues in your inbox