m522t commited on
Commit
fb7bd9e
·
verified ·
1 Parent(s): 85d7f58

Upload open_groundingdino model

Browse files
README.md CHANGED
@@ -1,16 +1,18 @@
1
  ---
2
- library_name: grounding-dino
3
  tags:
4
  - object-detection
5
  - grounding
6
  - vision
7
  - custom-dataset
 
8
  license: mit
 
9
  ---
10
 
11
  # Custom GroundingDINO Model
12
 
13
- This is a custom trained GroundingDINO model for object detection and grounding.
14
 
15
  ## Model Details
16
 
@@ -18,8 +20,39 @@ This is a custom trained GroundingDINO model for object detection and grounding.
18
  - **Number of Classes**: 1180
19
  - **Training Dataset**: Custom dataset with 1180 object classes
20
  - **Architecture**: GroundingDINO with Swin-T backbone
 
21
 
22
- ## Usage
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
 
24
  ```python
25
  from model_loader import ModelLoader, quick_inference
@@ -30,271 +63,84 @@ results = quick_inference('your_image.jpg')
30
  # Or load model manually
31
  model = ModelLoader.load_model(
32
  checkpoint_path='pytorch_model.bin',
33
- config_path='config.py',
34
  device='cuda'
35
  )
36
 
37
  label_map = ModelLoader.load_label_map('label_map.json')
38
  ```
39
 
 
 
 
 
 
 
 
 
 
40
  ## Classes
41
 
