deep-learning-analytics/segformer_semantic_segmentation
144
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 