config.json
2.6 KB · 131 lines · json Raw
1 {
2 "_name_or_path": "valentinafeve/yolos-fashionpedia",
3 "architectures": [
4 "YolosForObjectDetection"
5 ],
6 "attention_probs_dropout_prob": 0.0,
7 "auxiliary_loss": false,
8 "bbox_cost": 5,
9 "bbox_loss_coefficient": 5,
10 "class_cost": 1,
11 "eos_coefficient": 0.1,
12 "giou_cost": 2,
13 "giou_loss_coefficient": 2,
14 "hidden_act": "gelu",
15 "hidden_dropout_prob": 0.0,
16 "hidden_size": 384,
17 "id2label": {
18 "0": "shirt, blouse",
19 "1": "top, t-shirt, sweatshirt",
20 "2": "sweater",
21 "3": "cardigan",
22 "4": "jacket",
23 "5": "vest",
24 "6": "pants",
25 "7": "shorts",
26 "8": "skirt",
27 "9": "coat",
28 "10": "dress",
29 "11": "jumpsuit",
30 "12": "cape",
31 "13": "glasses",
32 "14": "hat",
33 "15": "headband, head covering, hair accessory",
34 "16": "tie",
35 "17": "glove",
36 "18": "watch",
37 "19": "belt",
38 "20": "leg warmer",
39 "21": "tights, stockings",
40 "22": "sock",
41 "23": "shoe",
42 "24": "bag, wallet",
43 "25": "scarf",
44 "26": "umbrella",
45 "27": "hood",
46 "28": "collar",
47 "29": "lapel",
48 "30": "epaulette",
49 "31": "sleeve",
50 "32": "pocket",
51 "33": "neckline",
52 "34": "buckle",
53 "35": "zipper",
54 "36": "applique",
55 "37": "bead",
56 "38": "bow",
57 "39": "flower",
58 "40": "fringe",
59 "41": "ribbon",
60 "42": "rivet",
61 "43": "ruffle",
62 "44": "sequin",
63 "45": "tassel"
64 },
65 "image_size": [
66 512,
67 864
68 ],
69 "initializer_range": 0.02,
70 "intermediate_size": 1536,
71 "label2id": {
72 "applique": 36,
73 "bag, wallet": 24,
74 "bead": 37,
75 "belt": 19,
76 "bow": 38,
77 "buckle": 34,
78 "cape": 12,
79 "cardigan": 3,
80 "coat": 9,
81 "collar": 28,
82 "dress": 10,
83 "epaulette": 30,
84 "flower": 39,
85 "fringe": 40,
86 "glasses": 13,
87 "glove": 17,
88 "hat": 14,
89 "headband, head covering, hair accessory": 15,
90 "hood": 27,
91 "jacket": 4,
92 "jumpsuit": 11,
93 "lapel": 29,
94 "leg warmer": 20,
95 "neckline": 33,
96 "pants": 6,
97 "pocket": 32,
98 "ribbon": 41,
99 "rivet": 42,
100 "ruffle": 43,
101 "scarf": 25,
102 "sequin": 44,
103 "shirt, blouse": 0,
104 "shoe": 23,
105 "shorts": 7,
106 "skirt": 8,
107 "sleeve": 31,
108 "sock": 22,
109 "sweater": 2,
110 "tassel": 45,
111 "tie": 16,
112 "tights, stockings": 21,
113 "top, t-shirt, sweatshirt": 1,
114 "umbrella": 26,
115 "vest": 5,
116 "watch": 18,
117 "zipper": 35
118 },
119 "layer_norm_eps": 1e-12,
120 "model_type": "yolos",
121 "num_attention_heads": 6,
122 "num_channels": 3,
123 "num_detection_tokens": 100,
124 "num_hidden_layers": 12,
125 "patch_size": 16,
126 "qkv_bias": true,
127 "torch_dtype": "float32",
128 "transformers_version": "4.23.1",
129 "use_mid_position_embeddings": true
130 }
131