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