supermarketNRVSMELOC / config.json
nrvtrue's picture
Upload folder using huggingface_hub
506762b verified
raw
history blame
No virus
2.47 kB
{
"_name_or_path": "microsoft/resnet-50",
"_num_labels": 43,
"architectures": [
"ResNetForImageClassification"
],
"depths": [
3,
4,
6,
3
],
"downsample_in_bottleneck": false,
"downsample_in_first_stage": false,
"embedding_size": 64,
"hidden_act": "relu",
"hidden_sizes": [
256,
512,
1024,
2048
],
"id2label": {
"0": "Apple",
"1": "Asparagus",
"2": "Aubergine",
"3": "Avocado",
"4": "Banana",
"5": "Brown-Cap-Mushroom",
"6": "Cabbage",
"7": "Carrots",
"8": "Cucumber",
"9": "Garlic",
"10": "Ginger",
"11": "Juice",
"12": "Kiwi",
"13": "Leek",
"14": "Lemon",
"15": "Lime",
"16": "Mango",
"17": "Melon",
"18": "Milk",
"19": "Nectarine",
"20": "Oat-Milk",
"21": "Oatghurt",
"22": "Onion",
"23": "Orange",
"24": "Papaya",
"25": "Passion-Fruit",
"26": "Peach",
"27": "Pear",
"28": "Pepper",
"29": "Pineapple",
"30": "Plum",
"31": "Pomegranate",
"32": "Potato",
"33": "Red-Beet",
"34": "Red-Grapefruit",
"35": "Satsumas",
"36": "Sour-Cream",
"37": "Sour-Milk",
"38": "Soy-Milk",
"39": "Soyghurt",
"40": "Tomato",
"41": "Yoghurt",
"42": "Zucchini"
},
"label2id": {
"Apple": 0,
"Asparagus": 1,
"Aubergine": 2,
"Avocado": 3,
"Banana": 4,
"Brown-Cap-Mushroom": 5,
"Cabbage": 6,
"Carrots": 7,
"Cucumber": 8,
"Garlic": 9,
"Ginger": 10,
"Juice": 11,
"Kiwi": 12,
"Leek": 13,
"Lemon": 14,
"Lime": 15,
"Mango": 16,
"Melon": 17,
"Milk": 18,
"Nectarine": 19,
"Oat-Milk": 20,
"Oatghurt": 21,
"Onion": 22,
"Orange": 23,
"Papaya": 24,
"Passion-Fruit": 25,
"Peach": 26,
"Pear": 27,
"Pepper": 28,
"Pineapple": 29,
"Plum": 30,
"Pomegranate": 31,
"Potato": 32,
"Red-Beet": 33,
"Red-Grapefruit": 34,
"Satsumas": 35,
"Sour-Cream": 36,
"Sour-Milk": 37,
"Soy-Milk": 38,
"Soyghurt": 39,
"Tomato": 40,
"Yoghurt": 41,
"Zucchini": 42
},
"layer_type": "bottleneck",
"model_type": "resnet",
"num_channels": 3,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"problem_type": "single_label_classification",
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.44.2"
}