config.json

{
    "model_type": "SegformerForSemanticSegmentation",
    "backbone": "nvidia/segformer-b3-finetuned-ade-512-512",
    "input_resolution": "Dynamic (Resized by Image Processor)",
    "learning_rate": 6e-05,
    "epochs": 150,
    "batch_size": 8,
    "train_sample_fraction": 0.7,
    "classes": {
        "0": "background",
        "1": "damage",
        "2": "root"
    }
}