firework8 commited on
Commit
71ca1c1
·
verified ·
1 Parent(s): fd62532

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