42
- This model can detect the following 1180 classes:
43
-
44
-
45
- - blue and purple polka dot block - blue and purple polka dot bowl - blue and purple polka dot container - blue and purple polka dot cross - blue and purple polka dot diamond
46
- - blue and purple polka dot flower - blue and purple polka dot frame - blue and purple polka dot heart - blue and purple polka dot hexagon - blue and purple polka dot l-shaped block
47
- - blue and purple polka dot letter a - blue and purple polka dot letter e - blue and purple polka dot letter g - blue and purple polka dot letter m - blue and purple polka dot letter r
48
- - blue and purple polka dot letter t - blue and purple polka dot letter v - blue and purple polka dot line - blue and purple polka dot pallet - blue and purple polka dot pan
49
- - blue and purple polka dot pentagon - blue and purple polka dot ring - blue and purple polka dot round - blue and purple polka dot square - blue and purple polka dot star
50
- - blue and purple polka dot three-sided rectangle - blue and purple polka dot triangle - blue and purple stripe block - blue and purple stripe bowl - blue and purple stripe cross
51
- - blue and purple stripe diamond - blue and purple stripe flower - blue and purple stripe heart - blue and purple stripe hexagon - blue and purple stripe l-shaped block
52
- - blue and purple stripe letter a - blue and purple stripe letter e - blue and purple stripe letter g - blue and purple stripe letter m - blue and purple stripe letter r
53
- - blue and purple stripe letter t - blue and purple stripe letter v - blue and purple stripe pallet - blue and purple stripe pentagon - blue and purple stripe ring
54
- - blue and purple stripe round - blue and purple stripe small block - blue and purple stripe square - blue and purple stripe star - blue and purple stripe triangle
55
- - blue bowl - blue container - blue cross - blue diamond - blue flower
56
- - blue frame - blue heart - blue hexagon - blue l-shaped block - blue letter a
57
- - blue letter e - blue letter g - blue letter m - blue letter r - blue letter t
58
- - blue letter v - blue line - blue paisley block - blue paisley bowl - blue paisley container
59
- - blue paisley diamond - blue paisley flower - blue paisley frame - blue paisley heart - blue paisley l-shaped block
60
- - blue paisley letter a - blue paisley letter e - blue paisley letter m - blue paisley letter r - blue paisley letter t
61
- - blue paisley letter v - blue paisley line - blue paisley pallet - blue paisley pan - blue paisley pentagon
62
- - blue paisley ring - blue paisley round - blue paisley square - blue paisley star - blue paisley three-sided rectangle
63
- - blue paisley triangle - blue pallet - blue pan - blue pentagon - blue ring
64
- - blue round - blue shorter block - blue square - blue star - blue swirl block
65
- - blue swirl bowl - blue swirl container - blue swirl cross - blue swirl diamond - blue swirl flower
66
- - blue swirl frame - blue swirl heart - blue swirl hexagon - blue swirl l-shaped block - blue swirl letter a
67
- - blue swirl letter e - blue swirl letter g - blue swirl letter m - blue swirl letter t - blue swirl letter v
68
- - blue swirl line - blue swirl pallet - blue swirl pan - blue swirl pentagon - blue swirl ring
69
- - blue swirl round - blue swirl square - blue swirl star - blue swirl three-sided rectangle - blue swirl triangle
70
- - blue three-sided rectangle - blue triangle - brick block - brick cross - brick diamond
71
- - brick flower - brick heart - brick hexagon - brick l-shaped block - brick letter a
72
- - brick letter e - brick letter g - brick letter m - brick letter r - brick letter t
73
- - brick letter v - brick pentagon - brick ring - brick round - brick shorter block
74
- - brick small block - brick star - brick triangle - checkerboard block - checkerboard bowl
75
- - checkerboard cross - checkerboard diamond - checkerboard flower - checkerboard heart - checkerboard hexagon
76
- - checkerboard l-shaped block - checkerboard letter a - checkerboard letter e - checkerboard letter g - checkerboard letter m
77
- - checkerboard letter r - checkerboard letter t - checkerboard letter v - checkerboard pentagon - checkerboard ring
78
- - checkerboard round - checkerboard shorter block - checkerboard small block - checkerboard star - checkerboard triangle
79
- - cyan bowl - cyan container - cyan cross - cyan diamond - cyan flower
80
- - cyan frame - cyan heart - cyan hexagon - cyan l-shaped block - cyan letter a
81
- - cyan letter e - cyan letter g - cyan letter m - cyan letter r - cyan letter t
82
- - cyan letter v - cyan line - cyan pallet - cyan pan - cyan pentagon
83
- - cyan ring - cyan round - cyan shorter block - cyan square - cyan star
84
- - cyan three-sided rectangle - cyan triangle - granite block - granite cross - granite diamond
85
- - granite flower - granite heart - granite hexagon - granite l-shaped block - granite letter a
86
- - granite letter e - granite letter g - granite letter m - granite letter r - granite letter t
87
- - granite letter v - granite pentagon - granite ring - granite round - granite shorter block
88
- - granite small block - granite star - granite triangle - green and blue polka dot block - green and blue polka dot bowl
89
- - green and blue polka dot container - green and blue polka dot cross - green and blue polka dot frame - green and blue polka dot heart - green and blue polka dot hexagon
90
- - green and blue polka dot l-shaped block - green and blue polka dot letter a - green and blue polka dot letter e - green and blue polka dot letter g - green and blue polka dot letter m
91
- - green and blue polka dot letter r - green and blue polka dot letter t - green and blue polka dot letter v - green and blue polka dot line - green and blue polka dot pallet
92
- - green and blue polka dot pan - green and blue polka dot pentagon - green and blue polka dot ring - green and blue polka dot round - green and blue polka dot shorter block
93
- - green and blue polka dot square - green and blue polka dot star - green and blue polka dot three-sided rectangle - green and blue polka dot triangle - green and blue stripe block
94
- - green and blue stripe cross - green and blue stripe diamond - green and blue stripe flower - green and blue stripe heart - green and blue stripe hexagon
95
- - green and blue stripe l-shaped block - green and blue stripe letter a - green and blue stripe letter e - green and blue stripe letter g - green and blue stripe letter m
96
- - green and blue stripe letter r - green and blue stripe letter t - green and blue stripe letter v - green and blue stripe pentagon - green and blue stripe ring
97
- - green and blue stripe round - green and blue stripe small block - green and blue stripe star - green and blue stripe triangle - green and purple polka dot block
98
- - green and purple polka dot bowl - green and purple polka dot container - green and purple polka dot cross - green and purple polka dot diamond - green and purple polka dot flower
99
- - green and purple polka dot frame - green and purple polka dot heart - green and purple polka dot hexagon - green and purple polka dot l-shaped block - green and purple polka dot letter a
100
- - green and purple polka dot letter e - green and purple polka dot letter g - green and purple polka dot letter m - green and purple polka dot letter r - green and purple polka dot letter t
101
- - green and purple polka dot letter v - green and purple polka dot line - green and purple polka dot pallet - green and purple polka dot pan - green and purple polka dot pentagon
102
- - green and purple polka dot ring - green and purple polka dot round - green and purple polka dot shorter block - green and purple polka dot square - green and purple polka dot star
103
- - green and purple polka dot three-sided rectangle - green and purple polka dot triangle - green and purple stripe block - green and purple stripe container - green and purple stripe cross
104
- - green and purple stripe diamond - green and purple stripe flower - green and purple stripe heart - green and purple stripe hexagon - green and purple stripe l-shaped block
105
- - green and purple stripe letter a - green and purple stripe letter e - green and purple stripe letter g - green and purple stripe letter m - green and purple stripe letter r
106
- - green and purple stripe letter t - green and purple stripe letter v - green and purple stripe pentagon - green and purple stripe ring - green and purple stripe round
107
- - green and purple stripe small block - green and purple stripe star - green and purple stripe triangle - green bowl - green container
108
- - green cross - green diamond - green flower - green frame - green heart
109
- - green hexagon - green l-shaped block - green letter a - green letter e - green letter g
110
- - green letter m - green letter r - green letter t - green letter v - green line
111
- - green paisley block - green paisley bowl - green paisley container - green paisley cross - green paisley diamond
112
- - green paisley flower - green paisley frame - green paisley heart - green paisley hexagon - green paisley l-shaped block
113
- - green paisley letter a - green paisley letter e - green paisley letter g - green paisley letter m - green paisley letter r
114
- - green paisley letter t - green paisley letter v - green paisley line - green paisley pallet - green paisley pan
115
- - green paisley pentagon - green paisley ring - green paisley round - green paisley square - green paisley star
116
- - green paisley three-sided rectangle - green paisley triangle - green pallet - green pan - green pentagon
117
- - green ring - green round - green shorter block - green square - green star
118
- - green swirl block - green swirl bowl - green swirl container - green swirl cross - green swirl diamond
119
- - green swirl flower - green swirl frame - green swirl heart - green swirl hexagon - green swirl l-shaped block
120
- - green swirl letter a - green swirl letter e - green swirl letter g - green swirl letter m - green swirl letter r
121
- - green swirl letter t - green swirl letter v - green swirl line - green swirl pallet - green swirl pan
122
- - green swirl pentagon - green swirl round - green swirl square - green swirl star - green swirl three-sided rectangle
123
- - green swirl triangle - green three-sided rectangle - green triangle - magma block - magma cross
124
- - magma diamond - magma flower - magma heart - magma hexagon - magma l-shaped block
125
- - magma letter a - magma letter e - magma letter g - magma letter m - magma letter r
126
- - magma letter t - magma letter v - magma pallet - magma pentagon - magma ring
127
- - magma round - magma small block - magma star - magma triangle - olive block
128
- - olive bowl - olive container - olive cross - olive diamond - olive flower
129
- - olive frame - olive heart - olive hexagon - olive l-shaped block - olive letter a
130
- - olive letter e - olive letter g - olive letter m - olive letter r - olive letter t
131
- - olive letter v - olive line - olive pallet - olive pan - olive pentagon
132
- - olive ring - olive round - olive shorter block - olive square - olive star
133
- - olive three-sided rectangle - olive triangle - orange bowl - orange container - orange cross
134
- - orange diamond - orange flower - orange frame - orange heart - orange hexagon
135
- - orange l-shaped block - orange letter a - orange letter e - orange letter g - orange letter m
136
- - orange letter r - orange letter t - orange letter v - orange line - orange pallet
137
- - orange pan - orange pentagon - orange ring - orange round - orange shorter block
138
- - orange square - orange star - orange three-sided rectangle - orange triangle - pink block
139
- - pink bowl - pink container - pink cross - pink diamond - pink flower
140
- - pink frame - pink heart - pink hexagon - pink l-shaped block - pink letter a
141
- - pink letter e - pink letter g - pink letter m - pink letter r - pink letter t
142
- - pink letter v - pink line - pink pallet - pink pan - pink ring
143
- - pink round - pink shorter block - pink square - pink three-sided rectangle - pink triangle
144
- - plastic block - plastic cross - plastic diamond - plastic flower - plastic heart
145
- - plastic hexagon - plastic l-shaped block - plastic letter a - plastic letter e - plastic letter g
146
- - plastic letter m - plastic letter r - plastic letter t - plastic letter v - plastic pentagon
147
- - plastic ring - plastic round - plastic shorter block - plastic small block - plastic star
148
- - plastic triangle - polka dot block - polka dot bowl - polka dot cross - polka dot diamond
149
- - polka dot flower - polka dot heart - polka dot hexagon - polka dot l-shaped block - polka dot letter a
150
- - polka dot letter e - polka dot letter g - polka dot letter m - polka dot letter r - polka dot letter t
151
- - polka dot letter v - polka dot pan - polka dot pentagon - polka dot ring - polka dot round
152
- - polka dot small block - polka dot star - polka dot triangle - purple block - purple bowl
153
- - purple container - purple cross - purple diamond - purple flower - purple frame
154
- - purple heart - purple hexagon - purple l-shaped block - purple letter a - purple letter e
155
- - purple letter m - purple letter r - purple letter t - purple letter v - purple line
156
- - purple paisley block - purple paisley bowl - purple paisley container - purple paisley cross - purple paisley diamond
157
- - purple paisley flower - purple paisley frame - purple paisley heart - purple paisley hexagon - purple paisley l-shaped block
158
- - purple paisley letter a - purple paisley letter e - purple paisley letter g - purple paisley letter m - purple paisley letter r
159
- - purple paisley letter t - purple paisley letter v - purple paisley line - purple paisley pallet - purple paisley pan
160
- - purple paisley pentagon - purple paisley ring - purple paisley square - purple paisley star - purple paisley three-sided rectangle
161
- - purple paisley triangle - purple pallet - purple pan - purple pentagon - purple ring
162
- - purple square - purple star - purple swirl block - purple swirl bowl - purple swirl container
163
- - purple swirl cross - purple swirl diamond - purple swirl flower - purple swirl frame - purple swirl heart
164
- - purple swirl hexagon - purple swirl l-shaped block - purple swirl letter a - purple swirl letter e - purple swirl letter g
165
- - purple swirl letter m - purple swirl letter r - purple swirl letter t - purple swirl letter v - purple swirl line
166
- - purple swirl pallet - purple swirl pan - purple swirl pentagon - purple swirl ring - purple swirl round
167
- - purple swirl shorter block - purple swirl square - purple swirl star - purple swirl three-sided rectangle - purple swirl triangle
168
- - purple three-sided rectangle - purple triangle - rainbow block - rainbow cross - rainbow diamond
169
- - rainbow flower - rainbow heart - rainbow hexagon - rainbow l-shaped block - rainbow letter a
170
- - rainbow letter e - rainbow letter g - rainbow letter m - rainbow letter r - rainbow letter t
171
- - rainbow letter v - rainbow pentagon - rainbow ring - rainbow round - rainbow shorter block
172
- - rainbow small block - rainbow square - rainbow star - rainbow triangle - red and blue polka dot block
173
- - red and blue polka dot cross - red and blue polka dot diamond - red and blue polka dot flower - red and blue polka dot heart - red and blue polka dot hexagon
174
- - red and blue polka dot l-shaped block - red and blue polka dot letter a - red and blue polka dot letter e - red and blue polka dot letter g - red and blue polka dot letter m
175
- - red and blue polka dot letter r - red and blue polka dot letter t - red and blue polka dot letter v - red and blue polka dot pallet - red and blue polka dot pentagon
176
- - red and blue polka dot ring - red and blue polka dot round - red and blue polka dot shorter block - red and blue polka dot small block - red and blue polka dot star
177
- - red and blue polka dot triangle - red and blue stripe block - red and blue stripe cross - red and blue stripe diamond - red and blue stripe flower
178
- - red and blue stripe heart - red and blue stripe hexagon - red and blue stripe l-shaped block - red and blue stripe letter a - red and blue stripe letter e
179
- - red and blue stripe letter g - red and blue stripe letter m - red and blue stripe letter r - red and blue stripe letter t - red and blue stripe letter v
180
- - red and blue stripe pentagon - red and blue stripe ring - red and blue stripe round - red and blue stripe small block - red and blue stripe star
181
- - red and blue stripe triangle - red and green polka dot block - red and green polka dot cross - red and green polka dot diamond - red and green polka dot flower
182
- - red and green polka dot heart - red and green polka dot hexagon - red and green polka dot l-shaped block - red and green polka dot letter a - red and green polka dot letter e
183
- - red and green polka dot letter g - red and green polka dot letter m - red and green polka dot letter r - red and green polka dot letter t - red and green polka dot letter v
184
- - red and green polka dot pan - red and green polka dot pentagon - red and green polka dot ring - red and green polka dot round - red and green polka dot small block
185
- - red and green polka dot star - red and green polka dot triangle - red and green stripe block - red and green stripe bowl - red and green stripe cross
186
- - red and green stripe diamond - red and green stripe flower - red and green stripe heart - red and green stripe hexagon - red and green stripe l-shaped block
187
- - red and green stripe letter a - red and green stripe letter e - red and green stripe letter g - red and green stripe letter m - red and green stripe letter r
188
- - red and green stripe letter t - red and green stripe letter v - red and green stripe pentagon - red and green stripe ring - red and green stripe round
189
- - red and green stripe shorter block - red and green stripe small block - red and green stripe star - red and green stripe triangle - red and purple polka dot block
190
- - red and purple polka dot cross - red and purple polka dot diamond - red and purple polka dot flower - red and purple polka dot heart - red and purple polka dot hexagon
191
- - red and purple polka dot l-shaped block - red and purple polka dot letter a - red and purple polka dot letter e - red and purple polka dot letter g - red and purple polka dot letter m
192
- - red and purple polka dot letter r - red and purple polka dot letter t - red and purple polka dot letter v - red and purple polka dot pentagon - red and purple polka dot ring
193
- - red and purple polka dot round - red and purple polka dot small block - red and purple polka dot star - red and purple polka dot triangle - red and purple stripe block
194
- - red and purple stripe cross - red and purple stripe diamond - red and purple stripe flower - red and purple stripe heart - red and purple stripe hexagon
195
- - red and purple stripe l-shaped block - red and purple stripe letter a - red and purple stripe letter e - red and purple stripe letter g - red and purple stripe letter m
196
- - red and purple stripe letter r - red and purple stripe letter t - red and purple stripe letter v - red and purple stripe pentagon - red and purple stripe ring
197
- - red and purple stripe round - red and purple stripe shorter block - red and purple stripe small block - red and purple stripe star - red and purple stripe triangle
198
- - red and yellow polka dot block - red and yellow polka dot cross - red and yellow polka dot diamond - red and yellow polka dot flower - red and yellow polka dot heart
199
- - red and yellow polka dot hexagon - red and yellow polka dot l-shaped block - red and yellow polka dot letter a - red and yellow polka dot letter e - red and yellow polka dot letter g
200
- - red and yellow polka dot letter m - red and yellow polka dot letter r - red and yellow polka dot letter t - red and yellow polka dot letter v - red and yellow polka dot pentagon
201
- - red and yellow polka dot ring - red and yellow polka dot round - red and yellow polka dot shorter block - red and yellow polka dot small block - red and yellow polka dot star
202
- - red and yellow polka dot triangle - red and yellow stripe block - red and yellow stripe bowl - red and yellow stripe cross - red and yellow stripe diamond
203
- - red and yellow stripe flower - red and yellow stripe heart - red and yellow stripe hexagon - red and yellow stripe l-shaped block - red and yellow stripe letter a
204
- - red and yellow stripe letter e - red and yellow stripe letter g - red and yellow stripe letter m - red and yellow stripe letter r - red and yellow stripe letter t
205
- - red and yellow stripe letter v - red and yellow stripe pentagon - red and yellow stripe ring - red and yellow stripe round - red and yellow stripe small block
206
- - red and yellow stripe star - red and yellow stripe triangle - red block - red bowl - red container
207
- - red cross - red diamond - red flower - red frame - red heart
208
- - red hexagon - red l-shaped block - red letter a - red letter e - red letter g
209
- - red letter m - red letter r - red letter t - red letter v - red line
210
- - red paisley block - red paisley bowl - red paisley container - red paisley cross - red paisley diamond
211
- - red paisley flower - red paisley frame - red paisley heart - red paisley hexagon - red paisley l-shaped block
212
- - red paisley letter a - red paisley letter e - red paisley letter g - red paisley letter m - red paisley letter r
213
- - red paisley letter t - red paisley letter v - red paisley line - red paisley pallet - red paisley pan
214
- - red paisley pentagon - red paisley ring - red paisley round - red paisley shorter block - red paisley square
215
- - red paisley star - red paisley three-sided rectangle - red paisley triangle - red pallet - red pan
216
- - red pentagon - red ring - red round - red shorter block - red square
217
- - red star - red swirl block - red swirl bowl - red swirl container - red swirl cross
218
- - red swirl diamond - red swirl flower - red swirl frame - red swirl heart - red swirl hexagon
219
- - red swirl l-shaped block - red swirl letter a - red swirl letter e - red swirl letter g - red swirl letter m
220
- - red swirl letter r - red swirl letter t - red swirl letter v - red swirl line - red swirl pallet
221
- - red swirl pan - red swirl pentagon - red swirl ring - red swirl round - red swirl shorter block
222
- - red swirl square - red swirl star - red swirl three-sided rectangle - red swirl triangle - red three-sided rectangle
223
- - red triangle - tiger block - tiger cross - tiger diamond - tiger flower
224
- - tiger heart - tiger hexagon - tiger l-shaped block - tiger letter a - tiger letter e
225
- - tiger letter g - tiger letter m - tiger letter r - tiger letter t - tiger letter v
226
- - tiger pentagon - tiger ring - tiger round - tiger small block - tiger square
227
- - tiger star - tiger triangle - tiles block - tiles cross - tiles diamond
228
- - tiles flower - tiles heart - tiles hexagon - tiles l-shaped block - tiles letter a
229
- - tiles letter e - tiles letter g - tiles letter m - tiles letter r - tiles letter t
230
- - tiles letter v - tiles pentagon - tiles ring - tiles round - tiles shorter block
231
- - tiles small block - tiles square - tiles star - tiles triangle - wooden block
232
- - wooden cross - wooden diamond - wooden flower - wooden heart - wooden hexagon
233
- - wooden l-shaped block - wooden letter a - wooden letter e - wooden letter g - wooden letter m
234
- - wooden letter r - wooden letter t - wooden letter v - wooden pentagon - wooden ring
235
- - wooden round - wooden shorter block - wooden small block - wooden star - wooden triangle
236
- - yellow and blue polka dot block - yellow and blue polka dot bowl - yellow and blue polka dot container - yellow and blue polka dot cross - yellow and blue polka dot diamond
237
- - yellow and blue polka dot flower - yellow and blue polka dot frame - yellow and blue polka dot heart - yellow and blue polka dot hexagon - yellow and blue polka dot l-shaped block
238
- - yellow and blue polka dot letter a - yellow and blue polka dot letter e - yellow and blue polka dot letter g - yellow and blue polka dot letter m - yellow and blue polka dot letter r
239
- - yellow and blue polka dot letter t - yellow and blue polka dot letter v - yellow and blue polka dot line - yellow and blue polka dot pallet - yellow and blue polka dot pan
240
- - yellow and blue polka dot pentagon - yellow and blue polka dot ring - yellow and blue polka dot round - yellow and blue polka dot square - yellow and blue polka dot star
241
- - yellow and blue polka dot three-sided rectangle - yellow and blue polka dot triangle - yellow and blue stripe block - yellow and blue stripe cross - yellow and blue stripe diamond
242
- - yellow and blue stripe flower - yellow and blue stripe heart - yellow and blue stripe hexagon - yellow and blue stripe l-shaped block - yellow and blue stripe letter a
243
- - yellow and blue stripe letter e - yellow and blue stripe letter g - yellow and blue stripe letter m - yellow and blue stripe letter r - yellow and blue stripe letter t
244
- - yellow and blue stripe letter v - yellow and blue stripe pan - yellow and blue stripe pentagon - yellow and blue stripe ring - yellow and blue stripe round
245
- - yellow and blue stripe small block - yellow and blue stripe star - yellow and blue stripe triangle - yellow and green polka dot block - yellow and green polka dot bowl
246
- - yellow and green polka dot container - yellow and green polka dot cross - yellow and green polka dot diamond - yellow and green polka dot flower - yellow and green polka dot frame
247
- - yellow and green polka dot heart - yellow and green polka dot hexagon - yellow and green polka dot l-shaped block - yellow and green polka dot letter a - yellow and green polka dot letter e
248
- - yellow and green polka dot letter g - yellow and green polka dot letter m - yellow and green polka dot letter r - yellow and green polka dot letter t - yellow and green polka dot letter v
249
- - yellow and green polka dot line - yellow and green polka dot pallet - yellow and green polka dot pan - yellow and green polka dot pentagon - yellow and green polka dot ring
250
- - yellow and green polka dot round - yellow and green polka dot shorter block - yellow and green polka dot square - yellow and green polka dot star - yellow and green polka dot three-sided rectangle
251
- - yellow and green stripe block - yellow and green stripe bowl - yellow and green stripe cross - yellow and green stripe diamond - yellow and green stripe flower
252
- - yellow and green stripe heart - yellow and green stripe hexagon - yellow and green stripe l-shaped block - yellow and green stripe letter a - yellow and green stripe letter e
253
- - yellow and green stripe letter g - yellow and green stripe letter m - yellow and green stripe letter r - yellow and green stripe letter t - yellow and green stripe letter v
254
- - yellow and green stripe pentagon - yellow and green stripe ring - yellow and green stripe round - yellow and green stripe shorter block - yellow and green stripe small block
255
- - yellow and green stripe star - yellow and green stripe triangle - yellow and purple polka dot block - yellow and purple polka dot bowl - yellow and purple polka dot container
256
- - yellow and purple polka dot cross - yellow and purple polka dot diamond - yellow and purple polka dot flower - yellow and purple polka dot frame - yellow and purple polka dot heart
257
- - yellow and purple polka dot hexagon - yellow and purple polka dot l-shaped block - yellow and purple polka dot letter a - yellow and purple polka dot letter e - yellow and purple polka dot letter g
258
- - yellow and purple polka dot letter m - yellow and purple polka dot letter r - yellow and purple polka dot letter t - yellow and purple polka dot letter v - yellow and purple polka dot line
259
- - yellow and purple polka dot pallet - yellow and purple polka dot pan - yellow and purple polka dot pentagon - yellow and purple polka dot ring - yellow and purple polka dot round
260
- - yellow and purple polka dot shorter block - yellow and purple polka dot square - yellow and purple polka dot star - yellow and purple polka dot three-sided rectangle - yellow and purple polka dot triangle
261
- - yellow and purple stripe block - yellow and purple stripe bowl - yellow and purple stripe cross - yellow and purple stripe diamond - yellow and purple stripe flower
262
- - yellow and purple stripe heart - yellow and purple stripe hexagon - yellow and purple stripe l-shaped block - yellow and purple stripe letter a - yellow and purple stripe letter e
263
- - yellow and purple stripe letter g - yellow and purple stripe letter m - yellow and purple stripe letter r - yellow and purple stripe letter t - yellow and purple stripe letter v
264
- - yellow and purple stripe pentagon - yellow and purple stripe ring - yellow and purple stripe round - yellow and purple stripe small block - yellow and purple stripe star
265
- - yellow and purple stripe triangle - yellow bowl - yellow container - yellow cross - yellow diamond
266
- - yellow flower - yellow frame - yellow heart - yellow hexagon - yellow l-shaped block
267
- - yellow letter e - yellow letter g - yellow letter m - yellow letter r - yellow letter v
268
- - yellow line - yellow paisley block - yellow paisley bowl - yellow paisley container - yellow paisley cross
269
- - yellow paisley diamond - yellow paisley flower - yellow paisley frame - yellow paisley heart - yellow paisley hexagon
270
- - yellow paisley l-shaped block - yellow paisley letter a - yellow paisley letter e - yellow paisley letter g - yellow paisley letter m
271
- - yellow paisley letter r - yellow paisley letter t - yellow paisley letter v - yellow paisley line - yellow paisley pallet
272
- - yellow paisley pan - yellow paisley pentagon - yellow paisley ring - yellow paisley round - yellow paisley square
273
- - yellow paisley star - yellow paisley three-sided rectangle - yellow paisley triangle - yellow pallet - yellow pan
274
- - yellow pentagon - yellow ring - yellow round - yellow shorter block - yellow square
275
- - yellow star - yellow swirl block - yellow swirl bowl - yellow swirl container - yellow swirl cross
276
- - yellow swirl diamond - yellow swirl flower - yellow swirl frame - yellow swirl heart - yellow swirl hexagon
277
- - yellow swirl l-shaped block - yellow swirl letter a - yellow swirl letter e - yellow swirl letter g - yellow swirl letter m
278
- - yellow swirl letter r - yellow swirl letter t - yellow swirl line - yellow swirl pallet - yellow swirl pan
279
- - yellow swirl pentagon - yellow swirl ring - yellow swirl round - yellow swirl shorter block - yellow swirl square
280
- - yellow swirl star - yellow swirl three-sided rectangle - yellow swirl triangle - yellow three-sided rectangle - yellow triangle
281
-
282
- ## Files
283
-
284
- - `pytorch_model.bin`: Model weights
285
- - `config.py`: Model configuration
286
- - `label_map.json`: Class label mapping
287
- - `model_card.json`: Model metadata
288
 
