qubvel-org/segmentation_models.pytorch

Add upsampling parameter to Segformer

Ouverte

#1 106 ouverte le 26 mars 2025

 (1 commentaire) (1 réaction) (0 personne assignée)Python (1 842 forks)github user discovery
featuregood first issue

Métriques du dépôt

Stars
 (11 704 étoiles)
Métriques de merge PR
 (Merge moyen 4h 14m) (24 PRs mergées en 30 j)

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

Guide contributeur