agrobot_base/Python/OAK/datasets/oak-fcc-3/config.json

{
    "camera": "oak-fcc-3",
    "modelo": "segformer_b1",
    "model_name": "copypaste",
    "main_class_name": "cana", 
    "es_classes": "", 
    "model_to_use": "geral",
    "raw_size": [1280, 800], 
    "resolucao": [1024, 640], 
    "roi_inicio": 0.0,
    "roi_tamanho": 1.0,
    "shaves": 3,
    "channels": 5,
    "input_channels": ["R", "G", "B", "RE", "NIR"],
    "use_ndvi": false,
    "backbone": "nvidia/mit-b1",
    "fusion_mode": "stacked",
    "stats_source_tag": "stacked_raw5",
    "module_params_json": "calibration/module_params.json",
    "ckpt_test": "best_target",
    "multi_head": true,
    "heads": {
        "semantic": {
            "enabled": true,
            "type": "multiclass",
            "num_classes": 3,
            "mask_dir": "masks",
            "classes": {"chao": 0, "cana": 1, "erva": 2},
            "ignore_index": 255,
            "loss_weight": 0.20
        },
        "vegetation": {
            "enabled": true,
            "type": "binary",
            "num_classes": 2,
            "mask_dir": "masks_vegetation",
            "classes": {"background": 0, "vegetation": 1},
            "ignore_index": 255,
            "loss_weight": 0.20
        },
        "cana": {
            "enabled": true,
            "type": "binary",
            "num_classes": 2,
            "mask_dir": "masks_cana",
            "classes": {"not_cana": 0, "cana": 1},
            "ignore_index": 255,
            "loss_weight": 0.25
        },
        "target": {
            "enabled": true,
            "type": "binary",
            "num_classes": 2,
            "mask_dir": "__derived_target__",
            "classes": {"background": 0, "target": 1},
            "ignore_index": 255,
            "loss_weight": 0.35,
            "derived": true
        }
    },
    "target_distillation": {
        "enabled": true,

        "hard_weight": 0.85,
        "distill_weight": 0.15,

        "rampup_enabled": true,
        "start_epoch": 8,
        "rampup_epochs": 12,

        "w_sem_erva": 0.45,
        "w_veg_not_cana": 0.35,
        "w_veg_suppressed": 0.20,
        "cana_suppression_power": 1.5,

        "teacher_min": 0.0,
        "teacher_max": 1.0,
        "detach_teacher": true
    }
}