{ "_name_or_path": "nvidia/segformer-b3-finetuned-cityscapes-1024-1024", "architectures": [ "SegformerForSemanticSegmentation" ], "attention_probs_dropout_prob": 0.0, "classifier_dropout_prob": 0.1, "decoder_hidden_size": 768, "depths": [ 3, 4, 18, 3 ], "downsampling_rates": [ 1, 4, 8, 16 ], "drop_path_rate": 0.1, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_sizes": [ 64, 128, 320, 512 ], "id2label": { "0": "unlabeled", "1": "ego vehicle", "2": "rectification border", "3": "out of roi", "4": "static", "5": "dynamic", "6": "ground", "7": "road", "8": "sidewalk", "9": "parking", "10": "rail track", "11": "building", "12": "wall", "13": "fence", "14": "guard rail", "15": "bridge", "16": "tunnel", "17": "pole", "18": "polegroup", "19": "traffic light", "20": "traffic sign", "21": "vegetation", "22": "terrain", "23": "sky", "24": "person", "25": "rider", "26": "car", "27": "truck", "28": "bus", "29": "caravan", "30": "trailer", "31": "train", "32": "motorcycle", "33": "bicycle", "34": "license plate" }, "image_size": 224, "initializer_range": 0.02, "label2id": { "bicycle": 33, "bridge": 15, "building": 11, "bus": 28, "car": 26, "caravan": 29, "dynamic": 5, "ego vehicle": 1, "fence": 13, "ground": 6, "guard rail": 14, "license plate": 34, "motorcycle": 32, "out of roi": 3, "parking": 9, "person": 24, "pole": 17, "polegroup": 18, "rail track": 10, "rectification border": 2, "rider": 25, "road": 7, "sidewalk": 8, "sky": 23, "static": 4, "terrain": 22, "traffic light": 19, "traffic sign": 20, "trailer": 30, "train": 31, "truck": 27, "tunnel": 16, "unlabeled": 0, "vegetation": 21, "wall": 12 }, "layer_norm_eps": 1e-06, "mlp_ratios": [ 4, 4, 4, 4 ], "model_type": "segformer", "num_attention_heads": [ 1, 2, 5, 8 ], "num_channels": 3, "num_encoder_blocks": 4, "patch_sizes": [ 7, 3, 3, 3 ], "reshape_last_stage": true, "semantic_loss_ignore_index": 255, "sr_ratios": [ 8, 4, 2, 1 ], "strides": [ 4, 2, 2, 2 ], "torch_dtype": "float32", "transformers_version": "4.47.1" }