289
  ## Citation
290
 
291
  If you use this model, please cite the original GroundingDINO paper:
292
 
293
  ```bibtex
294
- @article{liu2023grounding,
295
- title={Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection},
296
- author={Liu, Shilong and Zeng, Zhaoyang and Ren, Tianhe and Li, Feng and Zhang, Hao and Yang, Jie and Li, Chunyuan and Yang, Jianwei and Su, Hang and Zhu, Jun and others},
297
- journal={arXiv preprint arXiv:2303.05499},
298
- year={2023}
299
- }
300
  ```
 
 
 
 
 
1
  ---
2
+ library_name: transformers
3
  tags:
4
  - object-detection
5
  - grounding
6
  - vision
7
  - custom-dataset
8
+ - groundingdino
9
  license: mit
10
+ pipeline_tag: object-detection
11
  ---
12
 
13
  # Custom GroundingDINO Model
14
 
15
+ This is a custom trained GroundingDINO model for object detection and grounding, compatible with the Hugging Face Transformers library.
16
 
17
  ## Model Details
18
 
 
20
  - **Number of Classes**: 1180
21
  - **Training Dataset**: Custom dataset with 1180 object classes
22
  - **Architecture**: GroundingDINO with Swin-T backbone
23
+ - **Transformers Compatible**: ✅ Yes
24
 
25
+ ## Usage with Transformers
26
+
27
+ ```python
28
+ from transformers import AutoModel, AutoConfig, AutoTokenizer
29
+ import torch
30
+ from PIL import Image
31
+
32
+ # Load model and config
33
+ model = AutoModel.from_pretrained("your_username/your_model_name")
34
+ config = AutoConfig.from_pretrained("your_username/your_model_name")
35
+
36
+ # Load label map
37
+ import json
38
+ with open("label_map.json", "r") as f:
39
+ label_map = json.load(f)
40
+
41
+ # Prepare text prompt
42
+ text_prompt = ". ".join(list(label_map.values())[:100]) + "."
43
+
44
+ # Load and preprocess image
45
+ image = Image.open("your_image.jpg").convert("RGB")
46
+ # Add your image preprocessing here
47
+
48
+ # Run inference
49
+ with torch.no_grad():
50
+ outputs = model(images=image, text_prompts=[text_prompt])
51
+ logits = outputs.logits
52
+ boxes = outputs.boxes
53
+ ```
54
+
55
+ ## Usage with Original Implementation
56
 
57
  ```python
58
  from model_loader import ModelLoader, quick_inference
 
63
  # Or load model manually
64
  model = ModelLoader.load_model(
65
  checkpoint_path='pytorch_model.bin',
66
+ config_path='original_config.py',
67
  device='cuda'
68
  )
69
 
70
  label_map = ModelLoader.load_label_map('label_map.json')
71
  ```
