qubvel-org/segmentation_models.pytorch

Add upsampling parameter to Segformer

Aperta

#1106 aperta il 26 mar 2025

 (1 commento) (1 reazione) (0 assegnatari)Python (1842 fork)github user discovery
featuregood first issue

Metriche repository

Star
 (11.704 stelle)
Metriche merge PR
 (Merge medio 4h 14m) (24 PR mergiate in 30 g)

Descrizione

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

Guida contributor