CoolFace
Modelpublic

naufalso/segformer-b2-carla-car-body-segmentation

sourceHugging Faceotherupdated 2y agoView on Hugging Face
0likes10downloads
config.json81 linesDownload Raw Back to root
1{2  "_name_or_path": "nvidia/segformer-b2-finetuned-ade-512-512",3  "architectures": [4    "SegformerForSemanticSegmentation"5  ],6  "attention_probs_dropout_prob": 0.0,7  "classifier_dropout_prob": 0.1,8  "decoder_hidden_size": 768,9  "depths": [10    3,11    4,12    6,13    314  ],15  "downsampling_rates": [16    1,17    4,18    8,19    1620  ],21  "drop_path_rate": 0.1,22  "hidden_act": "gelu",23  "hidden_dropout_prob": 0.0,24  "hidden_sizes": [25    64,26    128,27    320,28    51229  ],30  "id2label": {31    "0": "background",32    "1": "car_body",33    "2": "car_non_body"34  },35  "image_size": 224,36  "initializer_range": 0.02,37  "label2id": {38    "background": 0,39    "car_body": 1,40    "car_non_body": 241  },42  "layer_norm_eps": 1e-06,43  "mlp_ratios": [44    4,45    4,46    4,47    448  ],49  "model_type": "segformer",50  "num_attention_heads": [51    1,52    2,53    5,54    855  ],56  "num_channels": 3,57  "num_encoder_blocks": 4,58  "patch_sizes": [59    7,60    3,61    3,62    363  ],64  "reshape_last_stage": true,65  "semantic_loss_ignore_index": 255,66  "sr_ratios": [67    8,68    4,69    2,70    171  ],72  "strides": [73    4,74    2,75    2,76    277  ],78  "torch_dtype": "float32",79  "transformers_version": "4.41.2"80}81