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-resnet18",
3 "encoder_weights": "imagenet",
4 "encoder_depth": 3,
5 "psp_out_channels": 512,
6 "psp_use_batchnorm": True,
7 "psp_dropout": 0.2,
8 "in_channels": 3,
9 "classes": 1,
10 "activation": None,
11 "upsampling": 8,
12 "aux_params": None
13}