CoolFace
Modelpublic

deep-learning-analytics/segformer_semantic_segmentation

sourceHugging Faceupdated 5y agoView on Hugging Face
1likes44downloads
config.json123 linesDownload Raw Back to root
1{2  "_name_or_path": "./segformer_model",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    6,12    40,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": "unlabeled",32    "1": "paved-area",33    "2": "dirt",34    "3": "grass",35    "4": "gravel",36    "5": "water",37    "6": "rocks",38    "7": "pool",39    "8": "vegetation",40    "9": "roof",41    "10": "wall",42    "11": "window",43    "12": "door",44    "13": "fence",45    "14": "fence-pole",46    "15": "person",47    "16": "dog",48    "17": "car",49    "18": "bicycle",50    "19": "tree",51    "20": "bald-tree",52    "21": "ar-marker",53    "22": "obstacle",54    "23": "conflicting"55  },56  "image_size": 224,57  "initializer_range": 0.02,58  "label2id": {59    "ar-marker": 21,60    "bald-tree": 20,61    "bicycle": 18,62    "car": 17,63    "conflicting": 23,64    "dirt": 2,65    "dog": 16,66    "door": 12,67    "fence": 13,68    "fence-pole": 14,69    "grass": 3,70    "gravel": 4,71    "obstacle": 22,72    "paved-area": 1,73    "person": 15,74    "pool": 7,75    "rocks": 6,76    "roof": 9,77    "tree": 19,78    "unlabeled": 0,79    "vegetation": 8,80    "wall": 10,81    "water": 5,82    "window": 1183  },84  "layer_norm_eps": 1e-06,85  "mlp_ratios": [86    4,87    4,88    4,89    490  ],91  "model_type": "segformer",92  "num_attention_heads": [93    1,94    2,95    5,96    897  ],98  "num_channels": 3,99  "num_encoder_blocks": 4,100  "patch_sizes": [101    7,102    3,103    3,104    3105  ],106  "reshape_last_stage": true,107  "semantic_loss_ignore_index": 255,108  "sr_ratios": [109    8,110    4,111    2,112    1113  ],114  "strides": [115    4,116    2,117    2,118    2119  ],120  "torch_dtype": "float32",121  "transformers_version": "4.15.0"122}123