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-test_vit",
3 "encoder_depth": 4,
4 "encoder_weights": "imagenet",
5 "encoder_output_indices": None,
6 "decoder_readout": "cat",
7 "decoder_intermediate_channels": (16, 32, 64, 64),
8 "decoder_fusion_channels": 16,
9 "in_channels": 3,
10 "classes": 1,
11 "activation": None,
12 "aux_params": None,
13 "dynamic_img_size": True
14}