|
{ |
|
"architectures": [ |
|
"MobileViTv2ForSemanticSegmentation" |
|
], |
|
"aspp_dropout_prob": 0.1, |
|
"aspp_out_channels": 512, |
|
"atrous_rates": [ |
|
6, |
|
12, |
|
18 |
|
], |
|
"attn_dropout": 0.0, |
|
"classifier_dropout_prob": 0.1, |
|
"conv_kernel_size": 3, |
|
"expand_ratio": 2.0, |
|
"ffn_dropout": 0.0, |
|
"hidden_act": "swish", |
|
"id2label": { |
|
"0": "background", |
|
"1": "aeroplane", |
|
"2": "bicycle", |
|
"3": "bird", |
|
"4": "boat", |
|
"5": "bottle", |
|
"6": "bus", |
|
"7": "car", |
|
"8": "cat", |
|
"9": "chair", |
|
"10": "cow", |
|
"11": "diningtable", |
|
"12": "dog", |
|
"13": "horse", |
|
"14": "motorbike", |
|
"15": "person", |
|
"16": "pottedplant", |
|
"17": "sheep", |
|
"18": "sofa", |
|
"19": "train", |
|
"20": "tvmonitor" |
|
}, |
|
"image_size": 512, |
|
"initializer_range": 0.02, |
|
"label2id": { |
|
"aeroplane": 1, |
|
"background": 0, |
|
"bicycle": 2, |
|
"bird": 3, |
|
"boat": 4, |
|
"bottle": 5, |
|
"bus": 6, |
|
"car": 7, |
|
"cat": 8, |
|
"chair": 9, |
|
"cow": 10, |
|
"diningtable": 11, |
|
"dog": 12, |
|
"horse": 13, |
|
"motorbike": 14, |
|
"person": 15, |
|
"pottedplant": 16, |
|
"sheep": 17, |
|
"sofa": 18, |
|
"train": 19, |
|
"tvmonitor": 20 |
|
}, |
|
"layer_norm_eps": 1e-05, |
|
"mlp_ratio": 2.0, |
|
"model_type": "mobilevitv2", |
|
"num_channels": 3, |
|
"output_stride": 16, |
|
"patch_size": 2, |
|
"semantic_loss_ignore_index": 255, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.29.0.dev0", |
|
"width_multiplier": 1.5 |
|
} |
|
|