Hugging Face's logo

valentinocc
/
dog-breed-classifier

Image Classification
Transformers
Safetensors
PyTorch
mobilenet_v2
dog-breed
mobilenet
computer-vision
Model card Files
xet
Community
3
valentinocc commited on
Commit
9bb4e93
·
verified ·
1 Parent(s): 7295e34

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +263 -252
config.json CHANGED
@@ -1,253 +1,264 @@
1
- {
2
- "model_type": "mobilenet_v2",
3
- "num_classes": 120,
4
- "image_size": [
5
- 224,
6
- 224
7
- ],
8
- "architecture": "mobilenet_v2_custom",
9
- "id2label": {
10
- "0": "Chihuahua",
11
- "1": "Japanese Spaniel",
12
- "2": "Maltese Dog",
13
- "3": "Pekinese",
14
- "4": "Shih-Tzu",
15
- "5": "Blenheim Spaniel",
16
- "6": "Papillon",
17
- "7": "Toy Terrier",
18
- "8": "Rhodesian Ridgeback",
19
- "9": "Afghan Hound",
20
- "10": "Basset",
21
- "11": "Beagle",
22
- "12": "Bloodhound",
23
- "13": "Bluetick",
24
- "14": "Black-And-Tan Coonhound",
25
- "15": "Walker Hound",
26
- "16": "English Foxhound",
27
- "17": "Redbone",
28
- "18": "Borzoi",
29
- "19": "Irish Wolfhound",
30
- "20": "Italian Greyhound",
31
- "21": "Whippet",
32
- "22": "Ibizan Hound",
33
- "23": "Norwegian Elkhound",
34
- "24": "Otterhound",
35
- "25": "Saluki",
36
- "26": "Scottish Deerhound",
37
- "27": "Weimaraner",
38
- "28": "Staffordshire Bullterrier",
39
- "29": "American Staffordshire Terrier",
40
- "30": "Bedlington Terrier",
41
- "31": "Border Terrier",
42
- "32": "Kerry Blue Terrier",
43
- "33": "Irish Terrier",
44
- "34": "Norfolk Terrier",
45
- "35": "Norwich Terrier",
46
- "36": "Yorkshire Terrier",
47
- "37": "Wire-Haired Fox Terrier",
48
- "38": "Lakeland Terrier",
49
- "39": "Sealyham Terrier",
50
- "40": "Airedale",
51
- "41": "Cairn",
52
- "42": "Australian Terrier",
53
- "43": "Dandie Dinmont",
54
- "44": "Boston Bull",
55
- "45": "Miniature Schnauzer",
56
- "46": "Giant Schnauzer",
57
- "47": "Standard Schnauzer",
58
- "48": "Scotch Terrier",
59
- "49": "Tibetan Terrier",
60
- "50": "Silky Terrier",
61
- "51": "Soft-Coated Wheaten Terrier",
62
- "52": "West Highland White Terrier",
63
- "53": "Lhasa",
64
- "54": "Flat-Coated Retriever",
65
- "55": "Curly-Coated Retriever",
66
- "56": "Golden Retriever",
67
- "57": "Labrador Retriever",
68
- "58": "Chesapeake Bay Retriever",
69
- "59": "German Short-Haired Pointer",
70
- "60": "Vizsla",
71
- "61": "English Setter",
72
- "62": "Irish Setter",
73
- "63": "Gordon Setter",
74
- "64": "Brittany Spaniel",
75
- "65": "Clumber",
76
- "66": "English Springer",
77
- "67": "Welsh Springer Spaniel",
78
- "68": "Cocker Spaniel",
79
- "69": "Sussex Spaniel",
80
- "70": "Irish Water Spaniel",
81
- "71": "Kuvasz",
82
- "72": "Schipperke",
83
- "73": "Groenendael",
84
- "74": "Malinois",
85
- "75": "Briard",
86
- "76": "Kelpie",
87
- "77": "Komondor",
88
- "78": "Old English Sheepdog",
89
- "79": "Shetland Sheepdog",
90
- "80": "Collie",
91
- "81": "Border Collie",
92
- "82": "Bouvier Des Flandres",
93
- "83": "Rottweiler",
94
- "84": "German Shepherd",
95
- "85": "Doberman",
96
- "86": "Miniature Pinscher",
97
- "87": "Greater Swiss Mountain Dog",
98
- "88": "Bernese Mountain Dog",
99
- "89": "Appenzeller",
100
- "90": "Entlebucher",
101
- "91": "Boxer",
102
- "92": "Bull Mastiff",
103
- "93": "Tibetan Mastiff",
104
- "94": "French Bulldog",
105
- "95": "Great Dane",
106
- "96": "Saint Bernard",
107
- "97": "Eskimo Dog",
108
- "98": "Malamute",
109
- "99": "Siberian Husky",
110
- "100": "Affenpinscher",
111
- "101": "Basenji",
112
- "102": "Pug",
113
- "103": "Leonberg",
114
- "104": "Newfoundland",
115
- "105": "Great Pyrenees",
116
- "106": "Samoyed",
117
- "107": "Pomeranian",
118
- "108": "Chow",
119
- "109": "Keeshond",
120
- "110": "Brabancon Griffon",
121
- "111": "Pembroke",
122
- "112": "Cardigan",
123
- "113": "Toy Poodle",
124
- "114": "Miniature Poodle",
125
- "115": "Standard Poodle",
126
- "116": "Mexican Hairless",
127
- "117": "Dingo",
128
- "118": "Dhole",
129
- "119": "African Hunting Dog"
130
- },
131
- "label2id": {
132
- "Chihuahua": "0",
133
- "Japanese Spaniel": "1",
134
- "Maltese Dog": "2",
135
- "Pekinese": "3",
136
- "Shih-Tzu": "4",
137
- "Blenheim Spaniel": "5",
138
- "Papillon": "6",
139
- "Toy Terrier": "7",
140
- "Rhodesian Ridgeback": "8",
141
- "Afghan Hound": "9",
142
- "Basset": "10",
143
- "Beagle": "11",
144
- "Bloodhound": "12",
145
- "Bluetick": "13",
146
- "Black-And-Tan Coonhound": "14",
147
- "Walker Hound": "15",
148
- "English Foxhound": "16",
149
- "Redbone": "17",
150
- "Borzoi": "18",
151
- "Irish Wolfhound": "19",
152
- "Italian Greyhound": "20",
153
- "Whippet": "21",
154
- "Ibizan Hound": "22",
155
- "Norwegian Elkhound": "23",
156
- "Otterhound": "24",
157
- "Saluki": "25",
158
- "Scottish Deerhound": "26",
159
- "Weimaraner": "27",
160
- "Staffordshire Bullterrier": "28",
161
- "American Staffordshire Terrier": "29",
162
- "Bedlington Terrier": "30",
163
- "Border Terrier": "31",
164
- "Kerry Blue Terrier": "32",
165
- "Irish Terrier": "33",
166
- "Norfolk Terrier": "34",
167
- "Norwich Terrier": "35",
168
- "Yorkshire Terrier": "36",
169
- "Wire-Haired Fox Terrier": "37",
170
- "Lakeland Terrier": "38",
171
- "Sealyham Terrier": "39",
172
- "Airedale": "40",
173
- "Cairn": "41",
174
- "Australian Terrier": "42",
175
- "Dandie Dinmont": "43",
176
- "Boston Bull": "44",
177
- "Miniature Schnauzer": "45",
178
- "Giant Schnauzer": "46",
179
- "Standard Schnauzer": "47",
180
- "Scotch Terrier": "48",
181
- "Tibetan Terrier": "49",
182
- "Silky Terrier": "50",
183
- "Soft-Coated Wheaten Terrier": "51",
184
- "West Highland White Terrier": "52",
185
- "Lhasa": "53",
186
- "Flat-Coated Retriever": "54",
187
- "Curly-Coated Retriever": "55",
188
- "Golden Retriever": "56",
189
- "Labrador Retriever": "57",
190
- "Chesapeake Bay Retriever": "58",
191
- "German Short-Haired Pointer": "59",
192
- "Vizsla": "60",
193
- "English Setter": "61",
194
- "Irish Setter": "62",
195
- "Gordon Setter": "63",
196
- "Brittany Spaniel": "64",
197
- "Clumber": "65",
198
- "English Springer": "66",
199
- "Welsh Springer Spaniel": "67",
200
- "Cocker Spaniel": "68",
201
- "Sussex Spaniel": "69",
202
- "Irish Water Spaniel": "70",
203
- "Kuvasz": "71",
204
- "Schipperke": "72",
205
- "Groenendael": "73",
206
- "Malinois": "74",
207
- "Briard": "75",
208
- "Kelpie": "76",
209
- "Komondor": "77",
210
- "Old English Sheepdog": "78",
211
- "Shetland Sheepdog": "79",
212
- "Collie": "80",
213
- "Border Collie": "81",
214
- "Bouvier Des Flandres": "82",
215
- "Rottweiler": "83",
216
- "German Shepherd": "84",
217
- "Doberman": "85",
218
- "Miniature Pinscher": "86",
219
- "Greater Swiss Mountain Dog": "87",
220
- "Bernese Mountain Dog": "88",
221
- "Appenzeller": "89",
222
- "Entlebucher": "90",
223
- "Boxer": "91",
224
- "Bull Mastiff": "92",
225
- "Tibetan Mastiff": "93",
226
- "French Bulldog": "94",
227
- "Great Dane": "95",
228
- "Saint Bernard": "96",
229
- "Eskimo Dog": "97",
230
- "Malamute": "98",
231
- "Siberian Husky": "99",
232
- "Affenpinscher": "100",
233
- "Basenji": "101",
234
- "Pug": "102",
235
- "Leonberg": "103",
236
- "Newfoundland": "104",
237
- "Great Pyrenees": "105",
238
- "Samoyed": "106",
239
- "Pomeranian": "107",
240
- "Chow": "108",
241
- "Keeshond": "109",
242
- "Brabancon Griffon": "110",
243
- "Pembroke": "111",
244
- "Cardigan": "112",
245
- "Toy Poodle": "113",
246
- "Miniature Poodle": "114",
247
- "Standard Poodle": "115",
248
- "Mexican Hairless": "116",
249
- "Dingo": "117",
250
- "Dhole": "118",
251
- "African Hunting Dog": "119"
252
- }
 
 
 
 
 
 
 
 
 
 
 
253
  }
 
