Segformer
Collection
15 items
•
Updated
•
1
Table of Contents:
import segmentation_models_pytorch as smp
model = smp.from_pretrained("smp-hub/segformer-b5-640x640-ade-160k")
model_init_params = {
"encoder_name": "mit_b5",
"encoder_depth": 5,
"encoder_weights": None,
"decoder_segmentation_channels": 768,
"in_channels": 3,
"classes": 150,
"activation": None,
"aux_params": None
}
[More Information Needed]
Dataset name: [More Information Needed]
This model has been pushed to the Hub using the PytorchModelHubMixin