Hugging Face's logo

bhumong
/
fruit-classifier-efficientnet-b0

Image Classification
PyTorch
English
efficientnet
torchvision
fruits
fruits-360
transfer-learning
neptune-ai
Model card Files
xet
Community
1
bhumong commited on
Commit
1b56e54
·
verified ·
1 Parent(s): 1edfd97

Update config.json with correct class labels

Browse files
Files changed (1) hide show
  1. config.json +163 -152
config.json CHANGED
@@ -2,161 +2,172 @@
2
  "model_type": "efficientnet",
3
  "num_labels": 76,
4
  "id2label": {
5
- "0": "LABEL_0",
6
- "1": "LABEL_1",
7
- "2": "LABEL_2",
8
- "3": "LABEL_3",
9
- "4": "LABEL_4",
10
- "5": "LABEL_5",
11
- "6": "LABEL_6",
12
- "7": "LABEL_7",
13
- "8": "LABEL_8",
14
- "9": "LABEL_9",
15
- "10": "LABEL_10",
16
- "11": "LABEL_11",
17
- "12": "LABEL_12",
18
- "13": "LABEL_13",
19
- "14": "LABEL_14",
20
- "15": "LABEL_15",
21
- "16": "LABEL_16",
22
- "17": "LABEL_17",
23
- "18": "LABEL_18",
24
- "19": "LABEL_19",
25
- "20": "LABEL_20",
26
- "21": "LABEL_21",
27
- "22": "LABEL_22",
28
- "23": "LABEL_23",
29
- "24": "LABEL_24",
30
- "25": "LABEL_25",
31
- "26": "LABEL_26",
32
- "27": "LABEL_27",
33
- "28": "LABEL_28",
34
- "29": "LABEL_29",
35
- "30": "LABEL_30",
36
- "31": "LABEL_31",
37
- "32": "LABEL_32",
38
- "33": "LABEL_33",
39
- "34": "LABEL_34",
40
- "35": "LABEL_35",
41
- "36": "LABEL_36",
42
- "37": "LABEL_37",
43
- "38": "LABEL_38",
44
- "39": "LABEL_39",
45
- "40": "LABEL_40",
46
- "41": "LABEL_41",
47
- "42": "LABEL_42",
48
- "43": "LABEL_43",
49
- "44": "LABEL_44",
50
- "45": "LABEL_45",
51
- "46": "LABEL_46",
52
- "47": "LABEL_47",
53
- "48": "LABEL_48",
54
- "49": "LABEL_49",
55
- "50": "LABEL_50",
56
- "51": "LABEL_51",
57
- "52": "LABEL_52",
58
- "53": "LABEL_53",
59
- "54": "LABEL_54",
60
- "55": "LABEL_55",
61
- "56": "LABEL_56",
62
- "57": "LABEL_57",
63
- "58": "LABEL_58",
64
- "59": "LABEL_59",
65
- "60": "LABEL_60",
66
- "61": "LABEL_61",
67
- "62": "LABEL_62",
68
- "63": "LABEL_63",
69
- "64": "LABEL_64",
70
- "65": "LABEL_65",
71
- "66": "LABEL_66",
72
- "67": "LABEL_67",
73
- "68": "LABEL_68",
74
- "69": "LABEL_69",
75
- "70": "LABEL_70",
76
- "71": "LABEL_71",
77
- "72": "LABEL_72",
78
- "73": "LABEL_73",
79
- "74": "LABEL_74",
80
- "75": "LABEL_75"
81
  },
