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": "resnet34",
3 "encoder_depth": 5,
4 "encoder_weights": "imagenet",
5 "decoder_pyramid_channels": 256,
6 "decoder_segmentation_channels": 128,
7 "decoder_merge_policy": "add",
8 "decoder_dropout": 0.2,
9 "in_channels": 3,
10 "classes": 12,
11 "activation": None,
12 "upsampling": 4,
13 "aux_params": None
14}1{
2 "test_per_image_iou": 0.7410496473312378,
3 "test_dataset_iou": 0.7378937005996704
4}