config.json

{
  "task": "image-segmentation",
  "model_name": "segformer-mit-b4",
  "model_kwargs": {
    "encoder_name": "mit_b4",
    "encoder_weights": "imagenet",
    "in_channels": 3,
    "num_classes": 4,
  },
  "classes": ["clear", "thick cloud", "thin cloud", "cloud shadow"],
  "input_bands": ["B04", "B03", "B02"]
}