82
  "label2id": {
83
- "LABEL_0": "0",
84
- "LABEL_1": "1",
85
- "LABEL_2": "2",
86
- "LABEL_3": "3",
87
- "LABEL_4": "4",
88
- "LABEL_5": "5",
89
- "LABEL_6": "6",
90
- "LABEL_7": "7",
91
- "LABEL_8": "8",
92
- "LABEL_9": "9",
93
- "LABEL_10": "10",
94
- "LABEL_11": "11",
95
- "LABEL_12": "12",
96
- "LABEL_13": "13",
97
- "LABEL_14": "14",
98
- "LABEL_15": "15",
99
- "LABEL_16": "16",
100
- "LABEL_17": "17",
101
- "LABEL_18": "18",
102
- "LABEL_19": "19",
103
- "LABEL_20": "20",
104
- "LABEL_21": "21",
105
- "LABEL_22": "22",
106
- "LABEL_23": "23",
107
- "LABEL_24": "24",
108
- "LABEL_25": "25",
109
- "LABEL_26": "26",
110
- "LABEL_27": "27",
111
- "LABEL_28": "28",
112
- "LABEL_29": "29",
113
- "LABEL_30": "30",
114
- "LABEL_31": "31",
115
- "LABEL_32": "32",
116
- "LABEL_33": "33",
117
- "LABEL_34": "34",
118
- "LABEL_35": "35",
119
- "LABEL_36": "36",
120
- "LABEL_37": "37",
121
- "LABEL_38": "38",
122
- "LABEL_39": "39",
123
- "LABEL_40": "40",
124
- "LABEL_41": "41",
125
- "LABEL_42": "42",
126
- "LABEL_43": "43",
127
- "LABEL_44": "44",
128
- "LABEL_45": "45",
129
- "LABEL_46": "46",
130
- "LABEL_47": "47",
131
- "LABEL_48": "48",
132
- "LABEL_49": "49",
133
- "LABEL_50": "50",
134
- "LABEL_51": "51",
135
- "LABEL_52": "52",
136
- "LABEL_53": "53",
137
- "LABEL_54": "54",
138
- "LABEL_55": "55",
139
- "LABEL_56": "56",
140
- "LABEL_57": "57",
141
- "LABEL_58": "58",
142
- "LABEL_59": "59",
143
- "LABEL_60": "60",
144
- "LABEL_61": "61",
145
- "LABEL_62": "62",
146
- "LABEL_63": "63",
147
- "LABEL_64": "64",
148
- "LABEL_65": "65",
149
- "LABEL_66": "66",
150
- "LABEL_67": "67",
151
- "LABEL_68": "68",
152
- "LABEL_69": "69",
153
- "LABEL_70": "70",
154
- "LABEL_71": "71",
155
- "LABEL_72": "72",
156
- "LABEL_73": "73",
157
- "LABEL_74": "74",
158
- "LABEL_75": "75"
159
  },
 
 
 
 
 
 
 
 
 
 
 
160
  "_custom_architecture": true,
161
  "_model_class_name": "EfficientNet",
162
  "_pytorch_version": "2.6.0+cu124"
 
2
  "model_type": "efficientnet",
3
  "num_labels": 76,
4
  "id2label": {
5
+ "0": "Apple",
6
+ "1": "Apricot",
7
+ "2": "Avocado",
8
+ "3": "Banana",
9
+ "4": "Beans",
10
+ "5": "Beetroot",
11
+ "6": "Blackberrie",
12
+ "7": "Blueberry",
13
+ "8": "Cabbage",
14
+ "9": "Cactus",
15
+ "10": "Caju",
16
+ "11": "Cantaloupe",
17
+ "12": "Carambula",
18
+ "13": "Carrot",
19
+ "14": "Cauliflower",
20
+ "15": "Cherimoya",
21
+ "16": "Cherry",
22
+ "17": "Chestnut",
23
+ "18": "Clementine",
24
+ "19": "Cocos",
25
+ "20": "Corn",
26
+ "21": "Cucumber",
27
+ "22": "Dates",
28
+ "23": "Eggplant",
29
+ "24": "Fig",
30
+ "25": "Ginger",
31
+ "26": "Gooseberry",
32
+ "27": "Granadilla",
33
+ "28": "Grape",
34
+ "29": "Grapefruit",
35
+ "30": "Guava",
36
+ "31": "Hazelnut",
37
+ "32": "Huckleberry",
38
+ "33": "Kaki",
39
+ "34": "Kiwi",
40
+ "35": "Kohlrabi",
41
+ "36": "Kumquats",
42
+ "37": "Lemon",
43
+ "38": "Limes",
44
+ "39": "Lychee",
45
+ "40": "Mandarine",
46
+ "41": "Mango",
47
+ "42": "Mangostan",
48
+ "43": "Maracuja",
49
+ "44": "Melon",
50
+ "45": "Mulberry",
51
+ "46": "Nectarine",
52
+ "47": "Nut",
53
+ "48": "Onion",
54
+ "49": "Orange",
55
+ "50": "Papaya",
56
+ "51": "Passion",
57
+ "52": "Peach",
58
+ "53": "Pear",
59
+ "54": "Pepino",
60
+ "55": "Pepper",
61
+ "56": "Physalis",
62
+ "57": "Pineapple",
63
+ "58": "Pistachio",
64
+ "59": "Pitahaya",
65
+ "60": "Plum",
66
+ "61": "Pomegranate",
67
+ "62": "Pomelo",
68
+ "63": "Potato",
69
+ "64": "Quince",
70
+ "65": "Rambutan",
71
+ "66": "Raspberry",
72
+ "67": "Redcurrant",
73
+ "68": "Salak",
74
+ "69": "Strawberry",
75
+ "70": "Tamarillo",
76
+ "71": "Tangelo",
77
+ "72": "Tomato",
78
+ "73": "Walnut",
79
+ "74": "Watermelon",
80
+ "75": "Zucchini"
81
  },
