Views
No views yet
1import segmentation_models_pytorch as smp
2
3model = smp.from_pretrained("<save-directory-or-this-repo>")1model_init_params = {
2 "encoder_name": "tu-vit_large_patch16_384",
3 "encoder_depth": 4,
4 "encoder_weights": None,
5 "feature_dim": 256,
6 "in_channels": 3,
7 "classes": 150,
8 "activation": None,
9 "aux_params": None,
10 "output_stride": None
11}