| { | |
| "_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" | |
| } |