amaye15 commited on
Commit
a0eb65a
1 Parent(s): 78ceb40

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