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-sam2_hiera_base_plus.fb_r896",
3 "encoder_depth": 5,
4 "encoder_weights": "imagenet",
5 "decoder_use_norm": "batchnorm",
6 "decoder_channels": (256, 128, 64, 32, 16),
7 "decoder_attention_type": None,
8 "decoder_interpolation": "nearest",
9 "in_channels": 3,
10 "classes": 4,
11 "activation": None,
12 "aux_params": None
13}