SegFly-Firefly / Firefly_RGB /config.json
markus-42's picture
Initial commit
94cbfa6
Raw
History Blame Contribute Delete
962 Bytes
{
"architectures": ["FireflyForSemanticSegmentationRGB"],
"model_type": "firefly-rgb",
"num_labels": 15,
"backbone": "dinov3_vitb16",
"image_size": 640,
"embedding_dim": 256,
"backbone_embed_dim": 768,
"patch_size": 16,
"dropout_ratio": 0.1,
"semantic_loss_ignore_index": 255,
"id2label": {
"0": "road",
"1": "walkway",
"2": "dirt",
"3": "gravel",
"4": "grass",
"5": "vegetation",
"6": "tree",
"7": "ground_obstacles",
"8": "vehicle",
"9": "water",
"10": "building",
"11": "roof",
"12": "parking_lot",
"13": "constructions",
"14": "truck"
},
"label2id": {
"road": 0,
"walkway": 1,
"dirt": 2,
"gravel": 3,
"grass": 4,
"vegetation": 5,
"tree": 6,
"ground_obstacles": 7,
"vehicle": 8,
"water": 9,
"building": 10,
"roof": 11,
"parking_lot": 12,
"constructions": 13,
"truck": 14
},
"transformers_version": "4.37.2"
}