leafmed-models / model-v2 /config.json
RippedKek's picture
Upload 6 files
d03fc57 verified
{
"_name_or_path": "/kaggle/working/medicinal_plants_image_detection/checkpoint-4460",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Seethapala",
"1": "Coriender",
"2": "Arali",
"3": "Raktachandini",
"4": "Gauva",
"5": "Neem",
"6": "Pepper",
"7": "Nooni",
"8": "Jasmine",
"9": "Ganike",
"10": "Brahmi",
"11": "Geranium",
"12": "Hibiscus",
"13": "Guava",
"14": "Tulsi",
"15": "Pomegranate",
"16": "Rose",
"17": "Tamarind",
"18": "Mint",
"19": "Insulin",
"20": "Basale",
"21": "Wood_sorel",
"22": "Nithyapushpa",
"23": "Bamboo",
"24": "Jackfruit",
"25": "Palak(Spinach)",
"26": "Ashoka",
"27": "Pappaya",
"28": "Lemon_grass",
"29": "Betel",
"30": "Avacado",
"31": "Betel_Nut",
"32": "Amruta_Balli",
"33": "Bhrami",
"34": "Henna",
"35": "Castor",
"36": "Catharanthus",
"37": "Amla",
"38": "Curry_Leaf",
"39": "Nagadali",
"40": "Aloevera",
"41": "Mango",
"42": "Papaya",
"43": "Lemon",
"44": "Doddpathre",
"45": "Doddapatre",
"46": "Honge",
"47": "Sapota",
"48": "Ekka",
"49": "Curry",
"50": "Ashwagandha",
"51": "Tulasi"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Aloevera": 40,
"Amla": 37,
"Amruta_Balli": 32,
"Arali": 2,
"Ashoka": 26,
"Ashwagandha": 50,
"Avacado": 30,
"Bamboo": 23,
"Basale": 20,
"Betel": 29,
"Betel_Nut": 31,
"Bhrami": 33,
"Brahmi": 10,
"Castor": 35,
"Catharanthus": 36,
"Coriender": 1,
"Curry": 49,
"Curry_Leaf": 38,
"Doddapatre": 45,
"Doddpathre": 44,
"Ekka": 48,
"Ganike": 9,
"Gauva": 4,
"Geranium": 11,
"Guava": 13,
"Henna": 34,
"Hibiscus": 12,
"Honge": 46,
"Insulin": 19,
"Jackfruit": 24,
"Jasmine": 8,
"Lemon": 43,
"Lemon_grass": 28,
"Mango": 41,
"Mint": 18,
"Nagadali": 39,
"Neem": 5,
"Nithyapushpa": 22,
"Nooni": 7,
"Palak(Spinach)": 25,
"Papaya": 42,
"Pappaya": 27,
"Pepper": 6,
"Pomegranate": 15,
"Raktachandini": 3,
"Rose": 16,
"Sapota": 47,
"Seethapala": 0,
"Tamarind": 17,
"Tulasi": 51,
"Tulsi": 14,
"Wood_sorel": 21
},
"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.36.2"
}