| { |
| "_name_or_path": "custom-unet-cloud-segmentation", |
| "architectures": [ |
| "UnetModel" |
| ], |
| "model_type": "semantic_segmentation", |
| "in_channels": 3, |
| "out_channels": 2, |
| "base_filters": 64, |
| "depth": 4, |
| "downsampling_blocks": ["ConvBlock", "ConvBlock", "ConvBlock", "ConvBlock"], |
| "upsampling_blocks": ["UpConvBlock", "UpConvBlock", "UpConvBlock", "UpConvBlock"], |
| "id2label": { |
| "0": "Background/Clear_Sky", |
| "1": "Cloud" |
| }, |
| "label2id": { |
| "Background/Clear_Sky": 0, |
| "Cloud": 1 |
| }, |
| "num_labels": 2, |
| "image_size": [256, 256], |
| "pytorch_version": "2.1.0" |
| } |
