dinov2-nivra-finetuned / class_names.json
datdevsteve's picture
Upload 6 files
209490f verified
{
"id2label": {
"0": "atopic_dermatitis",
"1": "basal_cell_carcinoma",
"2": "eczema",
"3": "fungal_infections",
"4": "keratosis",
"5": "melanocytic_nevus",
"6": "melanoma",
"7": "psoriasis",
"8": "viral_warts"
},
"label2id": {
"atopic_dermatitis": 0,
"basal_cell_carcinoma": 1,
"eczema": 2,
"fungal_infections": 3,
"keratosis": 4,
"melanocytic_nevus": 5,
"melanoma": 6,
"psoriasis": 7,
"viral_warts": 8
},
"class_names": [
"atopic_dermatitis",
"basal_cell_carcinoma",
"eczema",
"fungal_infections",
"keratosis",
"melanocytic_nevus",
"melanoma",
"psoriasis",
"viral_warts"
],
"num_labels": 9,
"class_weights": {
"0": 1.0,
"1": 1.0,
"2": 1.0,
"3": 1.0,
"4": 1.0,
"5": 1.0,
"6": 1.0,
"7": 1.0,
"8": 1.0
},
"version": "2.0-kaggle",
"model_name": "microsoft/beit-base-patch16-224-pt22k-ft22k",
"image_size": {
"height": 224,
"width": 224
},
"platform": "kaggle",
"improvements": [
"Medical-safe augmentation",
"Class weights for imbalance",
"Reduced batch size (16)",
"Optimized epochs (20)",
"Dropout (0.1)",
"Early stopping",
"Test-time augmentation support"
]
}