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
f2a330d
·
verified ·
1 Parent(s): 1344788

Upload 2 files

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