abdusahmbzuai commited on
Commit
f5d9c1f
1 Parent(s): cc8c9df

commit files to HF hub

Browse files
README.md ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-classification
4
+ - pytorch
5
+ - huggingpics
6
+ metrics:
7
+ - accuracy
8
+
9
+ model-index:
10
+ - name: CarViT
11
+ results:
12
+ - task:
13
+ name: Image Classification
14
+ type: image-classification
15
+ metrics:
16
+ - name: Accuracy
17
+ type: accuracy
18
+ value: 0.8122026920318604
19
+ ---
20
+
21
+ # CarViT
22
+
23
+
24
+ CarViT - Car make classifier that can identify 40 manufacturers.
25
+
26
+
27
+ ## Example Images
28
+
29
+
30
+ #### Acura
31
+
32
+ ![Acura](images/Acura.jpg)
33
+
34
+ #### Alfa Romeo
35
+
36
+ ![Alfa Romeo](images/Alfa_Romeo.jpg)
37
+
38
+ #### Aston Martin
39
+
40
+ ![Aston Martin](images/Aston_Martin.jpg)
41
+
42
+ #### Audi
43
+
44
+ ![Audi](images/Audi.jpg)
45
+
46
+ #### BMW
47
+
48
+ ![BMW](images/BMW.jpg)
49
+
50
+ #### Bentley
51
+
52
+ ![Bentley](images/Bentley.jpg)
53
+
54
+ #### Buick
55
+
56
+ ![Buick](images/Buick.jpg)
57
+
58
+ #### Cadillac
59
+
60
+ ![Cadillac](images/Cadillac.jpg)
61
+
62
+ #### Chevrolet
63
+
64
+ ![Chevrolet](images/Chevrolet.jpg)
65
+
66
+ #### Chrysler
67
+
68
+ ![Chrysler](images/Chrysler.jpg)
69
+
70
+ #### Dodge
71
+
72
+ ![Dodge](images/Dodge.jpg)
73
+
74
+ #### FIAT
75
+
76
+ ![FIAT](images/FIAT.jpg)
77
+
78
+ #### Ferrari
79
+
80
+ ![Ferrari](images/Ferrari.jpg)
81
+
82
+ #### Ford
83
+
84
+ ![Ford](images/Ford.jpg)
85
+
86
+ #### GMC
87
+
88
+ ![GMC](images/GMC.jpg)
89
+
90
+ #### Genesis
91
+
92
+ ![Genesis](images/Genesis.jpg)
93
+
94
+ #### Honda
95
+
96
+ ![Honda](images/Honda.jpg)
97
+
98
+ #### Hyundai
99
+
100
+ ![Hyundai](images/Hyundai.jpg)
101
+
102
+ #### INFINITI
103
+
104
+ ![INFINITI](images/INFINITI.jpg)
105
+
106
+ #### Jaguar
107
+
108
+ ![Jaguar](images/Jaguar.jpg)
109
+
110
+ #### Jeep
111
+
112
+ ![Jeep](images/Jeep.jpg)
113
+
114
+ #### Kia
115
+
116
+ ![Kia](images/Kia.jpg)
117
+
118
+ #### Lamborghini
119
+
120
+ ![Lamborghini](images/Lamborghini.jpg)
121
+
122
+ #### Land Rover
123
+
124
+ ![Land Rover](images/Land_Rover.jpg)
125
+
126
+ #### Lexus
127
+
128
+ ![Lexus](images/Lexus.jpg)
129
+
130
+ #### Lincoln
131
+
132
+ ![Lincoln](images/Lincoln.jpg)
133
+
134
+ #### MINI
135
+
136
+ ![MINI](images/MINI.jpg)
137
+
138
+ #### Maserati
139
+
140
+ ![Maserati](images/Maserati.jpg)
141
+
142
+ #### Mazda
143
+
144
+ ![Mazda](images/Mazda.jpg)
145
+
146
+ #### McLaren
147
+
148
+ ![McLaren](images/McLaren.jpg)
149
+
150
+ #### Mercedes-Benz
151
+
152
+ ![Mercedes-Benz](images/Mercedes-Benz.jpg)
153
+
154
+ #### Mitsubishi
155
+
156
+ ![Mitsubishi](images/Mitsubishi.jpg)
157
+
158
+ #### Nissan
159
+
160
+ ![Nissan](images/Nissan.jpg)
161
+
162
+ #### Porsche
163
+
164
+ ![Porsche](images/Porsche.jpg)
165
+
166
+ #### Ram
167
+
168
+ ![Ram](images/Ram.jpg)
169
+
170
+ #### Rolls-Royce
171
+
172
+ ![Rolls-Royce](images/Rolls-Royce.jpg)
173
+
174
+ #### Subaru
175
+
176
+ ![Subaru](images/Subaru.jpg)
177
+
178
+ #### Tesla
179
+
180
+ ![Tesla](images/Tesla.jpg)
181
+
182
+ #### Toyota
183
+
184
+ ![Toyota](images/Toyota.jpg)
185
+
186
+ #### Volkswagen
187
+
188
+ ![Volkswagen](images/Volkswagen.jpg)
189
+
190
+ #### Volvo
191
+
192
+ ![Volvo](images/Volvo.jpg)
193
+
194
+ #### smart
195
+
196
+ ![smart](images/smart.jpg)
config.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Acura",
13
+ "1": "Alfa Romeo",
14
+ "10": "Dodge",
15
+ "11": "FIAT",
16
+ "12": "Ferrari",
17
+ "13": "Ford",
18
+ "14": "GMC",
19
+ "15": "Genesis",
20
+ "16": "Honda",
21
+ "17": "Hyundai",
22
+ "18": "INFINITI",
23
+ "19": "Jaguar",
24
+ "2": "Aston Martin",
25
+ "20": "Jeep",
26
+ "21": "Kia",
27
+ "22": "Lamborghini",
28
+ "23": "Land Rover",
29
+ "24": "Lexus",
30
+ "25": "Lincoln",
31
+ "26": "MINI",
32
+ "27": "Maserati",
33
+ "28": "Mazda",
34
+ "29": "McLaren",
35
+ "3": "Audi",
36
+ "30": "Mercedes-Benz",
37
+ "31": "Mitsubishi",
38
+ "32": "Nissan",
39
+ "33": "Porsche",
40
+ "34": "Ram",
41
+ "35": "Rolls-Royce",
42
+ "36": "Subaru",
43
+ "37": "Tesla",
44
+ "38": "Toyota",
45
+ "39": "Volkswagen",
46
+ "4": "BMW",
47
+ "40": "Volvo",
48
+ "41": "smart",
49
+ "5": "Bentley",
50
+ "6": "Buick",
51
+ "7": "Cadillac",
52
+ "8": "Chevrolet",
53
+ "9": "Chrysler"
54
+ },
55
+ "image_size": 224,
56
+ "initializer_range": 0.02,
57
+ "intermediate_size": 3072,
58
+ "label2id": {
59
+ "Acura": "0",
60
+ "Alfa Romeo": "1",
61
+ "Aston Martin": "2",
62
+ "Audi": "3",
63
+ "BMW": "4",
64
+ "Bentley": "5",
65
+ "Buick": "6",
66
+ "Cadillac": "7",
67
+ "Chevrolet": "8",
68
+ "Chrysler": "9",
69
+ "Dodge": "10",
70
+ "FIAT": "11",
71
+ "Ferrari": "12",
72
+ "Ford": "13",
73
+ "GMC": "14",
74
+ "Genesis": "15",
75
+ "Honda": "16",
76
+ "Hyundai": "17",
77
+ "INFINITI": "18",
78
+ "Jaguar": "19",
79
+ "Jeep": "20",
80
+ "Kia": "21",
81
+ "Lamborghini": "22",
82
+ "Land Rover": "23",
83
+ "Lexus": "24",
84
+ "Lincoln": "25",
85
+ "MINI": "26",
86
+ "Maserati": "27",
87
+ "Mazda": "28",
88
+ "McLaren": "29",
89
+ "Mercedes-Benz": "30",
90
+ "Mitsubishi": "31",
91
+ "Nissan": "32",
92
+ "Porsche": "33",
93
+ "Ram": "34",
94
+ "Rolls-Royce": "35",
95
+ "Subaru": "36",
96
+ "Tesla": "37",
97
+ "Toyota": "38",
98
+ "Volkswagen": "39",
99
+ "Volvo": "40",
100
+ "smart": "41"
101
+ },
102
+ "layer_norm_eps": 1e-12,
103
+ "model_type": "vit",
104
+ "num_attention_heads": 12,
105
+ "num_channels": 3,
106
+ "num_hidden_layers": 12,
107
+ "patch_size": 16,
108
+ "problem_type": "single_label_classification",
109
+ "qkv_bias": true,
110
+ "torch_dtype": "float32",
111
+ "transformers_version": "4.17.0"
112
+ }
images/Acura.jpg ADDED
Binary file (16.1 kB)
images/Alfa_Romeo.jpg ADDED
Binary file (7.51 kB)
images/Aston_Martin.jpg ADDED
Binary file (10 kB)
images/Audi.jpg ADDED
Binary file (7.67 kB)
images/BMW.jpg ADDED
Binary file (13.8 kB)
images/Bentley.jpg ADDED
Binary file (12 kB)
images/Buick.jpg ADDED
Binary file (10.7 kB)
images/Cadillac.jpg ADDED
Binary file (12.6 kB)
images/Chevrolet.jpg ADDED
Binary file (10.7 kB)
images/Chrysler.jpg ADDED
Binary file (6.74 kB)
images/Dodge.jpg ADDED
Binary file (11.9 kB)
images/FIAT.jpg ADDED
Binary file (7.45 kB)
images/Ferrari.jpg ADDED
Binary file (14.8 kB)
images/Ford.jpg ADDED
Binary file (5.5 kB)
images/GMC.jpg ADDED
Binary file (10.1 kB)
images/Genesis.jpg ADDED
Binary file (13 kB)
images/Honda.jpg ADDED
Binary file (8.73 kB)
images/Hyundai.jpg ADDED
Binary file (9.35 kB)
images/INFINITI.jpg ADDED
Binary file (9.04 kB)
images/Jaguar.jpg ADDED
Binary file (14 kB)
images/Jeep.jpg ADDED
Binary file (12.4 kB)
images/Kia.jpg ADDED
Binary file (11.5 kB)
images/Lamborghini.jpg ADDED
Binary file (7.71 kB)
images/Land_Rover.jpg ADDED
Binary file (14.4 kB)
images/Lexus.jpg ADDED
Binary file (8.7 kB)
images/Lincoln.jpg ADDED
Binary file (22 kB)
images/MINI.jpg ADDED
Binary file (17 kB)
images/Maserati.jpg ADDED
Binary file (19.7 kB)
images/Mazda.jpg ADDED
Binary file (10.5 kB)
images/McLaren.jpg ADDED
Binary file (14.4 kB)
images/Mercedes-Benz.jpg ADDED
Binary file (10.1 kB)
images/Mitsubishi.jpg ADDED
Binary file (10.9 kB)
images/Nissan.jpg ADDED
Binary file (10 kB)
images/Porsche.jpg ADDED
Binary file (19.4 kB)
images/Ram.jpg ADDED
Binary file (9.9 kB)
images/Rolls-Royce.jpg ADDED
Binary file (20 kB)
images/Subaru.jpg ADDED
Binary file (13.6 kB)
images/Tesla.jpg ADDED
Binary file (8.48 kB)
images/Toyota.jpg ADDED
Binary file (9.98 kB)
images/Volkswagen.jpg ADDED
Binary file (10.3 kB)
images/Volvo.jpg ADDED
Binary file (11.2 kB)
images/smart.jpg ADDED
Binary file (12.2 kB)
preprocessor_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "ViTFeatureExtractor",
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_std": [
11
+ 0.5,
12
+ 0.5,
13
+ 0.5
14
+ ],
15
+ "resample": 2,
16
+ "size": 224
17
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:599560fd9cae61076c8ef22751390736d87a3ce870aa3ad8260e08290af6f6f8
3
+ size 343400945
runs/events.out.tfevents.1648305255.f0cf4d607bb3.81.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1811879a780c0433aeccd77f13e8151b0e0c0f309163c49497ad94dcc8d22f5d
3
+ size 52301