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