Upload 5 files
Browse files
ntu60_xview/b_1/20231210_182658.log
ADDED
The diff for this file is too large to render.
See raw diff
|
|
ntu60_xview/b_1/20231210_182658.log.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
ntu60_xview/b_1/b_1.py
ADDED
@@ -0,0 +1,93 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
modality = 'b'
|
2 |
+
graph = 'nturgb+d'
|
3 |
+
work_dir = './work_dirs/test_prototype/ntu60_xview/b_1'
|
4 |
+
model = dict(
|
5 |
+
type='RecognizerGCN_7_1_1',
|
6 |
+
backbone=dict(
|
7 |
+
type='GCN_7_1_1',
|
8 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
9 |
+
graph_cfg=dict(
|
10 |
+
layout='nturgb+d',
|
11 |
+
mode='random',
|
12 |
+
num_filter=8,
|
13 |
+
init_off=0.04,
|
14 |
+
init_std=0.02)),
|
15 |
+
cls_head=dict(type='SimpleHead_7_4_3', num_classes=60, in_channels=384))
|
16 |
+
dataset_type = 'PoseDataset'
|
17 |
+
ann_file = '/data/lhd/pyskl_data/nturgbd/ntu60_3danno.pkl'
|
18 |
+
train_pipeline = [
|
19 |
+
dict(type='PreNormalize3D', align_spine=False),
|
20 |
+
dict(type='RandomRot', theta=0.2),
|
21 |
+
dict(type='GenSkeFeat', feats=['b']),
|
22 |
+
dict(type='UniformSampleDecode', clip_len=100),
|
23 |
+
dict(type='FormatGCNInput'),
|
24 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
25 |
+
dict(type='ToTensor', keys=['keypoint'])
|
26 |
+
]
|
27 |
+
val_pipeline = [
|
28 |
+
dict(type='PreNormalize3D', align_spine=False),
|
29 |
+
dict(type='GenSkeFeat', feats=['b']),
|
30 |
+
dict(type='UniformSampleDecode', clip_len=100, num_clips=1),
|
31 |
+
dict(type='FormatGCNInput'),
|
32 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
33 |
+
dict(type='ToTensor', keys=['keypoint'])
|
34 |
+
]
|
35 |
+
test_pipeline = [
|
36 |
+
dict(type='PreNormalize3D', align_spine=False),
|
37 |
+
dict(type='GenSkeFeat', feats=['b']),
|
38 |
+
dict(type='UniformSampleDecode', clip_len=100, num_clips=10),
|
39 |
+
dict(type='FormatGCNInput'),
|
40 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
41 |
+
dict(type='ToTensor', keys=['keypoint'])
|
42 |
+
]
|
43 |
+
data = dict(
|
44 |
+
videos_per_gpu=16,
|
45 |
+
workers_per_gpu=4,
|
46 |
+
test_dataloader=dict(videos_per_gpu=1),
|
47 |
+
train=dict(
|
48 |
+
type='PoseDataset',
|
49 |
+
ann_file='/data/lhd/pyskl_data/nturgbd/ntu60_3danno.pkl',
|
50 |
+
pipeline=[
|
51 |
+
dict(type='PreNormalize3D', align_spine=False),
|
52 |
+
dict(type='RandomRot', theta=0.2),
|
53 |
+
dict(type='GenSkeFeat', feats=['b']),
|
54 |
+
dict(type='UniformSampleDecode', clip_len=100),
|
55 |
+
dict(type='FormatGCNInput'),
|
56 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
57 |
+
dict(type='ToTensor', keys=['keypoint'])
|
58 |
+
],
|
59 |
+
split='xview_train'),
|
60 |
+
val=dict(
|
61 |
+
type='PoseDataset',
|
62 |
+
ann_file='/data/lhd/pyskl_data/nturgbd/ntu60_3danno.pkl',
|
63 |
+
pipeline=[
|
64 |
+
dict(type='PreNormalize3D', align_spine=False),
|
65 |
+
dict(type='GenSkeFeat', feats=['b']),
|
66 |
+
dict(type='UniformSampleDecode', clip_len=100, num_clips=1),
|
67 |
+
dict(type='FormatGCNInput'),
|
68 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
69 |
+
dict(type='ToTensor', keys=['keypoint'])
|
70 |
+
],
|
71 |
+
split='xview_val'),
|
72 |
+
test=dict(
|
73 |
+
type='PoseDataset',
|
74 |
+
ann_file='/data/lhd/pyskl_data/nturgbd/ntu60_3danno.pkl',
|
75 |
+
pipeline=[
|
76 |
+
dict(type='PreNormalize3D', align_spine=False),
|
77 |
+
dict(type='GenSkeFeat', feats=['b']),
|
78 |
+
dict(type='UniformSampleDecode', clip_len=100, num_clips=10),
|
79 |
+
dict(type='FormatGCNInput'),
|
80 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
81 |
+
dict(type='ToTensor', keys=['keypoint'])
|
82 |
+
],
|
83 |
+
split='xview_val'))
|
84 |
+
optimizer = dict(
|
85 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
86 |
+
optimizer_config = dict(grad_clip=None)
|
87 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
88 |
+
total_epochs = 150
|
89 |
+
checkpoint_config = dict(interval=1)
|
90 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
91 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
92 |
+
dist_params = dict(backend='nccl')
|
93 |
+
gpu_ids = range(0, 1)
|
ntu60_xview/b_1/best_pred.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3e3a6e3c1e5300187beec881e556b1a65b483abb069045c88f273395eb910249
|
3 |
+
size 7620756
|
ntu60_xview/b_1/best_top1_acc_epoch_148.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:42b0b0efba2c3b406b411da09fba57e51089c3a1b06b60167c5d66842c53d732
|
3 |
+
size 33472486
|