1
+ {
2
+ "_name_or_path": "valentinocc/dog-breed-classifier",
3
+ "architectures": [
4
+ "MobileNetV2ForImageClassification"
5
+ ],
6
+ "classifier_dropout": 0.2,
7
+ "hidden_act": "relu6",
8
+ "hidden_size": 1280,
9
+ "image_size": 224,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_eps": 1e-05,
12
+ "model_type": "mobilenet_v2",
13
+ "num_channels": 3,
14
+ "num_labels": 120,
15
+ "output_stride": 32,
16
+ "problem_type": "single_label_classification",
17
+ "torch_dtype": "float32",
18
+ "transformers_version": "4.21.0",
19
+ "width_mult": 1.0,
20
+ "id2label": {
21
+ "0": "Chihuahua",
22
+ "1": "Japanese Spaniel",
23
+ "2": "Maltese Dog",
24
+ "3": "Pekinese",
25
+ "4": "Shih-Tzu",
26
+ "5": "Blenheim Spaniel",
27
+ "6": "Papillon",
28
+ "7": "Toy Terrier",
29
+ "8": "Rhodesian Ridgeback",
30
+ "9": "Afghan Hound",
31
+ "10": "Basset",
32
+ "11": "Beagle",
33
+ "12": "Bloodhound",
34
+ "13": "Bluetick",
35
+ "14": "Black-And-Tan Coonhound",
36
+ "15": "Walker Hound",
37
+ "16": "English Foxhound",
38
+ "17": "Redbone",
39
+ "18": "Borzoi",
40
+ "19": "Irish Wolfhound",
41
+ "20": "Italian Greyhound",
42
+ "21": "Whippet",
43
+ "22": "Ibizan Hound",
44
+ "23": "Norwegian Elkhound",
45
+ "24": "Otterhound",
46
+ "25": "Saluki",
47
+ "26": "Scottish Deerhound",
48
+ "27": "Weimaraner",
49
+ "28": "Staffordshire Bullterrier",
50
+ "29": "American Staffordshire Terrier",
51
+ "30": "Bedlington Terrier",
52
+ "31": "Border Terrier",
53
+ "32": "Kerry Blue Terrier",
54
+ "33": "Irish Terrier",
55
+ "34": "Norfolk Terrier",
56
+ "35": "Norwich Terrier",
57
+ "36": "Yorkshire Terrier",
58
+ "37": "Wire-Haired Fox Terrier",
59
+ "38": "Lakeland Terrier",
60
+ "39": "Sealyham Terrier",
61
+ "40": "Airedale",
62
+ "41": "Cairn",
63
+ "42": "Australian Terrier",
64
+ "43": "Dandie Dinmont",
65
+ "44": "Boston Bull",
66
+ "45": "Miniature Schnauzer",
67
+ "46": "Giant Schnauzer",
68
+ "47": "Standard Schnauzer",
69
+ "48": "Scotch Terrier",
70
+ "49": "Tibetan Terrier",
71
+ "50": "Silky Terrier",
72
+ "51": "Soft-Coated Wheaten Terrier",
73
+ "52": "West Highland White Terrier",
74
+ "53": "Lhasa",
75
+ "54": "Flat-Coated Retriever",
76
+ "55": "Curly-Coated Retriever",
77
+ "56": "Golden Retriever",
78
+ "57": "Labrador Retriever",
79
+ "58": "Chesapeake Bay Retriever",
80
+ "59": "German Short-Haired Pointer",
81
+ "60": "Vizsla",
82
+ "61": "English Setter",
83
+ "62": "Irish Setter",
84
+ "63": "Gordon Setter",
85
+ "64": "Brittany Spaniel",
86
+ "65": "Clumber",
87
+ "66": "English Springer",
88
+ "67": "Welsh Springer Spaniel",
89
+ "68": "Cocker Spaniel",
90
+ "69": "Sussex Spaniel",
91
+ "70": "Irish Water Spaniel",
92
+ "71": "Kuvasz",
93
+ "72": "Schipperke",
94
+ "73": "Groenendael",
95
+ "74": "Malinois",
96
+ "75": "Briard",
97
+ "76": "Kelpie",
98
+ "77": "Komondor",
99
+ "78": "Old English Sheepdog",
100
+ "79": "Shetland Sheepdog",
101
+ "80": "Collie",
102
+ "81": "Border Collie",
103
+ "82": "Bouvier Des Flandres",
104
+ "83": "Rottweiler",
105
+ "84": "German Shepherd",
106
+ "85": "Doberman",
107
+ "86": "Miniature Pinscher",
108
+ "87": "Greater Swiss Mountain Dog",
109
+ "88": "Bernese Mountain Dog",
110
+ "89": "Appenzeller",
111
+ "90": "Entlebucher",
112
+ "91": "Boxer",
113
+ "92": "Bull Mastiff",
114
+ "93": "Tibetan Mastiff",
115
+ "94": "French Bulldog",
116
+ "95": "Great Dane",
117
+ "96": "Saint Bernard",
118
+ "97": "Eskimo Dog",
119
+ "98": "Malamute",
120
+ "99": "Siberian Husky",
121
+ "100": "Affenpinscher",
122
+ "101": "Basenji",
123
+ "102": "Pug",
124
+ "103": "Leonberg",
125
+ "104": "Newfoundland",
126
+ "105": "Great Pyrenees",
127
+ "106": "Samoyed",
128
+ "107": "Pomeranian",
129
+ "108": "Chow",
130
+ "109": "Keeshond",
131
+ "110": "Brabancon Griffon",
132
+ "111": "Pembroke",
133
+ "112": "Cardigan",
134
+ "113": "Toy Poodle",
135
+ "114": "Miniature Poodle",
136
+ "115": "Standard Poodle",
137
+ "116": "Mexican Hairless",
138
+ "117": "Dingo",
139
+ "118": "Dhole",
140
+ "119": "African Hunting Dog"
141
+ },
142
+ "label2id": {
143
+ "Chihuahua": 0,
144
+ "Japanese Spaniel": 1,
145
+ "Maltese Dog": 2,
146
+ "Pekinese": 3,
147
+ "Shih-Tzu": 4,
148
+ "Blenheim Spaniel": 5,
149
+ "Papillon": 6,
150
+ "Toy Terrier": 7,
151
+ "Rhodesian Ridgeback": 8,
152
+ "Afghan Hound": 9,
153
+ "Basset": 10,
154
+ "Beagle": 11,
155
+ "Bloodhound": 12,
156
+ "Bluetick": 13,
157
+ "Black-And-Tan Coonhound": 14,
158
+ "Walker Hound": 15,
159
+ "English Foxhound": 16,
160
+ "Redbone": 17,
161
+ "Borzoi": 18,
162
+ "Irish Wolfhound": 19,
163
+ "Italian Greyhound": 20,
164
+ "Whippet": 21,
165
+ "Ibizan Hound": 22,
166
+ "Norwegian Elkhound": 23,
167
+ "Otterhound": 24,
168
+ "Saluki": 25,
169
+ "Scottish Deerhound": 26,
170
+ "Weimaraner": 27,
171
+ "Staffordshire Bullterrier": 28,
172
+ "American Staffordshire Terrier": 29,
173
+ "Bedlington Terrier": 30,
174
+ "Border Terrier": 31,
175
+ "Kerry Blue Terrier": 32,
176
+ "Irish Terrier": 33,
177
+ "Norfolk Terrier": 34,
178
+ "Norwich Terrier": 35,
179
+ "Yorkshire Terrier": 36,
180
+ "Wire-Haired Fox Terrier": 37,
181
+ "Lakeland Terrier": 38,
182
+ "Sealyham Terrier": 39,
183
+ "Airedale": 40,
184
+ "Cairn": 41,
185
+ "Australian Terrier": 42,
186
+ "Dandie Dinmont": 43,
187
+ "Boston Bull": 44,
188
+ "Miniature Schnauzer": 45,
189
+ "Giant Schnauzer": 46,
190
+ "Standard Schnauzer": 47,
191
+ "Scotch Terrier": 48,
192
+ "Tibetan Terrier": 49,
193
+ "Silky Terrier": 50,
194
+ "Soft-Coated Wheaten Terrier": 51,
195
+ "West Highland White Terrier": 52,
196
+ "Lhasa": 53,
197
+ "Flat-Coated Retriever": 54,
198
+ "Curly-Coated Retriever": 55,
199
+ "Golden Retriever": 56,
200
+ "Labrador Retriever": 57,
201
+ "Chesapeake Bay Retriever": 58,
202
+ "German Short-Haired Pointer": 59,
203
+ "Vizsla": 60,
204
+ "English Setter": 61,
205
+ "Irish Setter": 62,
206
+ "Gordon Setter": 63,
207
+ "Brittany Spaniel": 64,
208
+ "Clumber": 65,
209
+ "English Springer": 66,
210
+ "Welsh Springer Spaniel": 67,
211
+ "Cocker Spaniel": 68,
212
+ "Sussex Spaniel": 69,
213
+ "Irish Water Spaniel": 70,
214
+ "Kuvasz": 71,
215
+ "Schipperke": 72,
216
+ "Groenendael": 73,
217
+ "Malinois": 74,
218
+ "Briard": 75,
219
+ "Kelpie": 76,
220
+ "Komondor": 77,
221
+ "Old English Sheepdog": 78,
222
+ "Shetland Sheepdog": 79,
223
+ "Collie": 80,
224
+ "Border Collie": 81,
225
+ "Bouvier Des Flandres": 82,
226
+ "Rottweiler": 83,
227
+ "German Shepherd": 84,
228
+ "Doberman": 85,
229
+ "Miniature Pinscher": 86,
230
+ "Greater Swiss Mountain Dog": 87,
231
+ "Bernese Mountain Dog": 88,
232
+ "Appenzeller": 89,
233
+ "Entlebucher": 90,
234
+ "Boxer": 91,
235
+ "Bull Mastiff": 92,
236
+ "Tibetan Mastiff": 93,
237
+ "French Bulldog": 94,
238
+ "Great Dane": 95,
239
+ "Saint Bernard": 96,
240
+ "Eskimo Dog": 97,
241
+ "Malamute": 98,
242
+ "Siberian Husky": 99,
243
+ "Affenpinscher": 100,
244
+ "Basenji": 101,
245
+ "Pug": 102,
246
+ "Leonberg": 103,
247
+ "Newfoundland": 104,
248
+ "Great Pyrenees": 105,
249
+ "Samoyed": 106,
250
+ "Pomeranian": 107,
251
+ "Chow": 108,
252
+ "Keeshond": 109,
253
+ "Brabancon Griffon": 110,
254
+ "Pembroke": 111,
255
+ "Cardigan": 112,
256
+ "Toy Poodle": 113,
257
+ "Miniature Poodle": 114,
258
+ "Standard Poodle": 115,
259
+ "Mexican Hairless": 116,
260
+ "Dingo": 117,
261
+ "Dhole": 118,
262
+ "African Hunting Dog": 119
263
+ }
264
  }