qubvel-org/segmentation_models.pytorch

Add upsampling parameter to Segformer

Open

#1,106 opened on Mar 26, 2025

 (1 comment) (1 reaction) (0 assignees)Python (1,842 forks)github user discovery
featuregood first issue

Repository metrics

Stars
 (11,704 stars)
PR merge metrics
 (Avg merge 4h 14m) (24 merged PRs in 30d)

Description

I did this locally and it works perfectly well, but I legally cannot make a pull request. It is simple and easy.

On file segmentation_models_pytorch/decoders/segformer/model.py:

Line 60: Add upsampling: int = 4,, after activation param and before aux_params. Line 85: swap the parameter value from 4 to upsampling.

This is all. I've done it locally and I think the community would benefit.

Thank you

Contributor guide