fpn/config.json

{
  "model": {
    "name": "fpn",
    "encoder_name": "efficientnet-b4",
    "encoder_depth": 5,
    "in_channels": 3,
    "num_classes": 7,
    "decoder_pyramid_channels": 256,
    "decoder_segmentation_channels": 128,
    "decoder_merge_policy": "add",
    "decoder_dropout": 0.2
  },
  "class_names": [
    "River",
    "Forest",
    "Lake",
    "Wetland",
    "Stream",
    "Building",
    "Road"
  ],
  "thresholds": {
    "River": 0.5,
    "Forest": 0.6,
    "Lake": 0.85,
    "Wetland": 0.2,
    "Stream": 0.65,
    "Building": 0.55,
    "Road": 0.55
  },
  "patch_size": 512
}