|
{ |
|
"_name_or_path": "google/vit-base-patch16-224-in21k", |
|
"architectures": [ |
|
"ViTForImageClassification" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"encoder_stride": 16, |
|
"finetuning_task": "image-classification", |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_size": 768, |
|
"id2label": { |
|
"0": "acne_scars", |
|
"1": "acne_closed_comedones", |
|
"2": "acne_level_2", |
|
"3": "melanoma", |
|
"4": "age_changes", |
|
"5": "dry_skin", |
|
"6": "hidradenitis_suppurativa", |
|
"7": "acne_excoriee", |
|
"8": "acne_level_0", |
|
"9": "acne_level_1", |
|
"10": "normal_skin", |
|
"11": "acne_mechanica", |
|
"12": "rosacea", |
|
"13": "inflammation", |
|
"14": "acne_level_3", |
|
"15": "acne", |
|
"16": "acne_histology", |
|
"17": "acne_open_comedones", |
|
"18": "acne_infantile", |
|
"19": "acne_comedones", |
|
"20": "acne_pustules", |
|
"21": "pigmentation", |
|
"22": "acne_cystic", |
|
"23": "milia", |
|
"24": "oily_skin", |
|
"25": "burn_skin" |
|
}, |
|
"image_size": 224, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"label2id": { |
|
"acne": 15, |
|
"acne_closed_comedones": 1, |
|
"acne_comedones": 19, |
|
"acne_cystic": 22, |
|
"acne_excoriee": 7, |
|
"acne_histology": 16, |
|
"acne_infantile": 18, |
|
"acne_level_0": 8, |
|
"acne_level_1": 9, |
|
"acne_level_2": 2, |
|
"acne_level_3": 14, |
|
"acne_mechanica": 11, |
|
"acne_open_comedones": 17, |
|
"acne_pustules": 20, |
|
"acne_scars": 0, |
|
"age_changes": 4, |
|
"burn_skin": 25, |
|
"dry_skin": 5, |
|
"hidradenitis_suppurativa": 6, |
|
"inflammation": 13, |
|
"melanoma": 3, |
|
"milia": 23, |
|
"normal_skin": 10, |
|
"oily_skin": 24, |
|
"pigmentation": 21, |
|
"rosacea": 12 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"model_type": "vit", |
|
"num_attention_heads": 12, |
|
"num_channels": 3, |
|
"num_hidden_layers": 12, |
|
"patch_size": 16, |
|
"problem_type": "single_label_classification", |
|
"qkv_bias": true, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.38.2" |
|
} |
|
|