72
 
73
+ ## Model Files
74
+
75
+ - `pytorch_model.bin`: Model weights (transformers format)
76
+ - `config.json`: Transformers configuration
77
+ - `modeling_groundingdino.py`: Custom model class
78
+ - `tokenizer_config.json`: Tokenizer configuration
79
+ - `label_map.json`: Class label mapping (1180 classes)
80
+ - `original_config.py`: Original training configuration
81
+
82
  ## Classes
83
 
84
+ This model can detect 1180 unique object classes including:
85
+
86
+ - blue and purple polka dot block
87
+ - blue and purple polka dot bowl
88
+ - blue and purple polka dot container
89
+ - blue and purple polka dot cross
90
+ - blue and purple polka dot diamond
91
+ - blue and purple polka dot flower
92
+ - blue and purple polka dot frame
93
+ - blue and purple polka dot heart
94
+ - blue and purple polka dot hexagon
95
+ - blue and purple polka dot l-shaped block
96
+ - blue and purple polka dot letter a
97
+ - blue and purple polka dot letter e
98
+ - blue and purple polka dot letter g
99
+ - blue and purple polka dot letter m
100
+ - blue and purple polka dot letter r
101
+ - blue and purple polka dot letter t
102
+ - blue and purple polka dot letter v
103
+ - blue and purple polka dot line
104
+ - blue and purple polka dot pallet
105
+ - blue and purple polka dot pan
106
+
107
+ ... and 1160 more classes.
108
+
109
+
110
+ ## Installation
111
+
112
+ ```bash
113
+ pip install transformers torch torchvision
114
+ ```
115
+
116
+ ## Example Classes
117
+
118
+ The model can detect objects with various:
119
+ - **Colors**: blue, red, green, yellow, purple, etc.
120
+ - **Patterns**: polka dot, stripe, paisley, swirl, checkerboard
121
+ - **Shapes**: block, bowl, container, cross, diamond, flower
122
+ - **Combinations**: "blue and purple polka dot block", "red stripe heart"
123
+
124
+ ## Performance
125
+
126
+ - **Model Size**: ~1.1 GB
127
+ - **Parameters**: ~172M
128
+ - **Training**: 12 epochs on custom dataset
129
+ - **Memory Usage**: ~2-4 GB GPU memory during inference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
130
 
131
  ## Citation
132
 
133
  If you use this model, please cite the original GroundingDINO paper:
134
 
135
  ```bibtex
136
+ @article{{liu2023grounding,
137
+ title={{Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection}},
138
+ author={{Liu, Shilong and Zeng, Zhaoyang and Ren, Tianhe and Li, Feng and Zhang, Hao and Yang, Jie and Li, Chunyuan and Yang, Jianwei and Su, Hang and Zhu, Jun and others}},
139
+ journal={{arXiv preprint arXiv:2303.05499}},
140
+ year={{2023}}
141
+ }}
142
  ```
