config.json

{
  "classes": 4,
  "decoder_atrous_rates": [
    12,
    24,
    36
  ],
  "decoder_channels": 256,
  "encoder_name": "mit_b2",
  "encoder_output_stride": 16,
  "encoder_weights": "imagenet",
  "image_size": [
    384,
    384
  ],
  "in_channels": 3,
  "model_name": "segformer"
}