Hugging Face's logo

mohsin416
/
autocatalogai-clip-multitask

Image Classification
Transformers
clip_multitask_classifier
clip
fashion
computer-vision
multi-task-learning
product-attribute-extraction
autocatalogai
Model card Files
xet
 Community
mohsin416 commited on
Commit
dee55d3
·
verified ·
1 Parent(s): a0c4806

Upload folder using huggingface_hub

Browse files
Files changed (6) hide show
  1. README.md +64 -0
  2. config.json +33 -0
  3. history.json +127 -0
  4. label_maps.json +556 -0
  5. metrics.json +56 -0
  6. model.pt +3 -0
README.md ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ pipeline_tag: image-classification
4
+ tags:
5
+ - clip
6
+ - fashion
7
+ - computer-vision
8
+ - multi-task-learning
9
+ - product-attribute-extraction
10
+ - autocatalogai
11
+ ---
12
+
13
+ # AutoCatalogAI CLIP Multi-Task Classifier
14
+
15
+ AutoCatalogAI is a fashion product attribute extraction model.
16
+
17
+ It predicts:
18
+
19
+ ['gender', 'masterCategory', 'subCategory', 'articleType', 'baseColour', 'season', 'usage']
20
+
21
+ ## Dataset
22
+
23
+ Dataset: `ashraq/fashion-product-images-small`
24
+
25
+ Split:
26
+
27
+ - Train: 70%
28
+ - Validation: 15%
29
+ - Test: 15%
30
+
31
+ ## Base Model
32
+
33
+ `openai/clip-vit-base-patch32`
34
+
35
+ ## Architecture
36
+
37
+ CLIP image encoder + multiple classification heads.
38
+
39
+ ## Overall Test Metrics
40
+
41
+ ```json
42
+ {
43
+ "average_accuracy": 0.8335026038852995,
44
+ "average_macro_f1": 0.6568447666058456,
45
+ "average_weighted_f1": 0.8421526081109854,
46
+ "average_top3_accuracy": 0.9711087581303888,
47
+ "exact_match_accuracy": 0.27938284677053393,
48
+ "avg_inference_time_ms_per_image": 1.5960319117829298,
49
+ "p95_inference_time_ms_per_image": 1.4948196709156036,
50
+ "test_samples": 6611
51
+ }
52
+ ```
53
+
54
+ ## Important
55
+
56
+ This model should be loaded with the AutoCatalogAI project code because it contains custom multi-task classifier heads.
57
+
58
+ Expected files:
59
+
60
+ - `model.pt`
61
+ - `config.json`
62
+ - `label_maps.json`
63
+ - `metrics.json`
64
+ - `README.md`
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project_name": "AutoCatalogAI",
3
+ "dataset_name": "ashraq/fashion-product-images-small",
4
+ "base_model_name": "openai/clip-vit-base-patch32",
5
+ "model_type": "clip_multitask_classifier",
6
+ "tasks": [
7
+ "gender",
8
+ "masterCategory",
9
+ "subCategory",
10
+ "articleType",
11
+ "baseColour",
12
+ "season",
13
+ "usage"
14
+ ],
15
+ "task_num_classes": {
16
+ "gender": 5,
17
+ "masterCategory": 7,
18
+ "subCategory": 45,
19
+ "articleType": 141,
20
+ "baseColour": 46,
21
+ "season": 4,
22
+ "usage": 8
23
+ },
24
+ "hidden_dim": 512,
25
+ "dropout": 0.2,
26
+ "unfreeze_last_n_vision_layers": 2,
27
+ "image_size": 224,
28
+ "train_ratio": 0.7,
29
+ "validation_ratio": 0.15,
30
+ "test_ratio": 0.15,
31
+ "best_epoch": 4,
32
+ "best_val_loss": 4.497780916771451
33
+ }
history.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "epoch": 1,
4
+ "train_loss": 9.432052525584561,
5
+ "val_loss": 5.63392734642766,
6
+ "train_acc": {
7
+ "gender": 0.771729274611399,
8
+ "masterCategory": 0.9251295336787565,
9
+ "subCategory": 0.7585492227979275,
10
+ "articleType": 0.6095531088082902,
11
+ "baseColour": 0.35518134715025906,
12
+ "season": 0.5852007772020725,
13
+ "usage": 0.7349417098445595
14
+ },
15
+ "val_acc": {
16
+ "gender": 0.8417794940249174,
17
+ "masterCategory": 0.9820350241545893,
18
+ "subCategory": 0.9260265700483091,
19
+ "articleType": 0.7716755657259089,
20
+ "baseColour": 0.5580981439105009,
21
+ "season": 0.6431397787948132,
22
+ "usage": 0.7978006610729723
23
+ },
24
+ "avg_train_acc": 0.6771835677276091,
25
+ "avg_val_acc": 0.7886507482474301
26
+ },
27
+ {
28
+ "epoch": 2,
29
+ "train_loss": 4.5827552318573,
30
+ "val_loss": 4.7799444958783575,
31
+ "train_acc": {
32
+ "gender": 0.8623704663212435,
33
+ "masterCategory": 0.9847150259067358,
34
+ "subCategory": 0.9139572538860103,
35
+ "articleType": 0.7956930051813471,
36
+ "baseColour": 0.555181347150259,
37
+ "season": 0.6761010362694301,
38
+ "usage": 0.8134391191709844
39
+ },
40
+ "val_acc": {
41
+ "gender": 0.8757468853292651,
42
+ "masterCategory": 0.9769021739130435,
43
+ "subCategory": 0.9274329392321383,
44
+ "articleType": 0.8141050088990592,
45
+ "baseColour": 0.5533069539791508,
46
+ "season": 0.6885329265191965,
47
+ "usage": 0.8391653953724891
48
+ },
49
+ "avg_train_acc": 0.8002081791265729,
50
+ "avg_val_acc": 0.8107417547491919
51
+ },
52
+ {
53
+ "epoch": 3,
54
+ "train_loss": 3.370936806337821,
55
+ "val_loss": 4.622166495392288,
56
+ "train_acc": {
57
+ "gender": 0.8881800518134715,
58
+ "masterCategory": 0.9912564766839378,
59
+ "subCategory": 0.9370142487046632,
60
+ "articleType": 0.8315414507772021,
61
+ "baseColour": 0.6044689119170984,
62
+ "season": 0.7088082901554404,
63
+ "usage": 0.8428432642487047
64
+ },
65
+ "val_acc": {
66
+ "gender": 0.8746424485125858,
67
+ "masterCategory": 0.9938103864734299,
68
+ "subCategory": 0.9453979150775489,
69
+ "articleType": 0.8184830282227308,
70
+ "baseColour": 0.599002034070684,
71
+ "season": 0.717820366132723,
72
+ "usage": 0.8351369819476226
73
+ },
74
+ "avg_train_acc": 0.8291589563286454,
75
+ "avg_val_acc": 0.8263275943481894
76
+ },
77
+ {
78
+ "epoch": 4,
79
+ "train_loss": 2.6022005854492978,
80
+ "val_loss": 4.497780916771451,
81
+ "train_acc": {
82
+ "gender": 0.9083225388601036,
83
+ "masterCategory": 0.9950453367875648,
84
+ "subCategory": 0.9501619170984456,
85
+ "articleType": 0.8564443005181347,
86
+ "baseColour": 0.6356217616580311,
87
+ "season": 0.741062176165803,
88
+ "usage": 0.867389896373057
89
+ },
90
+ "val_acc": {
91
+ "gender": 0.884455250444953,
92
+ "masterCategory": 0.9941123188405797,
93
+ "subCategory": 0.947813374014747,
94
+ "articleType": 0.8485252987541317,
95
+ "baseColour": 0.6131928553267226,
96
+ "season": 0.7170178616831935,
97
+ "usage": 0.8717740910246631
98
+ },
99
+ "avg_train_acc": 0.8505782753515915,
100
+ "avg_val_acc": 0.8395558642984272
101
+ },
102
+ {
103
+ "epoch": 5,
104
+ "train_loss": 2.2021738547117597,
105
+ "val_loss": 4.508289864097816,
106
+ "train_acc": {
107
+ "gender": 0.9151554404145078,
108
+ "masterCategory": 0.9965673575129533,
109
+ "subCategory": 0.9587435233160622,
110
+ "articleType": 0.8702720207253886,
111
+ "baseColour": 0.6543069948186528,
112
+ "season": 0.760071243523316,
113
+ "usage": 0.8798575129533679
114
+ },
115
+ "val_acc": {
116
+ "gender": 0.8894371345029239,
117
+ "masterCategory": 0.9932065217391305,
118
+ "subCategory": 0.9475114416475973,
119
+ "articleType": 0.8495820620391559,
120
+ "baseColour": 0.6156559877955758,
121
+ "season": 0.7180746249682176,
122
+ "usage": 0.874690122044241
123
+ },
124
+ "avg_train_acc": 0.862139156180607,
125
+ "avg_val_acc": 0.8411654135338343
126
+ }
127
+ ]
label_maps.json ADDED
@@ -0,0 +1,556 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "gender": {
3
+ "label2id": {
4
+ "Boys": 0,
5
+ "Girls": 1,
6
+ "Men": 2,
7
+ "Unisex": 3,
8
+ "Women": 4
9
+ },
10
+ "id2label": {
11
+ "0": "Boys",
12
+ "1": "Girls",
13
+ "2": "Men",
14
+ "3": "Unisex",
15
+ "4": "Women"
16
+ }
17
+ },
18
+ "masterCategory": {
19
+ "label2id": {
20
+ "Accessories": 0,
21
+ "Apparel": 1,
22
+ "Footwear": 2,
23
+ "Free Items": 3,
24
+ "Home": 4,
25
+ "Personal Care": 5,
26
+ "Sporting Goods": 6
27
+ },
28
+ "id2label": {
29
+ "0": "Accessories",
30
+ "1": "Apparel",
31
+ "2": "Footwear",
32
+ "3": "Free Items",
33
+ "4": "Home",
34
+ "5": "Personal Care",
35
+ "6": "Sporting Goods"
36
+ }
37
+ },
38
+ "subCategory": {
39
+ "label2id": {
40
+ "Accessories": 0,
41
+ "Apparel Set": 1,
42
+ "Bags": 2,
43
+ "Bath and Body": 3,
44
+ "Beauty Accessories": 4,
45
+ "Belts": 5,
46
+ "Bottomwear": 6,
47
+ "Cufflinks": 7,
48
+ "Dress": 8,
49
+ "Eyes": 9,
50
+ "Eyewear": 10,
51
+ "Flip Flops": 11,
52
+ "Fragrance": 12,
53
+ "Free Gifts": 13,
54
+ "Gloves": 14,
55
+ "Hair": 15,
56
+ "Headwear": 16,
57
+ "Home Furnishing": 17,
58
+ "Innerwear": 18,
59
+ "Jewellery": 19,
60
+ "Lips": 20,
61
+ "Loungewear and Nightwear": 21,
62
+ "Makeup": 22,
63
+ "Mufflers": 23,
64
+ "Nails": 24,
65
+ "Perfumes": 25,
66
+ "Sandal": 26,
67
+ "Saree": 27,
68
+ "Scarves": 28,
69
+ "Shoe Accessories": 29,
70
+ "Shoes": 30,
71
+ "Skin": 31,
72
+ "Skin Care": 32,
73
+ "Socks": 33,
74
+ "Sports Accessories": 34,
75
+ "Sports Equipment": 35,
76
+ "Stoles": 36,
77
+ "Ties": 37,
78
+ "Topwear": 38,
79
+ "Umbrellas": 39,
80
+ "Vouchers": 40,
81
+ "Wallets": 41,
82
+ "Watches": 42,
83
+ "Water Bottle": 43,
84
+ "Wristbands": 44
85
+ },
86
+ "id2label": {
87
+ "0": "Accessories",
88
+ "1": "Apparel Set",
89
+ "2": "Bags",
90
+ "3": "Bath and Body",
91
+ "4": "Beauty Accessories",
92
+ "5": "Belts",
93
+ "6": "Bottomwear",
94
+ "7": "Cufflinks",
95
+ "8": "Dress",
96
+ "9": "Eyes",
97
+ "10": "Eyewear",
98
+ "11": "Flip Flops",
99
+ "12": "Fragrance",
100
+ "13": "Free Gifts",
101
+ "14": "Gloves",
102
+ "15": "Hair",
103
+ "16": "Headwear",
104
+ "17": "Home Furnishing",
105
+ "18": "Innerwear",
106
+ "19": "Jewellery",
107
+ "20": "Lips",
108
+ "21": "Loungewear and Nightwear",
109
+ "22": "Makeup",
110
+ "23": "Mufflers",
111
+ "24": "Nails",
112
+ "25": "Perfumes",
113
+ "26": "Sandal",
114
+ "27": "Saree",
115
+ "28": "Scarves",
116
+ "29": "Shoe Accessories",
117
+ "30": "Shoes",
118
+ "31": "Skin",
119
+ "32": "Skin Care",
120
+ "33": "Socks",
121
+ "34": "Sports Accessories",
122
+ "35": "Sports Equipment",
123
+ "36": "Stoles",
124
+ "37": "Ties",
125
+ "38": "Topwear",
126
+ "39": "Umbrellas",
127
+ "40": "Vouchers",
128
+ "41": "Wallets",
129
+ "42": "Watches",
130
+ "43": "Water Bottle",
131
+ "44": "Wristbands"
132
+ }
133
+ },
134
+ "articleType": {
135
+ "label2id": {
136
+ "Accessory Gift Set": 0,
137
+ "Baby Dolls": 1,
138
+ "Backpacks": 2,
139
+ "Bangle": 3,
140
+ "Basketballs": 4,
141
+ "Bath Robe": 5,
142
+ "Beauty Accessory": 6,
143
+ "Belts": 7,
144
+ "Blazers": 8,
145
+ "Body Lotion": 9,
146
+ "Body Wash and Scrub": 10,
147
+ "Booties": 11,
148
+ "Boxers": 12,
149
+ "Bra": 13,
150
+ "Bracelet": 14,
151
+ "Briefs": 15,
152
+ "Camisoles": 16,
153
+ "Capris": 17,
154
+ "Caps": 18,
155
+ "Casual Shoes": 19,
156
+ "Churidar": 20,
157
+ "Clothing Set": 21,
158
+ "Clutches": 22,
159
+ "Compact": 23,
160
+ "Concealer": 24,
161
+ "Cufflinks": 25,
162
+ "Cushion Covers": 26,
163
+ "Deodorant": 27,
164
+ "Dresses": 28,
165
+ "Duffel Bag": 29,
166
+ "Dupatta": 30,
167
+ "Earrings": 31,
168
+ "Eye Cream": 32,
169
+ "Eyeshadow": 33,
170
+ "Face Moisturisers": 34,
171
+ "Face Scrub and Exfoliator": 35,
172
+ "Face Serum and Gel": 36,
173
+ "Face Wash and Cleanser": 37,
174
+ "Flats": 38,
175
+ "Flip Flops": 39,
176
+ "Footballs": 40,
177
+ "Formal Shoes": 41,
178
+ "Foundation and Primer": 42,
179
+ "Fragrance Gift Set": 43,
180
+ "Free Gifts": 44,
181
+ "Gloves": 45,
182
+ "Hair Accessory": 46,
183
+ "Hair Colour": 47,
184
+ "Handbags": 48,
185
+ "Hat": 49,
186
+ "Headband": 50,
187
+ "Heels": 51,
188
+ "Highlighter and Blush": 52,
189
+ "Innerwear Vests": 53,
190
+ "Ipad": 54,
191
+ "Jackets": 55,
192
+ "Jeans": 56,
193
+ "Jeggings": 57,
194
+ "Jewellery Set": 58,
195
+ "Jumpsuit": 59,
196
+ "Kajal and Eyeliner": 60,
197
+ "Key chain": 61,
198
+ "Kurta Sets": 62,
199
+ "Kurtas": 63,
200
+ "Kurtis": 64,
201
+ "Laptop Bag": 65,
202
+ "Leggings": 66,
203
+ "Lehenga Choli": 67,
204
+ "Lip Care": 68,
205
+ "Lip Gloss": 69,
206
+ "Lip Liner": 70,
207
+ "Lip Plumper": 71,
208
+ "Lipstick": 72,
209
+ "Lounge Pants": 73,
210
+ "Lounge Shorts": 74,
211
+ "Lounge Tshirts": 75,
212
+ "Makeup Remover": 76,
213
+ "Mascara": 77,
214
+ "Mask and Peel": 78,
215
+ "Mens Grooming Kit": 79,
216
+ "Messenger Bag": 80,
217
+ "Mobile Pouch": 81,
218
+ "Mufflers": 82,
219
+ "Nail Essentials": 83,
220
+ "Nail Polish": 84,
221
+ "Necklace and Chains": 85,
222
+ "Nehru Jackets": 86,
223
+ "Night suits": 87,
224
+ "Nightdress": 88,
225
+ "Patiala": 89,
226
+ "Pendant": 90,
227
+ "Perfume and Body Mist": 91,
228
+ "Rain Jacket": 92,
229
+ "Ring": 93,
230
+ "Robe": 94,
231
+ "Rompers": 95,
232
+ "Rucksacks": 96,
233
+ "Salwar": 97,
234
+ "Salwar and Dupatta": 98,
235
+ "Sandals": 99,
236
+ "Sarees": 100,
237
+ "Scarves": 101,
238
+ "Shapewear": 102,
239
+ "Shirts": 103,
240
+ "Shoe Accessories": 104,
241
+ "Shoe Laces": 105,
242
+ "Shorts": 106,
243
+ "Shrug": 107,
244
+ "Skirts": 108,
245
+ "Socks": 109,
246
+ "Sports Sandals": 110,
247
+ "Sports Shoes": 111,
248
+ "Stockings": 112,
249
+ "Stoles": 113,
250
+ "Sunglasses": 114,
251
+ "Sunscreen": 115,
252
+ "Suspenders": 116,
253
+ "Sweaters": 117,
254
+ "Sweatshirts": 118,
255
+ "Swimwear": 119,
256
+ "Tablet Sleeve": 120,
257
+ "Ties": 121,
258
+ "Ties and Cufflinks": 122,
259
+ "Tights": 123,
260
+ "Toner": 124,
261
+ "Tops": 125,
262
+ "Track Pants": 126,
263
+ "Tracksuits": 127,
264
+ "Travel Accessory": 128,
265
+ "Trolley Bag": 129,
266
+ "Trousers": 130,
267
+ "Trunk": 131,
268
+ "Tshirts": 132,
269
+ "Tunics": 133,
270
+ "Umbrellas": 134,
271
+ "Waist Pouch": 135,
272
+ "Waistcoat": 136,
273
+ "Wallets": 137,
274
+ "Watches": 138,
275
+ "Water Bottle": 139,
276
+ "Wristbands": 140
277
+ },
278
+ "id2label": {
279
+ "0": "Accessory Gift Set",
280
+ "1": "Baby Dolls",
281
+ "2": "Backpacks",
282
+ "3": "Bangle",
283
+ "4": "Basketballs",
284
+ "5": "Bath Robe",
285
+ "6": "Beauty Accessory",
286
+ "7": "Belts",
287
+ "8": "Blazers",
288
+ "9": "Body Lotion",
289
+ "10": "Body Wash and Scrub",
290
+ "11": "Booties",
291
+ "12": "Boxers",
292
+ "13": "Bra",
293
+ "14": "Bracelet",
294
+ "15": "Briefs",
295
+ "16": "Camisoles",
296
+ "17": "Capris",
297
+ "18": "Caps",
298
+ "19": "Casual Shoes",
299
+ "20": "Churidar",
300
+ "21": "Clothing Set",
301
+ "22": "Clutches",
302
+ "23": "Compact",
303
+ "24": "Concealer",
304
+ "25": "Cufflinks",
305
+ "26": "Cushion Covers",
306
+ "27": "Deodorant",
307
+ "28": "Dresses",
308
+ "29": "Duffel Bag",
309
+ "30": "Dupatta",
310
+ "31": "Earrings",
311
+ "32": "Eye Cream",
312
+ "33": "Eyeshadow",
313
+ "34": "Face Moisturisers",
314
+ "35": "Face Scrub and Exfoliator",
315
+ "36": "Face Serum and Gel",
316
+ "37": "Face Wash and Cleanser",
317
+ "38": "Flats",
318
+ "39": "Flip Flops",
319
+ "40": "Footballs",
320
+ "41": "Formal Shoes",
321
+ "42": "Foundation and Primer",
322
+ "43": "Fragrance Gift Set",
323
+ "44": "Free Gifts",
324
+ "45": "Gloves",
325
+ "46": "Hair Accessory",
326
+ "47": "Hair Colour",
327
+ "48": "Handbags",
328
+ "49": "Hat",
329
+ "50": "Headband",
330
+ "51": "Heels",
331
+ "52": "Highlighter and Blush",
332
+ "53": "Innerwear Vests",
333
+ "54": "Ipad",
334
+ "55": "Jackets",
335
+ "56": "Jeans",
336
+ "57": "Jeggings",
337
+ "58": "Jewellery Set",
338
+ "59": "Jumpsuit",
339
+ "60": "Kajal and Eyeliner",
340
+ "61": "Key chain",
341
+ "62": "Kurta Sets",
342
+ "63": "Kurtas",
343
+ "64": "Kurtis",
344
+ "65": "Laptop Bag",
345
+ "66": "Leggings",
346
+ "67": "Lehenga Choli",
347
+ "68": "Lip Care",
348
+ "69": "Lip Gloss",
349
+ "70": "Lip Liner",
350
+ "71": "Lip Plumper",
351
+ "72": "Lipstick",
352
+ "73": "Lounge Pants",
353
+ "74": "Lounge Shorts",
354
+ "75": "Lounge Tshirts",
355
+ "76": "Makeup Remover",
356
+ "77": "Mascara",
357
+ "78": "Mask and Peel",
358
+ "79": "Mens Grooming Kit",
359
+ "80": "Messenger Bag",
360
+ "81": "Mobile Pouch",
361
+ "82": "Mufflers",
362
+ "83": "Nail Essentials",
363
+ "84": "Nail Polish",
364
+ "85": "Necklace and Chains",
365
+ "86": "Nehru Jackets",
366
+ "87": "Night suits",
367
+ "88": "Nightdress",
368
+ "89": "Patiala",
369
+ "90": "Pendant",
370
+ "91": "Perfume and Body Mist",
371
+ "92": "Rain Jacket",
372
+ "93": "Ring",
373
+ "94": "Robe",
374
+ "95": "Rompers",
375
+ "96": "Rucksacks",
376
+ "97": "Salwar",
377
+ "98": "Salwar and Dupatta",
378
+ "99": "Sandals",
379
+ "100": "Sarees",
380
+ "101": "Scarves",
381
+ "102": "Shapewear",
382
+ "103": "Shirts",
383
+ "104": "Shoe Accessories",
384
+ "105": "Shoe Laces",
385
+ "106": "Shorts",
386
+ "107": "Shrug",
387
+ "108": "Skirts",
388
+ "109": "Socks",
389
+ "110": "Sports Sandals",
390
+ "111": "Sports Shoes",
391
+ "112": "Stockings",
392
+ "113": "Stoles",
393
+ "114": "Sunglasses",
394
+ "115": "Sunscreen",
395
+ "116": "Suspenders",
396
+ "117": "Sweaters",
397
+ "118": "Sweatshirts",
398
+ "119": "Swimwear",
399
+ "120": "Tablet Sleeve",
400
+ "121": "Ties",
401
+ "122": "Ties and Cufflinks",
402
+ "123": "Tights",
403
+ "124": "Toner",
404
+ "125": "Tops",
405
+ "126": "Track Pants",
406
+ "127": "Tracksuits",
407
+ "128": "Travel Accessory",
408
+ "129": "Trolley Bag",
409
+ "130": "Trousers",
410
+ "131": "Trunk",
411
+ "132": "Tshirts",
412
+ "133": "Tunics",
413
+ "134": "Umbrellas",
414
+ "135": "Waist Pouch",
415
+ "136": "Waistcoat",
416
+ "137": "Wallets",
417
+ "138": "Watches",
418
+ "139": "Water Bottle",
419
+ "140": "Wristbands"
420
+ }
421
+ },
422
+ "baseColour": {
423
+ "label2id": {
424
+ "Beige": 0,
425
+ "Black": 1,
426
+ "Blue": 2,
427
+ "Bronze": 3,
428
+ "Brown": 4,
429
+ "Burgundy": 5,
430
+ "Charcoal": 6,
431
+ "Coffee Brown": 7,
432
+ "Copper": 8,
433
+ "Cream": 9,
434
+ "Fluorescent Green": 10,
435
+ "Gold": 11,
436
+ "Green": 12,
437
+ "Grey": 13,
438
+ "Grey Melange": 14,
439
+ "Khaki": 15,
440
+ "Lavender": 16,
441
+ "Lime Green": 17,
442
+ "Magenta": 18,
443
+ "Maroon": 19,
444
+ "Mauve": 20,
445
+ "Metallic": 21,
446
+ "Multi": 22,
447
+ "Mushroom Brown": 23,
448
+ "Mustard": 24,
449
+ "Navy Blue": 25,
450
+ "Nude": 26,
451
+ "Off White": 27,
452
+ "Olive": 28,
453
+ "Orange": 29,
454
+ "Peach": 30,
455
+ "Pink": 31,
456
+ "Purple": 32,
457
+ "Red": 33,
458
+ "Rose": 34,
459
+ "Rust": 35,
460
+ "Sea Green": 36,
461
+ "Silver": 37,
462
+ "Skin": 38,
463
+ "Steel": 39,
464
+ "Tan": 40,
465
+ "Taupe": 41,
466
+ "Teal": 42,
467
+ "Turquoise Blue": 43,
468
+ "White": 44,
469
+ "Yellow": 45
470
+ },
471
+ "id2label": {
472
+ "0": "Beige",
473
+ "1": "Black",
474
+ "2": "Blue",
475
+ "3": "Bronze",
476
+ "4": "Brown",
477
+ "5": "Burgundy",
478
+ "6": "Charcoal",
479
+ "7": "Coffee Brown",
480
+ "8": "Copper",
481
+ "9": "Cream",
482
+ "10": "Fluorescent Green",
483
+ "11": "Gold",
484
+ "12": "Green",
485
+ "13": "Grey",
486
+ "14": "Grey Melange",
487
+ "15": "Khaki",
488
+ "16": "Lavender",
489
+ "17": "Lime Green",
490
+ "18": "Magenta",
491
+ "19": "Maroon",
492
+ "20": "Mauve",
493
+ "21": "Metallic",
494
+ "22": "Multi",
495
+ "23": "Mushroom Brown",
496
+ "24": "Mustard",
497
+ "25": "Navy Blue",
498
+ "26": "Nude",
499
+ "27": "Off White",
500
+ "28": "Olive",
501
+ "29": "Orange",
502
+ "30": "Peach",
503
+ "31": "Pink",
504
+ "32": "Purple",
505
+ "33": "Red",
506
+ "34": "Rose",
507
+ "35": "Rust",
508
+ "36": "Sea Green",
509
+ "37": "Silver",
510
+ "38": "Skin",
511
+ "39": "Steel",
512
+ "40": "Tan",
513
+ "41": "Taupe",
514
+ "42": "Teal",
515
+ "43": "Turquoise Blue",
516
+ "44": "White",
517
+ "45": "Yellow"
518
+ }
519
+ },
520
+ "season": {
521
+ "label2id": {
522
+ "Fall": 0,
523
+ "Spring": 1,
524
+ "Summer": 2,
525
+ "Winter": 3
526
+ },
527
+ "id2label": {
528
+ "0": "Fall",
529
+ "1": "Spring",
530
+ "2": "Summer",
531
+ "3": "Winter"
532
+ }
533
+ },
534
+ "usage": {
535
+ "label2id": {
536
+ "Casual": 0,
537
+ "Ethnic": 1,
538
+ "Formal": 2,
539
+ "Home": 3,
540
+ "Party": 4,
541
+ "Smart Casual": 5,
542
+ "Sports": 6,
543
+ "Travel": 7
544
+ },
545
+ "id2label": {
546
+ "0": "Casual",
547
+ "1": "Ethnic",
548
+ "2": "Formal",
549
+ "3": "Home",
550
+ "4": "Party",
551
+ "5": "Smart Casual",
552
+ "6": "Sports",
553
+ "7": "Travel"
554
+ }
555
+ }
556
+ }
metrics.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "task_metrics": {
3
+ "gender": {
4
+ "accuracy": 0.8886703978218121,
5
+ "macro_f1": 0.7766001480008942,
6
+ "weighted_f1": 0.8983867693730325,
7
+ "top3_accuracy": 0.9956133716533051
8
+ },
9
+ "masterCategory": {
10
+ "accuracy": 0.9933444259567388,
11
+ "macro_f1": 0.8629937982338353,
12
+ "weighted_f1": 0.9931059873925967,
13
+ "top3_accuracy": 0.999394947814249
14
+ },
15
+ "subCategory": {
16
+ "accuracy": 0.9429738314929663,
17
+ "macro_f1": 0.7108939726417577,
18
+ "weighted_f1": 0.9457045590919284,
19
+ "top3_accuracy": 0.9947057933746786
20
+ },
21
+ "articleType": {
22
+ "accuracy": 0.8410225381939193,
23
+ "macro_f1": 0.6639695724602702,
24
+ "weighted_f1": 0.849652792592164,
25
+ "top3_accuracy": 0.9711087581303888
26
+ },
27
+ "baseColour": {
28
+ "accuracy": 0.6011193465436394,
29
+ "macro_f1": 0.3409578550909414,
30
+ "weighted_f1": 0.6308289193022855,
31
+ "top3_accuracy": 0.8538798971411284
32
+ },
33
+ "season": {
34
+ "accuracy": 0.7070034790500681,
35
+ "macro_f1": 0.7280134716349071,
36
+ "weighted_f1": 0.7072817578439281,
37
+ "top3_accuracy": 0.9848736953562245
38
+ },
39
+ "usage": {
40
+ "accuracy": 0.8603842081379519,
41
+ "macro_f1": 0.5144845481783126,
42
+ "weighted_f1": 0.8701074711809624,
43
+ "top3_accuracy": 0.9981848434427469
44
+ }
45
+ },
46
+ "overall_metrics": {
47
+ "average_accuracy": 0.8335026038852995,
48
+ "average_macro_f1": 0.6568447666058456,
49
+ "average_weighted_f1": 0.8421526081109854,
50
+ "average_top3_accuracy": 0.9711087581303888,
51
+ "exact_match_accuracy": 0.27938284677053393,
52
+ "avg_inference_time_ms_per_image": 1.5960319117829298,
53
+ "p95_inference_time_ms_per_image": 1.4948196709156036,
54
+ "test_samples": 6611
55
+ }
56
+ }
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4c4c9721bcb8ab6db236745bdf5aaf3e2aecd0f3139d2f58f248dc43b7bd26c
3
+ size 613172298