82
  "label2id": {
83
+ "Apple": "0",
84
+ "Apricot": "1",
85
+ "Avocado": "2",
86
+ "Banana": "3",
87
+ "Beans": "4",
88
+ "Beetroot": "5",
89
+ "Blackberrie": "6",
90
+ "Blueberry": "7",
91
+ "Cabbage": "8",
92
+ "Cactus": "9",
93
+ "Caju": "10",
94
+ "Cantaloupe": "11",
95
+ "Carambula": "12",
96
+ "Carrot": "13",
97
+ "Cauliflower": "14",
98
+ "Cherimoya": "15",
99
+ "Cherry": "16",
100
+ "Chestnut": "17",
101
+ "Clementine": "18",
102
+ "Cocos": "19",
103
+ "Corn": "20",
104
+ "Cucumber": "21",
105
+ "Dates": "22",
106
+ "Eggplant": "23",
107
+ "Fig": "24",
108
+ "Ginger": "25",
109
+ "Gooseberry": "26",
110
+ "Granadilla": "27",
111
+ "Grape": "28",
112
+ "Grapefruit": "29",
113
+ "Guava": "30",
114
+ "Hazelnut": "31",
115
+ "Huckleberry": "32",
116
+ "Kaki": "33",
117
+ "Kiwi": "34",
118
+ "Kohlrabi": "35",
119
+ "Kumquats": "36",
120
+ "Lemon": "37",
121
+ "Limes": "38",
122
+ "Lychee": "39",
123
+ "Mandarine": "40",
124
+ "Mango": "41",
125
+ "Mangostan": "42",
126
+ "Maracuja": "43",
127
+ "Melon": "44",
128
+ "Mulberry": "45",
129
+ "Nectarine": "46",
130
+ "Nut": "47",
131
+ "Onion": "48",
132
+ "Orange": "49",
133
+ "Papaya": "50",
134
+ "Passion": "51",
135
+ "Peach": "52",
136
+ "Pear": "53",
137
+ "Pepino": "54",
138
+ "Pepper": "55",
139
+ "Physalis": "56",
140
+ "Pineapple": "57",
141
+ "Pistachio": "58",
142
+ "Pitahaya": "59",
143
+ "Plum": "60",
144
+ "Pomegranate": "61",
145
+ "Pomelo": "62",
146
+ "Potato": "63",
147
+ "Quince": "64",
148
+ "Rambutan": "65",
149
+ "Raspberry": "66",
150
+ "Redcurrant": "67",
151
+ "Salak": "68",
152
+ "Strawberry": "69",
153
+ "Tamarillo": "70",
154
+ "Tangelo": "71",
155
+ "Tomato": "72",
156
+ "Walnut": "73",
157
+ "Watermelon": "74",
158
+ "Zucchini": "75"
159
  },
160
+ "image_size": 100,
161
+ "normalization_mean": [
162
+ 0.485,
163
+ 0.456,
164
+ 0.406
165
+ ],
166
+ "normalization_std": [
167
+ 0.229,
168
+ 0.224,
169
+ 0.225
170
+ ],
171
  "_custom_architecture": true,
172
  "_model_class_name": "EfficientNet",
173
  "_pytorch_version": "2.6.0+cu124"