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

81 lines
2.0 KiB
JSON

{
"camera": "oak-fcc-3",
"modelo": "segformer_b1",
"model_name": "2026_08_18",
"main_class_name": "erva",
"es_classes": "",
"model_to_use": "geral",
"raw_size": [1280, 800],
"resolucao": [640, 400],
"roi_inicio": 0.0,
"roi_tamanho": 1.0,
"shaves": 3,
"source_channels": ["R", "G", "B", "RE", "NIR"],
"channels": 7,
"input_channels": ["R", "G", "B", "RE", "NIR", "NDVI", "NDRE"],
"derived_channels": {
"epsilon": 1e-6,
"clip_min": -1.0,
"clip_max": 1.0
},
"backbone": "nvidia/mit-b1",
"fusion_mode": "stacked",
"stats_source_tag": "stacked_raw5_ndvi_ndre",
"module_params_json": "calibration/module_params.json",
"ckpt_test": "best_score",
"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
}
}