qubvel-org/segmentation_models.pytorch
Add upsampling parameter to Segformer
Offen
#1.106 geöffnet am 26.03.2025
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