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
|