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": "mobileone_s4",
3 "encoder_depth": 5,
4 "encoder_weights": "imagenet",
5 "encoder_output_stride": 16,
6 "decoder_channels": 32,
7 "decoder_interpolation": "bilinear",
8 "in_channels": 3,
9 "classes": 1,
10 "activation": None,
11 "upsampling": 4,
12 "aux_params": None
13}1[
2 {
3 "test_per_image_iou": 0.8509070873260498,
4 "test_dataset_iou": 0.8780016303062439
5 }
6]