qubvel-org/segmentation_models.pytorch

Add upsampling parameter to Segformer

Aberta

#1.106 aberto em 26 de mar. de 2025

 (1 comentário) (1 reação) (0 responsável)Python (1.842 forks)github user discovery
featuregood first issue

Métricas do repositório

Stars
 (11.704 estrelas)
Métricas de merge de PR
 (Mesclagem média 4h 14m) (24 fundiu PRs em 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

Guia do colaborador