manh6054 commited on
Commit
d317543
·
verified ·
1 Parent(s): 1dc0e89

Upload 20 files

Browse files
checkpoint/avav2.2/K23/config.yaml ADDED
@@ -0,0 +1,370 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ BACKBONE2D:
2
+ YOLOv8:
3
+ ver : n
4
+
5
+ PRETRAIN:
6
+ n : weights/backbone2D/YOLOv8/v8_n.pth
7
+ s : weights/backbone2D/YOLOv8/v8_s.pth
8
+ m : weights/backbone2D/YOLOv8/v8_m.pth
9
+ l : weights/backbone2D/YOLOv8/v8_l.pth
10
+ x : weights/backbone2D/YOLOv8/v8_x.pth
11
+
12
+ BACKBONE3D:
13
+
14
+ MOBILENET:
15
+ width_mult: 2.0
16
+
17
+ PRETRAIN:
18
+ width_mult_0.5x : weights/backbone3D/mobilenet/kinetics_mobilenet_0.5x_RGB_16_best.pth
19
+ width_mult_1.0x : weights/backbone3D/mobilenet/kinetics_mobilenet_1.0x_RGB_16_best.pth
20
+ width_mult_1.5x : weights/backbone3D/mobilenet/kinetics_mobilenet_1.5x_RGB_16_best.pth
21
+ width_mult_2.0x : weights/backbone3D/mobilenet/kinetics_mobilenet_2.0x_RGB_16_best.pth
22
+
23
+ MOBILENETv2:
24
+ width_mult: 1.0
25
+
26
+ PRETRAIN:
27
+ width_mult_0.2x : weights/backbone3D/mobilenetv2/kinetics_mobilenetv2_0.2x_RGB_16_best.pth
28
+ width_mult_0.45x : weights/backbone3D/mobilenetv2/kinetics_mobilenetv2_0.45x_RGB_16_best.pth
29
+ width_mult_0.7x : weights/backbone3D/mobilenetv2/kinetics_mobilenetv2_0.7x_RGB_16_best.pth
30
+ width_mult_1.0x : weights/backbone3D/mobilenetv2/kinetics_mobilenetv2_1.0x_RGB_16_best.pth
31
+
32
+ SHUFFLENET:
33
+ width_mult: 2.0
34
+
35
+ PRETRAIN:
36
+ width_mult_0.5x : weights/backbone3D/shufflenet/kinetics_shufflenet_0.5x_G3_RGB_16_best.pth
37
+ width_mult_1.0x : weights/backbone3D/shufflenet/kinetics_shufflenet_1.0x_G3_RGB_16_best.pth
38
+ width_mult_1.5x : weights/backbone3D/shufflenet/kinetics_shufflenet_1.5x_G3_RGB_16_best.pth
39
+ width_mult_2.0x : weights/backbone3D/shufflenet/kinetics_shufflenet_2.0x_G3_RGB_16_best.pth
40
+
41
+ SHUFFLENETv2:
42
+ width_mult: 2.0
43
+
44
+ PRETRAIN:
45
+ width_mult_0.25x : weights/backbone3D/shufflenetv2/kinetics_shufflenetv2_0.25x_RGB_16_best.pth
46
+ width_mult_1.0x : weights/backbone3D/shufflenetv2/kinetics_shufflenetv2_1.0x_RGB_16_best.pth
47
+ width_mult_1.5x : weights/backbone3D/shufflenetv2/kinetics_shufflenetv2_1.5x_RGB_16_best.pth
48
+ width_mult_2.0x : weights/backbone3D/shufflenetv2/kinetics_shufflenetv2_2.0x_RGB_16_best.pth
49
+
50
+ I3D:
51
+ PRETRAIN:
52
+ default: weights/backbone3D/I3D/rgb_imagenet.pth
53
+
54
+ RESNET:
55
+ ver : 101
56
+
57
+ PRETRAIN:
58
+ ver_18 : weights/backbone3D/resnet/kinetics_resnet_18_RGB_16_best.pth
59
+ ver_50 : weights/backbone3D/resnet/kinetics_resnet_50_RGB_16_best.pth
60
+ ver_101: weights/backbone3D/resnet/kinetics_resnet_101_RGB_16_best.pth
61
+
62
+ RESNEXT:
63
+ ver : 101
64
+
65
+ PRETRAIN:
66
+ ver_101 : weights/backbone3D/resnext/resnext-101-kinetics.pth
67
+
68
+ LOSS:
69
+ TAL:
70
+ top_k : 10
71
+ alpha : 0.5
72
+ beta : 6.0
73
+ radius : 2.5
74
+ soft_label : False
75
+
76
+ scale_cls_loss : 0.5
77
+ scale_box_loss : 7.5
78
+ scale_dfl_loss : 1.5
79
+
80
+ SIMOTA:
81
+ top_k : 10
82
+ gamma : 0.5
83
+ radius : 2.5
84
+ mode : unbalance
85
+ soft_label : True
86
+
87
+ dynamic_k : False
88
+ dynamic_top_k : 10
89
+
90
+ scale_cls_loss : 0.5
91
+ scale_box_loss : 5.0
92
+ scale_dfl_loss : 0.5
93
+
94
+ config_path : config/ava_config.yaml
95
+ dataset : ava
96
+ loss : simota
97
+ active_checker : True
98
+ num_classes : 80
99
+ backbone2D : yolov8
100
+ backbone3D : shufflenetv2
101
+ fusion_module : CFAM
102
+ mode : decoupled
103
+ interchannels : [64, 64, 64]
104
+ pretrain_path : null
105
+ data_root : /home/manh/Datasets/AVA/AVA_Dataset
106
+ img_size : 224
107
+ clip_length : 16
108
+ batch_size : 8
109
+ num_workers : 6
110
+ acc_grad : 16
111
+ lr : 0.0001
112
+ weight_decay : 0.0005
113
+ max_step_warmup : 800
114
+ adjustlr_schedule : [3, 4, 5, 6, 7]
115
+ max_epoch : 9
116
+ lr_decay : 0.5
117
+ save_folder : weights/model_checkpoint
118
+ sampling_rate : 1
119
+ labelmap : evaluator/Evaluation/ava_action_list_v2.2_for_activitynet_2019.pbtxt
120
+ groundtruth : evaluator/Evaluation/ava_val_v2.2.csv
121
+ detections : ava_result/ava_predicted_file.csv
122
+
123
+
124
+ idx2name:
125
+ 0 : bend/bow (at the waist)
126
+ 1 : crawl
127
+ 2 : crouch/kneel
128
+ 3 : dance
129
+ 4 : fall down
130
+ 5 : get up
131
+ 6 : jump/leap
132
+ 7 : lie/sleep
133
+ 8 : martial art
134
+ 9 : run/jog
135
+ 10 : sit
136
+ 11 : stand
137
+ 12 : swim
138
+ 13 : walk
139
+ 14 : answer phone
140
+ 15 : brush teeth
141
+ 16 : carry/hold (an object)
142
+ 17 : catch (an object)
143
+ 18 : chop
144
+ 19 : climb (e.g., a mountain)
145
+ 20 : clink glass
146
+ 21 : close (e.g., a door, a box)
147
+ 22 : cook
148
+ 23 : cut
149
+ 24 : dig
150
+ 25 : dress/put on clothing
151
+ 26 : drink
152
+ 27 : drive (e.g., a car, a truck)
153
+ 28 : eat
154
+ 29 : enter
155
+ 30 : exit
156
+ 31 : extract
157
+ 32 : fishing
158
+ 33 : hit (an object)
159
+ 34 : kick (an object)
160
+ 35 : lift/pick up
161
+ 36 : listen (e.g., to music)
162
+ 37 : open (e.g., a window, a car door)
163
+ 38 : paint
164
+ 39 : play board game
165
+ 40 : play musical instrument
166
+ 41 : play with pets
167
+ 42 : point to (an object)
168
+ 43 : press
169
+ 44 : pull (an object)
170
+ 45 : push (an object)
171
+ 46 : put down
172
+ 47 : read
173
+ 48 : ride (e.g., a bike, a car, a horse)
174
+ 49 : row boat
175
+ 50 : sail boat
176
+ 51 : shoot
177
+ 52 : shovel
178
+ 53 : smoke
179
+ 54 : stir
180
+ 55 : take a photo
181
+ 56 : text on/look at a cellphone
182
+ 57 : throw
183
+ 58 : touch (an object)
184
+ 59 : turn (e.g., a screwdriver)
185
+ 60 : watch (e.g., TV)
186
+ 61 : work on a computer
187
+ 62 : write
188
+ 63 : fight/hit (a person)
189
+ 64 : give/serve (an object) to (a person)
190
+ 65 : grab (a person)
191
+ 66 : hand clap
192
+ 67 : hand shake
193
+ 68 : hand wave
194
+ 69 : hug (a person)
195
+ 70 : kick (a person)
196
+ 71 : kiss (a person)
197
+ 72 : lift (a person)
198
+ 73 : listen to (a person)
199
+ 74 : play with kids
200
+ 75 : push (another person)
201
+ 76 : sing to (e.g., self, a person, a group)
202
+ 77 : take (an object) from (a person)
203
+ 78 : talk to (e.g., self, a person, a group)
204
+ 79 : watch (a person)
205
+
206
+
207
+ train_class_count:
208
+ 0 : 8322
209
+ 1 : 157
210
+ 2 : 2309
211
+ 3 : 3239
212
+ 4 : 291
213
+ 5 : 1120
214
+ 6 : 245
215
+ 7 : 5349
216
+ 8 : 2102
217
+ 9 : 3308
218
+ 10 : 99884
219
+ 11 : 165387
220
+ 12 : 110
221
+ 13 : 40341
222
+ 14 : 3275
223
+ 15 : 20
224
+ 16 : 80025
225
+ 17 : 80
226
+ 18 : 36
227
+ 19 : 267
228
+ 20 : 53
229
+ 21 : 784
230
+ 22 : 65
231
+ 23 : 184
232
+ 24 : 41
233
+ 25 : 342
234
+ 26 : 2319
235
+ 27 : 1187
236
+ 28 : 3012
237
+ 29 : 221
238
+ 30 : 53
239
+ 31 : 7
240
+ 32 : 79
241
+ 33 : 177
242
+ 34 : 35
243
+ 35 : 518
244
+ 36 : 660
245
+ 37 : 1247
246
+ 38 : 67
247
+ 39 : 117
248
+ 40 : 1290
249
+ 41 : 138
250
+ 42 : 97
251
+ 43 : 94
252
+ 44 : 341
253
+ 45 : 385
254
+ 46 : 533
255
+ 47 : 2141
256
+ 48 : 4784
257
+ 49 : 62
258
+ 50 : 712
259
+ 51 : 219
260
+ 52 : 67
261
+ 53 : 2986
262
+ 54 : 59
263
+ 55 : 180
264
+ 56 : 415
265
+ 57 : 248
266
+ 58 : 17077
267
+ 59 : 135
268
+ 60 : 1983
269
+ 61 : 176
270
+ 62 : 773
271
+ 63 : 2691
272
+ 64 : 1073
273
+ 65 : 1993
274
+ 66 : 1175
275
+ 67 : 619
276
+ 68 : 343
277
+ 69 : 1101
278
+ 70 : 50
279
+ 71 : 714
280
+ 72 : 400
281
+ 73 : 106175
282
+ 74 : 127
283
+ 75 : 355
284
+ 76 : 1638
285
+ 77 : 608
286
+ 78 : 110151
287
+ 79 : 167115
288
+
289
+
290
+ class_ratio:
291
+ 79: 0.5059319458527529
292
+ 11: 0.5005430972490094
293
+ 78: 0.3317767554377424
294
+ 73: 0.32139321745252786
295
+ 10: 0.3018567607333167
296
+ 16: 0.24206491290886498
297
+ 13: 0.12267378359756043
298
+ 58: 0.05155061034502472
299
+ 0: 0.02512088050957867
300
+ 7: 0.016115395377806128
301
+ 48: 0.014466546112115732
302
+ 9: 0.010040529196366514
303
+ 14: 0.00986601595291753
304
+ 3: 0.00982990976461774
305
+ 28: 0.009101768300571982
306
+ 53: 0.008999467433722578
307
+ 63: 0.008108848122327766
308
+ 26: 0.007025662473334075
309
+ 2: 0.006983538586984321
310
+ 47: 0.006456990007612388
311
+ 8: 0.006369733385887896
312
+ 65: 0.006026724597039895
313
+ 60: 0.0059966361067900695
314
+ 76: 0.004943538948046204
315
+ 40: 0.003902477185402268
316
+ 37: 0.0037640701302530744
317
+ 27: 0.0035745126416791783
318
+ 66: 0.003571503792654196
319
+ 5: 0.0033819463040803003
320
+ 69: 0.0033187604745556682
321
+ 64: 0.003228495003806194
322
+ 21: 0.0023649553336362243
323
+ 62: 0.002337875692411382
324
+ 50: 0.0021633624489623983
325
+ 71: 0.002148318203837486
326
+ 36: 0.0020099111486882922
327
+ 67: 0.001862477546464151
328
+ 77: 0.0018293802071893438
329
+ 46: 0.0016037165303156584
330
+ 35: 0.0015615926439659037
331
+ 56: 0.0012486723453677265
332
+ 72: 0.0012035396099929893
333
+ 45: 0.0011644245726682172
334
+ 75: 0.0010681414038687781
335
+ 68: 0.001056106007768848
336
+ 44: 0.001035044064593971
337
+ 25: 0.001029026366544006
338
+ 4: 0.0008755750662698998
339
+ 19: 0.0008063715386953028
340
+ 57: 0.0007492034072206358
341
+ 6: 0.000737168011120706
342
+ 29: 0.0006769910306210565
343
+ 51: 0.0006679644835461091
344
+ 23: 0.0005536282205967751
345
+ 55: 0.0005446016735218277
346
+ 33: 0.0005325662774218978
347
+ 61: 0.0005295574283969153
348
+ 1: 0.0004723892969222483
349
+ 41: 0.0004152211654475813
350
+ 59: 0.00041221231642259886
351
+ 74: 0.00038212382617277413
352
+ 39: 0.00035504418494793186
353
+ 12: 0.00033398224177305457
354
+ 42: 0.00029185835542329993
355
+ 43: 0.0002828318083483525
356
+ 17: 0.00024070792199859788
357
+ 32: 0.0002376990729736154
358
+ 38: 0.0002015928846738257
359
+ 52: 0.0002015928846738257
360
+ 22: 0.00019557518662386078
361
+ 49: 0.00018654863954891334
362
+ 54: 0.00018053094149894842
363
+ 20: 0.0001594689983240711
364
+ 30: 0.0001594689983240711
365
+ 70: 0.00015044245124912366
366
+ 24: 0.00012336281002428142
367
+ 18: 0.00010831856489936904
368
+ 34: 0.00010530971587438658
369
+ 15: 6.017698049964947e-05
370
+ 31: 2.1061943174877314e-05
checkpoint/avav2.2/K23/ema_epoch_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30c3cde0f17af715ec6186f1a861b17233c3f4538d0dacf870ccbd01faeeb7b8
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33afe6b16990c4506f421d9af2c72d6c2bcddac0c050e2569f52b63b7e9c673c
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d926cde52343405216f3e97b107dccd3c1765cef008b4f637814545b6386c837
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0e7bf6a0c7eb7b089051dc909153dd6b7f03b4036f2a2b404c9ba1922150afd
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_5.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:916e59c3d6ad114c685a1822f855796182180c58c949761a8c40a8df1c73bb4b
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_6.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00ba43f3e8e4e048c72ba95fa4ce80690d01e53ec9df9ce24e9117cf1c0333b5
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_7.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23f975d70903b226d7d59a516170a7174536ea0753ee7750acc4f21229418a17
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_8.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13750426472106c9b69da1f2ac67057d7f525a918fc130b3b3947881676aca85
3
+ size 41333100
checkpoint/avav2.2/K23/ema_epoch_9.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8128f93553890a7dd062212b8719d974e640a60c8a406b6612b32bf2f7a857ae
3
+ size 41333100
checkpoint/avav2.2/K23/epoch_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcc7baba963b299d51886d2ef225798c9d32bd1732c735ab3c19d01314771728
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0842dcafda54cdacd5e72b623056804681c51019b256d634268513428f1d8c9
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:225526ff304606cd63585ee8a934ea82afac24b596cdbd11f0c0673babb477cb
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:652995b4b5bd57050ca7beae7ada44dce0336f491b30764a2e8ee0496623f5e2
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_5.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d976c412d3f918c07ddce9b0c61176f0b2768e75a233445fe319ac99412dd06
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_6.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96abee845dda7c0760966a0b14e51937a33b444d189a422316fa412c01e4d86c
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_7.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3f4fa94a81dd204c202a5da9608b26e50ff40fc2122e883c4570549e33f3d9d
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_8.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82e318e2865bab83e722e4ddfe491cbd384b14ab17f96c87004aa0e79598e3c9
3
+ size 41327296
checkpoint/avav2.2/K23/epoch_9.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5280f95e9fecb58e33237c9a8de6b714d4634a2e1ebafb9cc009c44c5351d53a
3
+ size 41327296
checkpoint/avav2.2/K23/logging.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ epoch : 9, update : 1440, loss = 3.6678619980812073