143
+
144
+ ## License
145
+
146
+ This model is released under the MIT License.
config.json ADDED
@@ -0,0 +1,128 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "groundingdino",
3
+ "architecture": "GroundingDINO",
4
+ "backbone": "Swin-T",
5
+ "num_classes": 1180,
6
+ "num_queries": 900,
7
+ "hidden_dim": 256,
8
+ "num_feature_levels": 4,
9
+ "nheads": 8,
10
+ "enc_layers": 6,
11
+ "dec_layers": 6,
12
+ "dim_feedforward": 2048,
13
+ "dropout": 0.0,
14
+ "max_text_len": 256,
15
+ "text_encoder_type": "bert-base-uncased",
16
+ "use_text_enhancer": true,
17
+ "use_fusion_layer": true,
18
+ "use_checkpoint": true,
19
+ "use_transformer_ckpt": true,
20
+ "use_text_cross_attention": true,
21
+ "text_dropout": 0.0,
22
+ "fusion_dropout": 0.0,
23
+ "fusion_droppath": 0.1,
24
+ "sub_sentence_present": true,
25
+ "max_labels": 50,
26
+ "label_list": [
27
+ "blue and purple polka dot block",
28
+ "blue and purple polka dot bowl",
29
+ "blue and purple polka dot container",
30
+ "blue and purple polka dot cross",
31
+ "blue and purple polka dot diamond",
32
+ "blue and purple polka dot flower",
33
+ "blue and purple polka dot frame",
34
+ "blue and purple polka dot heart",
35
+ "blue and purple polka dot hexagon",
36
+ "blue and purple polka dot l-shaped block",
37
+ "blue and purple polka dot letter a",
38
+ "blue and purple polka dot letter e",
39
+ "blue and purple polka dot letter g",
40
+ "blue and purple polka dot letter m",
41
+ "blue and purple polka dot letter r",
42
+ "blue and purple polka dot letter t",
43
+ "blue and purple polka dot letter v",
44
+ "blue and purple polka dot line",
45
+ "blue and purple polka dot pallet",
46
+ "blue and purple polka dot pan",
47
+ "blue and purple polka dot pentagon",
48
+ "blue and purple polka dot ring",
49
+ "blue and purple polka dot round",
50
+ "blue and purple polka dot square",
51
+ "blue and purple polka dot star",
52
+ "blue and purple polka dot three-sided rectangle",
53
+ "blue and purple polka dot triangle",
54
+ "blue and purple stripe block",
55
+ "blue and purple stripe bowl",
56
+ "blue and purple stripe cross",
57
+ "blue and purple stripe diamond",
58
+ "blue and purple stripe flower",
59
+ "blue and purple stripe heart",
60
+ "blue and purple stripe hexagon",
61
+ "blue and purple stripe l-shaped block",
62
+ "blue and purple stripe letter a",
63
+ "blue and purple stripe letter e",
64
+ "blue and purple stripe letter g",
65
+ "blue and purple stripe letter m",
66
+ "blue and purple stripe letter r",
67
+ "blue and purple stripe letter t",
68
+ "blue and purple stripe letter v",
69
+ "blue and purple stripe pallet",
70
+ "blue and purple stripe pentagon",
71
+ "blue and purple stripe ring",
72
+ "blue and purple stripe round",
73
+ "blue and purple stripe small block",
74
+ "blue and purple stripe square",
75
+ "blue and purple stripe star",
76
+ "blue and purple stripe triangle",
77
+ "blue bowl",
78
+ "blue container",
79
+ "blue cross",
80
+ "blue diamond",
81
+ "blue flower",
82
+ "blue frame",
83
+ "blue heart",
84
+ "blue hexagon",
85
+ "blue l-shaped block",
86
+ "blue letter a",
87
+ "blue letter e",
88
+ "blue letter g",
89
+ "blue letter m",
90
+ "blue letter r",
91
+ "blue letter t",
92
+ "blue letter v",
93
+ "blue line",
94
+ "blue paisley block",
95
+ "blue paisley bowl",
96
+ "blue paisley container",
97
+ "blue paisley diamond",
98
+ "blue paisley flower",
99
+ "blue paisley frame",
100
+ "blue paisley heart",
101
+ "blue paisley l-shaped block",
102
+ "blue paisley letter a",
103
+ "blue paisley letter e",
104
+ "blue paisley letter m",
105
+ "blue paisley letter r",
106
+ "blue paisley letter t",
107
+ "blue paisley letter v",
108
+ "blue paisley line",
109
+ "blue paisley pallet",
110
+ "blue paisley pan",
111
+ "blue paisley pentagon",
112
+ "blue paisley ring",
113
+ "blue paisley round",
114
+ "blue paisley square",
115
+ "blue paisley star",
116
+ "blue paisley three-sided rectangle",
117
+ "blue paisley triangle",
118
+ "blue pallet",
119
+ "blue pan",
120
+ "blue pentagon",
121
+ "blue ring",
122
+ "blue round",
123
+ "blue shorter block",
124
+ "blue square",
125
+ "blue star",
126
+ "blue swirl block"
127
+ ]
128
+ }
modeling_groundingdino.py ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Custom GroundingDINO model class for transformers compatibility.
3
+ """
4
+
5
+ import torch
6
+ import torch.nn as nn
7
+ from transformers import PreTrainedModel
8
+ from transformers.configuration_utils import PretrainedConfig
9
+
10
+
11
+ class GroundingDINOConfig(PretrainedConfig):
12
+ """Configuration class for GroundingDINO."""
13
+
14
+ model_type = "groundingdino"
15
+
16
+ def __init__(
17
+ self,
18
+ num_classes=1180,
19
+ num_queries=900,
20
+ hidden_dim=256,
21
+ num_feature_levels=4,
22
+ nheads=8,
23
+ enc_layers=6,
24
+ dec_layers=6,
25
+ dim_feedforward=2048,
26
+ dropout=0.0,
27
+ max_text_len=256,
28
+ text_encoder_type="bert-base-uncased",
29
+ backbone="swin_T_224_1k",
30
+ position_embedding="sine",
31
+ **kwargs
32
+ ):
33
+ super().__init__(**kwargs)
34
+ self.num_classes = num_classes
35
+ self.num_queries = num_queries
36
+ self.hidden_dim = hidden_dim
37
+ self.num_feature_levels = num_feature_levels
38
+ self.nheads = nheads
39
+ self.enc_layers = enc_layers
40
+ self.dec_layers = dec_layers
41
+ self.dim_feedforward = dim_feedforward
42
+ self.dropout = dropout
43
+ self.max_text_len = max_text_len
44
+ self.text_encoder_type = text_encoder_type
45
+ self.backbone = backbone
46
+ self.position_embedding = position_embedding
47
+
48
+
49
+ class GroundingDINOModel(PreTrainedModel):
50
+ """GroundingDINO model for transformers."""
51
+
52
+ config_class = GroundingDINOConfig
53
+
54
+ def __init__(self, config):
55
+ super().__init__(config)
56
+ self.config = config
57
+
58
+ # This is a placeholder - in practice, you would load the actual model architecture
59
+ # For now, we'll create a simple wrapper
60
+ self.model = None
61
+
62
+ def forward(self, images, text_prompts=None, return_dict=True):
63
+ """
64
+ Forward pass of the model.
65
+
66
+ Args:
67
+ images: Input images tensor
68
+ text_prompts: Text prompts for grounding
69
+ return_dict: Whether to return a dictionary
70
+
71
+ Returns:
72
+ Model outputs
73
+ """
74
+ if self.model is None:
75
+ raise NotImplementedError(
76
+ "Model architecture not implemented. "
77
+ "Please use the original GroundingDINO implementation for inference."
78
+ )
79
+
80
+ outputs = self.model(images, captions=text_prompts)
81
+
82
+ if return_dict:
83
+ return {
84
+ "logits": outputs.get("pred_logits", torch.tensor([])),
85
+ "boxes": outputs.get("pred_boxes", torch.tensor([])),
86
+ "last_hidden_state": outputs.get("last_hidden_state", torch.tensor([]))
87
+ }
88
+ else:
89
+ return outputs
original_config.py ADDED
@@ -0,0 +1,591 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data_aug_scales = [480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800]
2
+ data_aug_max_size = 1333
3
+ data_aug_scales2_resize = [400, 500, 600]
4
+ data_aug_scales2_crop = [384, 600]
5
+ data_aug_scale_overlap = None
6
+ batch_size = 2
7
+ modelname = 'groundingdino'
8
+ backbone = 'swin_T_224_1k'
9
+ position_embedding = 'sine'
10
+ pe_temperatureH = 20
11
+ pe_temperatureW = 20
12
+ return_interm_indices = [1, 2, 3]
13
+ enc_layers = 6
14
+ dec_layers = 6
15
+ pre_norm = False
16
+ dim_feedforward = 2048
17
+ hidden_dim = 256
18
+ dropout = 0.0
19
+ nheads = 8
20
+ num_queries = 900
21
+ query_dim = 4
22
+ num_patterns = 0
23
+ num_feature_levels = 4
24
+ enc_n_points = 4
25
+ dec_n_points = 4
26
+ two_stage_type = 'standard'
27
+ two_stage_bbox_embed_share = False
28
+ two_stage_class_embed_share = False
29
+ transformer_activation = 'relu'
30
+ dec_pred_bbox_embed_share = True
31
+ dn_box_noise_scale = 1.0
32
+ dn_label_noise_ratio = 0.5
33
+ dn_label_coef = 1.0
34
+ dn_bbox_coef = 1.0
35
+ embed_init_tgt = True
36
+ dn_labelbook_size = 1181
37
+ max_text_len = 256
38
+ text_encoder_type = 'bert-base-uncased'
39
+ use_text_enhancer = True
40
+ use_fusion_layer = True
41
+ use_checkpoint = True
42
+ use_transformer_ckpt = True
43
+ use_text_cross_attention = True
44
+ text_dropout = 0.0
45
+ fusion_dropout = 0.0
46
+ fusion_droppath = 0.1
47
+ sub_sentence_present = True
48
+ max_labels = 50
49
+ lr = 0.0001
50
+ backbone_freeze_keywords = None
51
+ freeze_keywords = ['bert']
52
+ lr_backbone = 1e-05
53
+ lr_backbone_names = ['backbone.0', 'bert']
54
+ lr_linear_proj_mult = 1e-05
55
+ lr_linear_proj_names = ['ref_point_head', 'sampling_offsets']
56
+ weight_decay = 0.0001
57
+ param_dict_type = 'ddetr_in_mmdet'
58
+ ddetr_lr_param = False
59
+ epochs = 12
60
+ lr_drop = 10
61
+ save_checkpoint_interval = 5
62
+ clip_max_norm = 0.1
63
+ onecyclelr = False
64
+ multi_step_lr = False
65
+ lr_drop_list = [10, 20]
66
+ frozen_weights = None
67
+ dilation = False
68
+ pdetr3_bbox_embed_diff_each_layer = False
69
+ pdetr3_refHW = -1
70
+ random_refpoints_xy = False
71
+ fix_refpoints_hw = -1
72
+ dabdetr_yolo_like_anchor_update = False
73
+ dabdetr_deformable_encoder = False
74
+ dabdetr_deformable_decoder = False
75
+ use_deformable_box_attn = False
76
+ box_attn_type = 'roi_align'
77
+ dec_layer_number = None
78
+ decoder_layer_noise = False
79
+ dln_xy_noise = 0.2
80
+ dln_hw_noise = 0.2
81
+ add_channel_attention = False
82
+ add_pos_value = False
83
+ two_stage_pat_embed = 0
84
+ two_stage_add_query_num = 0
85
+ two_stage_learn_wh = False
86
+ two_stage_default_hw = 0.05
87
+ two_stage_keep_all_tokens = False
88
+ num_select = 300
89
+ batch_norm_type = 'FrozenBatchNorm2d'
90
+ masks = False
91
+ aux_loss = True
92
+ set_cost_class = 1.0
93
+ set_cost_bbox = 5.0
94
+ set_cost_giou = 2.0
95
+ cls_loss_coef = 2.0
96
+ bbox_loss_coef = 5.0
97
+ giou_loss_coef = 2.0
98
+ enc_loss_coef = 1.0
99
+ interm_loss_coef = 1.0
100
+ no_interm_box_loss = False
101
+ mask_loss_coef = 1.0
102
+ dice_loss_coef = 1.0
103
+ focal_alpha = 0.25
104
+ focal_gamma = 2.0
105
+ decoder_sa_type = 'sa'
106
+ matcher_type = 'HungarianMatcher'
107
+ decoder_module_seq = ['sa', 'ca', 'ffn']
108
+ nms_iou_threshold = -1
109
+ dec_pred_class_embed_share = True
110
+ match_unstable_error = True
111
+ use_ema = False
112
+ ema_decay = 0.9997
113
+ ema_epoch = 0
114
+ use_detached_boxes_dec_out = False
115
+ use_coco_eval = False
116
+ dn_scalar = 100
117
+ label_list = [
118
+ 'blue and purple polka dot block', 'blue and purple polka dot bowl',
119
+ 'blue and purple polka dot container', 'blue and purple polka dot cross',
120
+ 'blue and purple polka dot diamond', 'blue and purple polka dot flower',
121
+ 'blue and purple polka dot frame', 'blue and purple polka dot heart',
122
+ 'blue and purple polka dot hexagon',
123
+ 'blue and purple polka dot l-shaped block',
124
+ 'blue and purple polka dot letter a', 'blue and purple polka dot letter e',
125
+ 'blue and purple polka dot letter g', 'blue and purple polka dot letter m',
126
+ 'blue and purple polka dot letter r', 'blue and purple polka dot letter t',
127
+ 'blue and purple polka dot letter v', 'blue and purple polka dot line',
128
+ 'blue and purple polka dot pallet', 'blue and purple polka dot pan',
129
+ 'blue and purple polka dot pentagon', 'blue and purple polka dot ring',
130
+ 'blue and purple polka dot round', 'blue and purple polka dot square',
131
+ 'blue and purple polka dot star',
132
+ 'blue and purple polka dot three-sided rectangle',
133
+ 'blue and purple polka dot triangle', 'blue and purple stripe block',
134
+ 'blue and purple stripe bowl', 'blue and purple stripe cross',
135
+ 'blue and purple stripe diamond', 'blue and purple stripe flower',
136
+ 'blue and purple stripe heart', 'blue and purple stripe hexagon',
137
+ 'blue and purple stripe l-shaped block', 'blue and purple stripe letter a',
138
+ 'blue and purple stripe letter e', 'blue and purple stripe letter g',
139
+ 'blue and purple stripe letter m', 'blue and purple stripe letter r',
140
+ 'blue and purple stripe letter t', 'blue and purple stripe letter v',
141
+ 'blue and purple stripe pallet', 'blue and purple stripe pentagon',
142
+ 'blue and purple stripe ring', 'blue and purple stripe round',
143
+ 'blue and purple stripe small block', 'blue and purple stripe square',
144
+ 'blue and purple stripe star', 'blue and purple stripe triangle',
145
+ 'blue bowl', 'blue container', 'blue cross', 'blue diamond', 'blue flower',
146
+ 'blue frame', 'blue heart', 'blue hexagon', 'blue l-shaped block',
147
+ 'blue letter a', 'blue letter e', 'blue letter g', 'blue letter m',
148
+ 'blue letter r', 'blue letter t', 'blue letter v', 'blue line',
149
+ 'blue paisley block', 'blue paisley bowl', 'blue paisley container',
150
+ 'blue paisley diamond', 'blue paisley flower', 'blue paisley frame',
151
+ 'blue paisley heart', 'blue paisley l-shaped block',
152
+ 'blue paisley letter a', 'blue paisley letter e', 'blue paisley letter m',
153
+ 'blue paisley letter r', 'blue paisley letter t', 'blue paisley letter v',
154
+ 'blue paisley line', 'blue paisley pallet', 'blue paisley pan',
155
+ 'blue paisley pentagon', 'blue paisley ring', 'blue paisley round',
156
+ 'blue paisley square', 'blue paisley star',
157
+ 'blue paisley three-sided rectangle', 'blue paisley triangle',
158
+ 'blue pallet', 'blue pan', 'blue pentagon', 'blue ring', 'blue round',
159
+ 'blue shorter block', 'blue square', 'blue star', 'blue swirl block',
160
+ 'blue swirl bowl', 'blue swirl container', 'blue swirl cross',
161
+ 'blue swirl diamond', 'blue swirl flower', 'blue swirl frame',
162
+ 'blue swirl heart', 'blue swirl hexagon', 'blue swirl l-shaped block',
163
+ 'blue swirl letter a', 'blue swirl letter e', 'blue swirl letter g',
164
+ 'blue swirl letter m', 'blue swirl letter t', 'blue swirl letter v',
165
+ 'blue swirl line', 'blue swirl pallet', 'blue swirl pan',
166
+ 'blue swirl pentagon', 'blue swirl ring', 'blue swirl round',
167
+ 'blue swirl square', 'blue swirl star', 'blue swirl three-sided rectangle',
168
+ 'blue swirl triangle', 'blue three-sided rectangle', 'blue triangle',
169
+ 'brick block', 'brick cross', 'brick diamond', 'brick flower',
170
+ 'brick heart', 'brick hexagon', 'brick l-shaped block', 'brick letter a',
171
+ 'brick letter e', 'brick letter g', 'brick letter m', 'brick letter r',
172
+ 'brick letter t', 'brick letter v', 'brick pentagon', 'brick ring',
173
+ 'brick round', 'brick shorter block', 'brick small block', 'brick star',
174
+ 'brick triangle', 'checkerboard block', 'checkerboard bowl',
175
+ 'checkerboard cross', 'checkerboard diamond', 'checkerboard flower',
176
+ 'checkerboard heart', 'checkerboard hexagon',
177
+ 'checkerboard l-shaped block', 'checkerboard letter a',
178
+ 'checkerboard letter e', 'checkerboard letter g', 'checkerboard letter m',
179
+ 'checkerboard letter r', 'checkerboard letter t', 'checkerboard letter v',
180
+ 'checkerboard pentagon', 'checkerboard ring', 'checkerboard round',
181
+ 'checkerboard shorter block', 'checkerboard small block',
182
+ 'checkerboard star', 'checkerboard triangle', 'cyan bowl',
183
+ 'cyan container', 'cyan cross', 'cyan diamond', 'cyan flower',
184
+ 'cyan frame', 'cyan heart', 'cyan hexagon', 'cyan l-shaped block',
185
+ 'cyan letter a', 'cyan letter e', 'cyan letter g', 'cyan letter m',
186
+ 'cyan letter r', 'cyan letter t', 'cyan letter v', 'cyan line',
187
+ 'cyan pallet', 'cyan pan', 'cyan pentagon', 'cyan ring', 'cyan round',
188
+ 'cyan shorter block', 'cyan square', 'cyan star',
189
+ 'cyan three-sided rectangle', 'cyan triangle', 'granite block',
190
+ 'granite cross', 'granite diamond', 'granite flower', 'granite heart',
191
+ 'granite hexagon', 'granite l-shaped block', 'granite letter a',
192
+ 'granite letter e', 'granite letter g', 'granite letter m',
193
+ 'granite letter r', 'granite letter t', 'granite letter v',
194
+ 'granite pentagon', 'granite ring', 'granite round',
195
+ 'granite shorter block', 'granite small block', 'granite star',
196
+ 'granite triangle', 'green and blue polka dot block',
197
+ 'green and blue polka dot bowl', 'green and blue polka dot container',
198
+ 'green and blue polka dot cross', 'green and blue polka dot frame',
199
+ 'green and blue polka dot heart', 'green and blue polka dot hexagon',
200
+ 'green and blue polka dot l-shaped block',
201
+ 'green and blue polka dot letter a', 'green and blue polka dot letter e',
202
+ 'green and blue polka dot letter g', 'green and blue polka dot letter m',
203
+ 'green and blue polka dot letter r', 'green and blue polka dot letter t',
204
+ 'green and blue polka dot letter v', 'green and blue polka dot line',
205
+ 'green and blue polka dot pallet', 'green and blue polka dot pan',
206
+ 'green and blue polka dot pentagon', 'green and blue polka dot ring',
207
+ 'green and blue polka dot round', 'green and blue polka dot shorter block',
208
+ 'green and blue polka dot square', 'green and blue polka dot star',
209
+ 'green and blue polka dot three-sided rectangle',
210
+ 'green and blue polka dot triangle', 'green and blue stripe block',
211
+ 'green and blue stripe cross', 'green and blue stripe diamond',
212
+ 'green and blue stripe flower', 'green and blue stripe heart',
213
+ 'green and blue stripe hexagon', 'green and blue stripe l-shaped block',
214
+ 'green and blue stripe letter a', 'green and blue stripe letter e',
215
+ 'green and blue stripe letter g', 'green and blue stripe letter m',
216
+ 'green and blue stripe letter r', 'green and blue stripe letter t',
217
+ 'green and blue stripe letter v', 'green and blue stripe pentagon',
218
+ 'green and blue stripe ring', 'green and blue stripe round',
219
+ 'green and blue stripe small block', 'green and blue stripe star',
220
+ 'green and blue stripe triangle', 'green and purple polka dot block',
221
+ 'green and purple polka dot bowl', 'green and purple polka dot container',
222
+ 'green and purple polka dot cross', 'green and purple polka dot diamond',
223
+ 'green and purple polka dot flower', 'green and purple polka dot frame',
224
+ 'green and purple polka dot heart', 'green and purple polka dot hexagon',
225
+ 'green and purple polka dot l-shaped block',
226
+ 'green and purple polka dot letter a',
227
+ 'green and purple polka dot letter e',
228
+ 'green and purple polka dot letter g',
229
+ 'green and purple polka dot letter m',
230
+ 'green and purple polka dot letter r',
231
+ 'green and purple polka dot letter t',
232
+ 'green and purple polka dot letter v', 'green and purple polka dot line',
233
+ 'green and purple polka dot pallet', 'green and purple polka dot pan',
234
+ 'green and purple polka dot pentagon', 'green and purple polka dot ring',
235
+ 'green and purple polka dot round',
236
+ 'green and purple polka dot shorter block',
237
+ 'green and purple polka dot square', 'green and purple polka dot star',
238
+ 'green and purple polka dot three-sided rectangle',
239
+ 'green and purple polka dot triangle', 'green and purple stripe block',
240
+ 'green and purple stripe container', 'green and purple stripe cross',
241
+ 'green and purple stripe diamond', 'green and purple stripe flower',
242
+ 'green and purple stripe heart', 'green and purple stripe hexagon',
243
+ 'green and purple stripe l-shaped block',
244
+ 'green and purple stripe letter a', 'green and purple stripe letter e',
245
+ 'green and purple stripe letter g', 'green and purple stripe letter m',
246
+ 'green and purple stripe letter r', 'green and purple stripe letter t',
247
+ 'green and purple stripe letter v', 'green and purple stripe pentagon',
248
+ 'green and purple stripe ring', 'green and purple stripe round',
249
+ 'green and purple stripe small block', 'green and purple stripe star',
250
+ 'green and purple stripe triangle', 'green bowl', 'green container',
251
+ 'green cross', 'green diamond', 'green flower', 'green frame',
252
+ 'green heart', 'green hexagon', 'green l-shaped block', 'green letter a',
253
+ 'green letter e', 'green letter g', 'green letter m', 'green letter r',
254
+ 'green letter t', 'green letter v', 'green line', 'green paisley block',
255
+ 'green paisley bowl', 'green paisley container', 'green paisley cross',
256
+ 'green paisley diamond', 'green paisley flower', 'green paisley frame',
257
+ 'green paisley heart', 'green paisley hexagon',
258
+ 'green paisley l-shaped block', 'green paisley letter a',
259
+ 'green paisley letter e', 'green paisley letter g',
260
+ 'green paisley letter m', 'green paisley letter r',
261
+ 'green paisley letter t', 'green paisley letter v', 'green paisley line',
262
+ 'green paisley pallet', 'green paisley pan', 'green paisley pentagon',
263
+ 'green paisley ring', 'green paisley round', 'green paisley square',
264
+ 'green paisley star', 'green paisley three-sided rectangle',
265
+ 'green paisley triangle', 'green pallet', 'green pan', 'green pentagon',
266
+ 'green ring', 'green round', 'green shorter block', 'green square',
267
+ 'green star', 'green swirl block', 'green swirl bowl',
268
+ 'green swirl container', 'green swirl cross', 'green swirl diamond',
269
+ 'green swirl flower', 'green swirl frame', 'green swirl heart',
270
+ 'green swirl hexagon', 'green swirl l-shaped block',
271
+ 'green swirl letter a', 'green swirl letter e', 'green swirl letter g',
272
+ 'green swirl letter m', 'green swirl letter r', 'green swirl letter t',
273
+ 'green swirl letter v', 'green swirl line', 'green swirl pallet',
274
+ 'green swirl pan', 'green swirl pentagon', 'green swirl round',
275
+ 'green swirl square', 'green swirl star',
276
+ 'green swirl three-sided rectangle', 'green swirl triangle',
277
+ 'green three-sided rectangle', 'green triangle', 'magma block',
278
+ 'magma cross', 'magma diamond', 'magma flower', 'magma heart',
279
+ 'magma hexagon', 'magma l-shaped block', 'magma letter a',
280
+ 'magma letter e', 'magma letter g', 'magma letter m', 'magma letter r',
281
+ 'magma letter t', 'magma letter v', 'magma pallet', 'magma pentagon',
282
+ 'magma ring', 'magma round', 'magma small block', 'magma star',
283
+ 'magma triangle', 'olive block', 'olive bowl', 'olive container',
284
+ 'olive cross', 'olive diamond', 'olive flower', 'olive frame',
285
+ 'olive heart', 'olive hexagon', 'olive l-shaped block', 'olive letter a',
286
+ 'olive letter e', 'olive letter g', 'olive letter m', 'olive letter r',
287
+ 'olive letter t', 'olive letter v', 'olive line', 'olive pallet',
288
+ 'olive pan', 'olive pentagon', 'olive ring', 'olive round',
289
+ 'olive shorter block', 'olive square', 'olive star',
290
+ 'olive three-sided rectangle', 'olive triangle', 'orange bowl',
291
+ 'orange container', 'orange cross', 'orange diamond', 'orange flower',
292
+ 'orange frame', 'orange heart', 'orange hexagon', 'orange l-shaped block',
293
+ 'orange letter a', 'orange letter e', 'orange letter g', 'orange letter m',
294
+ 'orange letter r', 'orange letter t', 'orange letter v', 'orange line',
295
+ 'orange pallet', 'orange pan', 'orange pentagon', 'orange ring',
296
+ 'orange round', 'orange shorter block', 'orange square', 'orange star',
297
+ 'orange three-sided rectangle', 'orange triangle', 'pink block',
298
+ 'pink bowl', 'pink container', 'pink cross', 'pink diamond', 'pink flower',
299
+ 'pink frame', 'pink heart', 'pink hexagon', 'pink l-shaped block',
300
+ 'pink letter a', 'pink letter e', 'pink letter g', 'pink letter m',
301
+ 'pink letter r', 'pink letter t', 'pink letter v', 'pink line',
302
+ 'pink pallet', 'pink pan', 'pink ring', 'pink round', 'pink shorter block',
303
+ 'pink square', 'pink three-sided rectangle', 'pink triangle',
304
+ 'plastic block', 'plastic cross', 'plastic diamond', 'plastic flower',
305
+ 'plastic heart', 'plastic hexagon', 'plastic l-shaped block',
306
+ 'plastic letter a', 'plastic letter e', 'plastic letter g',
307
+ 'plastic letter m', 'plastic letter r', 'plastic letter t',
308
+ 'plastic letter v', 'plastic pentagon', 'plastic ring', 'plastic round',
309
+ 'plastic shorter block', 'plastic small block', 'plastic star',
310
+ 'plastic triangle', 'polka dot block', 'polka dot bowl', 'polka dot cross',
311
+ 'polka dot diamond', 'polka dot flower', 'polka dot heart',
312
+ 'polka dot hexagon', 'polka dot l-shaped block', 'polka dot letter a',
313
+ 'polka dot letter e', 'polka dot letter g', 'polka dot letter m',
314
+ 'polka dot letter r', 'polka dot letter t', 'polka dot letter v',
315
+ 'polka dot pan', 'polka dot pentagon', 'polka dot ring', 'polka dot round',
316
+ 'polka dot small block', 'polka dot star', 'polka dot triangle',
317
+ 'purple block', 'purple bowl', 'purple container', 'purple cross',
318
+ 'purple diamond', 'purple flower', 'purple frame', 'purple heart',
319
+ 'purple hexagon', 'purple l-shaped block', 'purple letter a',
320
+ 'purple letter e', 'purple letter m', 'purple letter r', 'purple letter t',
321
+ 'purple letter v', 'purple line', 'purple paisley block',
322
+ 'purple paisley bowl', 'purple paisley container', 'purple paisley cross',
323
+ 'purple paisley diamond', 'purple paisley flower', 'purple paisley frame',
324
+ 'purple paisley heart', 'purple paisley hexagon',
325
+ 'purple paisley l-shaped block', 'purple paisley letter a',
326
+ 'purple paisley letter e', 'purple paisley letter g',
327
+ 'purple paisley letter m', 'purple paisley letter r',
328
+ 'purple paisley letter t', 'purple paisley letter v',
329
+ 'purple paisley line', 'purple paisley pallet', 'purple paisley pan',
330
+ 'purple paisley pentagon', 'purple paisley ring', 'purple paisley square',
331
+ 'purple paisley star', 'purple paisley three-sided rectangle',
332
+ 'purple paisley triangle', 'purple pallet', 'purple pan',
333
+ 'purple pentagon', 'purple ring', 'purple square', 'purple star',
334
+ 'purple swirl block', 'purple swirl bowl', 'purple swirl container',
335
+ 'purple swirl cross', 'purple swirl diamond', 'purple swirl flower',
336
+ 'purple swirl frame', 'purple swirl heart', 'purple swirl hexagon',
337
+ 'purple swirl l-shaped block', 'purple swirl letter a',
338
+ 'purple swirl letter e', 'purple swirl letter g', 'purple swirl letter m',
339
+ 'purple swirl letter r', 'purple swirl letter t', 'purple swirl letter v',
340
+ 'purple swirl line', 'purple swirl pallet', 'purple swirl pan',
341
+ 'purple swirl pentagon', 'purple swirl ring', 'purple swirl round',
342
+ 'purple swirl shorter block', 'purple swirl square', 'purple swirl star',
343
+ 'purple swirl three-sided rectangle', 'purple swirl triangle',
344
+ 'purple three-sided rectangle', 'purple triangle', 'rainbow block',
345
+ 'rainbow cross', 'rainbow diamond', 'rainbow flower', 'rainbow heart',
346
+ 'rainbow hexagon', 'rainbow l-shaped block', 'rainbow letter a',
347
+ 'rainbow letter e', 'rainbow letter g', 'rainbow letter m',
348
+ 'rainbow letter r', 'rainbow letter t', 'rainbow letter v',
349
+ 'rainbow pentagon', 'rainbow ring', 'rainbow round',
350
+ 'rainbow shorter block', 'rainbow small block', 'rainbow square',
351
+ 'rainbow star', 'rainbow triangle', 'red and blue polka dot block',
352
+ 'red and blue polka dot cross', 'red and blue polka dot diamond',
353
+ 'red and blue polka dot flower', 'red and blue polka dot heart',
354
+ 'red and blue polka dot hexagon', 'red and blue polka dot l-shaped block',
355
+ 'red and blue polka dot letter a', 'red and blue polka dot letter e',
356
+ 'red and blue polka dot letter g', 'red and blue polka dot letter m',
357
+ 'red and blue polka dot letter r', 'red and blue polka dot letter t',
358
+ 'red and blue polka dot letter v', 'red and blue polka dot pallet',
359
+ 'red and blue polka dot pentagon', 'red and blue polka dot ring',
360
+ 'red and blue polka dot round', 'red and blue polka dot shorter block',
361
+ 'red and blue polka dot small block', 'red and blue polka dot star',
362
+ 'red and blue polka dot triangle', 'red and blue stripe block',
363
+ 'red and blue stripe cross', 'red and blue stripe diamond',
364
+ 'red and blue stripe flower', 'red and blue stripe heart',
365
+ 'red and blue stripe hexagon', 'red and blue stripe l-shaped block',
366
+ 'red and blue stripe letter a', 'red and blue stripe letter e',
367
+ 'red and blue stripe letter g', 'red and blue stripe letter m',
368
+ 'red and blue stripe letter r', 'red and blue stripe letter t',
369
+ 'red and blue stripe letter v', 'red and blue stripe pentagon',
370
+ 'red and blue stripe ring', 'red and blue stripe round',
371
+ 'red and blue stripe small block', 'red and blue stripe star',
372
+ 'red and blue stripe triangle', 'red and green polka dot block',
373
+ 'red and green polka dot cross', 'red and green polka dot diamond',
374
+ 'red and green polka dot flower', 'red and green polka dot heart',
375
+ 'red and green polka dot hexagon', 'red and green polka dot l-shaped block',
376
+ 'red and green polka dot letter a', 'red and green polka dot letter e',
377
+ 'red and green polka dot letter g', 'red and green polka dot letter m',
378
+ 'red and green polka dot letter r', 'red and green polka dot letter t',
379
+ 'red and green polka dot letter v', 'red and green polka dot pan',
380
+ 'red and green polka dot pentagon', 'red and green polka dot ring',
381
+ 'red and green polka dot round', 'red and green polka dot small block',
382
+ 'red and green polka dot star', 'red and green polka dot triangle',
383
+ 'red and green stripe block', 'red and green stripe bowl',
384
+ 'red and green stripe cross', 'red and green stripe diamond',
385
+ 'red and green stripe flower', 'red and green stripe heart',
386
+ 'red and green stripe hexagon', 'red and green stripe l-shaped block',
387
+ 'red and green stripe letter a', 'red and green stripe letter e',
388
+ 'red and green stripe letter g', 'red and green stripe letter m',
389
+ 'red and green stripe letter r', 'red and green stripe letter t',
390
+ 'red and green stripe letter v', 'red and green stripe pentagon',
391
+ 'red and green stripe ring', 'red and green stripe round',
392
+ 'red and green stripe shorter block', 'red and green stripe small block',
393
+ 'red and green stripe star', 'red and green stripe triangle',
394
+ 'red and purple polka dot block', 'red and purple polka dot cross',
395
+ 'red and purple polka dot diamond', 'red and purple polka dot flower',
396
+ 'red and purple polka dot heart', 'red and purple polka dot hexagon',
397
+ 'red and purple polka dot l-shaped block',
398
+ 'red and purple polka dot letter a', 'red and purple polka dot letter e',
399
+ 'red and purple polka dot letter g', 'red and purple polka dot letter m',
400
+ 'red and purple polka dot letter r', 'red and purple polka dot letter t',
401
+ 'red and purple polka dot letter v', 'red and purple polka dot pentagon',
402
+ 'red and purple polka dot ring', 'red and purple polka dot round',
403
+ 'red and purple polka dot small block', 'red and purple polka dot star',
404
+ 'red and purple polka dot triangle', 'red and purple stripe block',
405
+ 'red and purple stripe cross', 'red and purple stripe diamond',
406
+ 'red and purple stripe flower', 'red and purple stripe heart',
407
+ 'red and purple stripe hexagon', 'red and purple stripe l-shaped block',
408
+ 'red and purple stripe letter a', 'red and purple stripe letter e',
409
+ 'red and purple stripe letter g', 'red and purple stripe letter m',
410
+ 'red and purple stripe letter r', 'red and purple stripe letter t',
411
+ 'red and purple stripe letter v', 'red and purple stripe pentagon',
412
+ 'red and purple stripe ring', 'red and purple stripe round',
413
+ 'red and purple stripe shorter block', 'red and purple stripe small block',
414
+ 'red and purple stripe star', 'red and purple stripe triangle',
415
+ 'red and yellow polka dot block', 'red and yellow polka dot cross',
416
+ 'red and yellow polka dot diamond', 'red and yellow polka dot flower',
417
+ 'red and yellow polka dot heart', 'red and yellow polka dot hexagon',
418
+ 'red and yellow polka dot l-shaped block',
419
+ 'red and yellow polka dot letter a', 'red and yellow polka dot letter e',
420
+ 'red and yellow polka dot letter g', 'red and yellow polka dot letter m',
421
+ 'red and yellow polka dot letter r', 'red and yellow polka dot letter t',
422
+ 'red and yellow polka dot letter v', 'red and yellow polka dot pentagon',
423
+ 'red and yellow polka dot ring', 'red and yellow polka dot round',
424
+ 'red and yellow polka dot shorter block',
425
+ 'red and yellow polka dot small block', 'red and yellow polka dot star',
426
+ 'red and yellow polka dot triangle', 'red and yellow stripe block',
427
+ 'red and yellow stripe bowl', 'red and yellow stripe cross',
428
+ 'red and yellow stripe diamond', 'red and yellow stripe flower',
429
+ 'red and yellow stripe heart', 'red and yellow stripe hexagon',
430
+ 'red and yellow stripe l-shaped block', 'red and yellow stripe letter a',
431
+ 'red and yellow stripe letter e', 'red and yellow stripe letter g',
432
+ 'red and yellow stripe letter m', 'red and yellow stripe letter r',
433
+ 'red and yellow stripe letter t', 'red and yellow stripe letter v',
434
+ 'red and yellow stripe pentagon', 'red and yellow stripe ring',
435
+ 'red and yellow stripe round', 'red and yellow stripe small block',
436
+ 'red and yellow stripe star', 'red and yellow stripe triangle',
437
+ 'red block', 'red bowl', 'red container', 'red cross', 'red diamond',
438
+ 'red flower', 'red frame', 'red heart', 'red hexagon',
439
+ 'red l-shaped block', 'red letter a', 'red letter e', 'red letter g',
440
+ 'red letter m', 'red letter r', 'red letter t', 'red letter v', 'red line',
441
+ 'red paisley block', 'red paisley bowl', 'red paisley container',
442
+ 'red paisley cross', 'red paisley diamond', 'red paisley flower',
443
+ 'red paisley frame', 'red paisley heart', 'red paisley hexagon',
444
+ 'red paisley l-shaped block', 'red paisley letter a',
445
+ 'red paisley letter e', 'red paisley letter g', 'red paisley letter m',
446
+ 'red paisley letter r', 'red paisley letter t', 'red paisley letter v',
447
+ 'red paisley line', 'red paisley pallet', 'red paisley pan',
448
+ 'red paisley pentagon', 'red paisley ring', 'red paisley round',
449
+ 'red paisley shorter block', 'red paisley square', 'red paisley star',
450
+ 'red paisley three-sided rectangle', 'red paisley triangle', 'red pallet',
451
+ 'red pan', 'red pentagon', 'red ring', 'red round', 'red shorter block',
452
+ 'red square', 'red star', 'red swirl block', 'red swirl bowl',
453
+ 'red swirl container', 'red swirl cross', 'red swirl diamond',
454
+ 'red swirl flower', 'red swirl frame', 'red swirl heart',
455
+ 'red swirl hexagon', 'red swirl l-shaped block', 'red swirl letter a',
456
+ 'red swirl letter e', 'red swirl letter g', 'red swirl letter m',
457
+ 'red swirl letter r', 'red swirl letter t', 'red swirl letter v',
458
+ 'red swirl line', 'red swirl pallet', 'red swirl pan', 'red swirl pentagon',
459
+ 'red swirl ring', 'red swirl round', 'red swirl shorter block',
460
+ 'red swirl square', 'red swirl star', 'red swirl three-sided rectangle',
461
+ 'red swirl triangle', 'red three-sided rectangle', 'red triangle',
462
+ 'tiger block', 'tiger cross', 'tiger diamond', 'tiger flower',
463
+ 'tiger heart', 'tiger hexagon', 'tiger l-shaped block', 'tiger letter a',
464
+ 'tiger letter e', 'tiger letter g', 'tiger letter m', 'tiger letter r',
465
+ 'tiger letter t', 'tiger letter v', 'tiger pentagon', 'tiger ring',
466
+ 'tiger round', 'tiger small block', 'tiger square', 'tiger star',
467
+ 'tiger triangle', 'tiles block', 'tiles cross', 'tiles diamond',
468
+ 'tiles flower', 'tiles heart', 'tiles hexagon', 'tiles l-shaped block',
469
+ 'tiles letter a', 'tiles letter e', 'tiles letter g', 'tiles letter m',
470
+ 'tiles letter r', 'tiles letter t', 'tiles letter v', 'tiles pentagon',
471
+ 'tiles ring', 'tiles round', 'tiles shorter block', 'tiles small block',
472
+ 'tiles square', 'tiles star', 'tiles triangle', 'wooden block',
473
+ 'wooden cross', 'wooden diamond', 'wooden flower', 'wooden heart',
474
+ 'wooden hexagon', 'wooden l-shaped block', 'wooden letter a',
475
+ 'wooden letter e', 'wooden letter g', 'wooden letter m', 'wooden letter r',
476
+ 'wooden letter t', 'wooden letter v', 'wooden pentagon', 'wooden ring',
477
+ 'wooden round', 'wooden shorter block', 'wooden small block',
478
+ 'wooden star', 'wooden triangle', 'yellow and blue polka dot block',
479
+ 'yellow and blue polka dot bowl', 'yellow and blue polka dot container',
480
+ 'yellow and blue polka dot cross', 'yellow and blue polka dot diamond',
481
+ 'yellow and blue polka dot flower', 'yellow and blue polka dot frame',
482
+ 'yellow and blue polka dot heart', 'yellow and blue polka dot hexagon',
483
+ 'yellow and blue polka dot l-shaped block',
484
+ 'yellow and blue polka dot letter a', 'yellow and blue polka dot letter e',
485
+ 'yellow and blue polka dot letter g', 'yellow and blue polka dot letter m',
486
+ 'yellow and blue polka dot letter r', 'yellow and blue polka dot letter t',
487
+ 'yellow and blue polka dot letter v', 'yellow and blue polka dot line',
488
+ 'yellow and blue polka dot pallet', 'yellow and blue polka dot pan',
489
+ 'yellow and blue polka dot pentagon', 'yellow and blue polka dot ring',
490
+ 'yellow and blue polka dot round', 'yellow and blue polka dot square',
491
+ 'yellow and blue polka dot star',
492
+ 'yellow and blue polka dot three-sided rectangle',
493
+ 'yellow and blue polka dot triangle', 'yellow and blue stripe block',
494
+ 'yellow and blue stripe cross', 'yellow and blue stripe diamond',
495
+ 'yellow and blue stripe flower', 'yellow and blue stripe heart',
496
+ 'yellow and blue stripe hexagon', 'yellow and blue stripe l-shaped block',
497
+ 'yellow and blue stripe letter a', 'yellow and blue stripe letter e',
498
+ 'yellow and blue stripe letter g', 'yellow and blue stripe letter m',
499
+ 'yellow and blue stripe letter t', 'yellow and blue stripe letter v',
500
+ 'yellow and blue stripe pan', 'yellow and blue stripe pentagon',
501
+ 'yellow and blue stripe ring', 'yellow and blue stripe round',
502
+ 'yellow and blue stripe small block', 'yellow and blue stripe star',
503
+ 'yellow and blue stripe triangle', 'yellow and green polka dot block',
504
+ 'yellow and green polka dot bowl', 'yellow and green polka dot container',
505
+ 'yellow and green polka dot cross', 'yellow and green polka dot diamond',
506
+ 'yellow and green polka dot flower', 'yellow and green polka dot frame',
507
+ 'yellow and green polka dot heart', 'yellow and green polka dot hexagon',
508
+ 'yellow and green polka dot l-shaped block',
509
+ 'yellow and green polka dot letter a',
510
+ 'yellow and green polka dot letter e',
511
+ 'yellow and green polka dot letter g',
512
+ 'yellow and green polka dot letter m',
513
+ 'yellow and green polka dot letter r',
514
+ 'yellow and green polka dot letter t',
515
+ 'yellow and green polka dot letter v', 'yellow and green polka dot line',
516
+ 'yellow and green polka dot pallet', 'yellow and green polka dot pan',
517
+ 'yellow and green polka dot pentagon', 'yellow and green polka dot ring',
518
+ 'yellow and green polka dot round',
519
+ 'yellow and green polka dot shorter block',
520
+ 'yellow and green polka dot square', 'yellow and green polka dot star',
521
+ 'yellow and green polka dot three-sided rectangle',
522
+ 'yellow and green stripe block', 'yellow and green stripe bowl',
523
+ 'yellow and green stripe cross', 'yellow and green stripe diamond',
524
+ 'yellow and green stripe flower', 'yellow and green stripe heart',
525
+ 'yellow and green stripe hexagon', 'yellow and green stripe l-shaped block',
526
+ 'yellow and green stripe letter a', 'yellow and green stripe letter e',
527
+ 'yellow and green stripe letter g', 'yellow and green stripe letter m',
528
+ 'yellow and green stripe letter r', 'yellow and green stripe letter t',
529
+ 'yellow and green stripe letter v', 'yellow and green stripe pentagon',
530
+ 'yellow and green stripe ring', 'yellow and green stripe round',
531
+ 'yellow and green stripe shorter block',
532
+ 'yellow and green stripe small block', 'yellow and green stripe star',
533
+ 'yellow and green stripe triangle', 'yellow and purple polka dot block',
534
+ 'yellow and purple polka dot bowl', 'yellow and purple polka dot container',
535
+ 'yellow and purple polka dot cross', 'yellow and purple polka dot diamond',
536
+ 'yellow and purple polka dot flower', 'yellow and purple polka dot frame',
537
+ 'yellow and purple polka dot heart', 'yellow and purple polka dot hexagon',
538
+ 'yellow and purple polka dot l-shaped block',
539
+ 'yellow and purple polka dot letter a',
540
+ 'yellow and purple polka dot letter e',
541
+ 'yellow and purple polka dot letter g',
542
+ 'yellow and purple polka dot letter m',
543
+ 'yellow and purple polka dot letter r',
544
+ 'yellow and purple polka dot letter t',
545
+ 'yellow and purple polka dot letter v', 'yellow and purple polka dot line',
546
+ 'yellow and purple polka dot pallet', 'yellow and purple polka dot pan',
547
+ 'yellow and purple polka dot pentagon', 'yellow and purple polka dot ring',
548
+ 'yellow and purple polka dot round',
549
+ 'yellow and purple polka dot shorter block',
550
+ 'yellow and purple polka dot square', 'yellow and purple polka dot star',
551
+ 'yellow and purple polka dot three-sided rectangle',
552
+ 'yellow and purple polka dot triangle', 'yellow and purple stripe block',
553
+ 'yellow and purple stripe bowl', 'yellow and purple stripe cross',
554
+ 'yellow and purple stripe diamond', 'yellow and purple stripe flower',
555
+ 'yellow and purple stripe heart', 'yellow and purple stripe hexagon',
556
+ 'yellow and purple stripe l-shaped block',
557
+ 'yellow and purple stripe letter a', 'yellow and purple stripe letter e',
558
+ 'yellow and purple stripe letter g', 'yellow and purple stripe letter m',
559
+ 'yellow and purple stripe letter r', 'yellow and purple stripe letter t',
560
+ 'yellow and purple stripe letter v', 'yellow and purple stripe pentagon',
561
+ 'yellow and purple stripe ring', 'yellow and purple stripe round',
562
+ 'yellow and purple stripe small block', 'yellow and purple stripe star',
563
+ 'yellow and purple stripe triangle', 'yellow bowl', 'yellow container',
564
+ 'yellow cross', 'yellow diamond', 'yellow flower', 'yellow frame',
565
+ 'yellow heart', 'yellow hexagon', 'yellow l-shaped block',
566
+ 'yellow letter e', 'yellow letter g', 'yellow letter m', 'yellow letter r',
567
+ 'yellow letter v', 'yellow line', 'yellow paisley block',
568
+ 'yellow paisley bowl', 'yellow paisley container', 'yellow paisley cross',
569
+ 'yellow paisley diamond', 'yellow paisley flower', 'yellow paisley frame',
570
+ 'yellow paisley heart', 'yellow paisley hexagon',
571
+ 'yellow paisley l-shaped block', 'yellow paisley letter a',
572
+ 'yellow paisley letter e', 'yellow paisley letter g',
573
+ 'yellow paisley letter m', 'yellow paisley letter r',
574
+ 'yellow paisley letter t', 'yellow paisley letter v', 'yellow paisley line',
575
+ 'yellow paisley pallet', 'yellow paisley pan', 'yellow paisley pentagon',
576
+ 'yellow paisley ring', 'yellow paisley round', 'yellow paisley square',
577
+ 'yellow paisley star', 'yellow paisley three-sided rectangle',
578
+ 'yellow paisley triangle', 'yellow pallet', 'yellow pan',
579
+ 'yellow pentagon', 'yellow ring', 'yellow round', 'yellow shorter block',
580
+ 'yellow square', 'yellow star', 'yellow swirl block', 'yellow swirl bowl',
581
+ 'yellow swirl container', 'yellow swirl cross', 'yellow swirl diamond',
582
+ 'yellow swirl flower', 'yellow swirl frame', 'yellow swirl heart',
583
+ 'yellow swirl hexagon', 'yellow swirl l-shaped block',
584
+ 'yellow swirl letter a', 'yellow swirl letter e', 'yellow swirl letter g',
585
+ 'yellow swirl letter m', 'yellow swirl letter r', 'yellow swirl letter t',
586
+ 'yellow swirl line', 'yellow swirl pallet', 'yellow swirl pan',
587
+ 'yellow swirl pentagon', 'yellow swirl ring', 'yellow swirl round',
588
+ 'yellow swirl shorter block', 'yellow swirl square', 'yellow swirl star',
589
+ 'yellow swirl three-sided rectangle', 'yellow swirl triangle',
590
+ 'yellow three-sided rectangle', 'yellow triangle'
591
+ ]
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8de4b52ad8e7782bd8415bea36307be62393edec8dfefad980bbd2a9f29ca261
3
- size 1199415991
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fedf9c9fb99f394ea159a963a029628839bc50589f9c1c172683c433ff76359c
3
+ size 691937450
tokenizer.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": "1.0",
3
+ "truncation": {
4
+ "max_length": 512,
5
+ "stride": 0,
6
+ "strategy": "longest_first"
7
+ },
8
+ "padding": {
9
+ "strategy": "longest",
10
+ "pad_to_multiple_of": null
11
+ }
12
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tokenizer_class": "BertTokenizer",
3
+ "model_max_length": 512,
4
+ "padding_side": "right",
5
+ "truncation": true,
6
+ "do_lower_case": true,
7
+ "vocab_size": 30522
8
+ }