qubvel-org/segmentation_models.pytorch

Add upsampling parameter to Segformer

Offen

#1.106 geöffnet am 26.03.2025

 (1 Kommentar) (1 Reaktion) (0 zugewiesene Personen)Python (1.842 Forks)github user discovery
featuregood first issue

Repository-Metriken

Stars
 (11.704 Sterne)
PR-Merge-Metriken
 (Durchschn. Merge 4h 14m) (24 gemergte PRs in 30 T)

Beschreibung

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