CoolFace
Modelpublic

SpotLab/MobileViT_DeepLabv3

sourceHugging Faceotherupdated 3y agoView on Hugging Face
0likes251downloads
config.json92 linesDownload Raw Back to root
1{2  "architectures": [3    "MobileViTForSemanticSegmentation"4  ],5  "aspp_dropout_prob": 0.1,6  "aspp_out_channels": 256,7  "atrous_rates": [8    6,9    12,10    1811  ],12  "attention_probs_dropout_prob": 0.0,13  "classifier_dropout_prob": 0.1,14  "conv_kernel_size": 3,15  "expand_ratio": 2.0,16  "hidden_act": "silu",17  "hidden_dropout_prob": 0.05,18  "hidden_sizes": [19    64,20    80,21    9622  ],23  "id2label": {24    "0": "background",25    "1": "aeroplane",26    "2": "bicycle",27    "3": "bird",28    "4": "boat",29    "5": "bottle",30    "6": "bus",31    "7": "car",32    "8": "cat",33    "9": "chair",34    "10": "cow",35    "11": "diningtable",36    "12": "dog",37    "13": "horse",38    "14": "motorbike",39    "15": "person",40    "16": "pottedplant",41    "17": "sheep",42    "18": "sofa",43    "19": "train",44    "20": "tvmonitor"45  },46  "image_size": 512,47  "initializer_range": 0.02,48  "label2id": {49    "aeroplane": 1,50    "background": 0,51    "bicycle": 2,52    "bird": 3,53    "boat": 4,54    "bottle": 5,55    "bus": 6,56    "car": 7,57    "cat": 8,58    "chair": 9,59    "cow": 10,60    "diningtable": 11,61    "dog": 12,62    "horse": 13,63    "motorbike": 14,64    "person": 15,65    "pottedplant": 16,66    "sheep": 17,67    "sofa": 18,68    "train": 19,69    "tvmonitor": 2070  },71  "layer_norm_eps": 1e-05,72  "mlp_ratio": 2.0,73  "model_type": "mobilevit",74  "neck_hidden_sizes": [75    16,76    16,77    24,78    48,79    64,80    80,81    32082  ],83  "num_attention_heads": 4,84  "num_channels": 3,85  "output_stride": 16,86  "patch_size": 2,87  "qkv_bias": true,88  "semantic_loss_ignore_index": 255,89  "torch_dtype": "float32",90  "transformers_version": "4.20.0.dev0"91}92