amaye15 commited on
Commit
ed3aa03
1 Parent(s): 1cf9848

Upload ViTForImageClassification

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