Upload 122 files
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- pku_mmd_xsub/b_1/20250702_012820.log +0 -0
- pku_mmd_xsub/b_1/20250702_012820.log.json +0 -0
- pku_mmd_xsub/b_1/b_1.py +98 -0
- pku_mmd_xsub/b_1/best_pred.pkl +3 -0
- pku_mmd_xsub/b_1/best_top1_acc_epoch_134.pth +3 -0
- pku_mmd_xsub/b_2/20250702_012904.log +0 -0
- pku_mmd_xsub/b_2/20250702_012904.log.json +0 -0
- pku_mmd_xsub/b_2/b_2.py +98 -0
- pku_mmd_xsub/b_2/best_pred.pkl +3 -0
- pku_mmd_xsub/b_2/best_top1_acc_epoch_137.pth +3 -0
- pku_mmd_xsub/b_3/20250702_012939.log +0 -0
- pku_mmd_xsub/b_3/20250702_012939.log.json +0 -0
- pku_mmd_xsub/b_3/b_3.py +98 -0
- pku_mmd_xsub/b_3/best_pred.pkl +3 -0
- pku_mmd_xsub/b_3/best_top1_acc_epoch_140.pth +3 -0
- pku_mmd_xsub/bm/20250702_121022.log +0 -0
- pku_mmd_xsub/bm/20250702_121022.log.json +0 -0
- pku_mmd_xsub/bm/best_pred.pkl +3 -0
- pku_mmd_xsub/bm/best_top1_acc_epoch_150.pth +3 -0
- pku_mmd_xsub/bm/bm.py +98 -0
- pku_mmd_xsub/j_1/20250702_013020.log +0 -0
- pku_mmd_xsub/j_1/20250702_013020.log.json +0 -0
- pku_mmd_xsub/j_1/best_pred.pkl +3 -0
- pku_mmd_xsub/j_1/best_top1_acc_epoch_147.pth +3 -0
- pku_mmd_xsub/j_1/j_1.py +98 -0
- pku_mmd_xsub/j_2/20250702_013047.log +0 -0
- pku_mmd_xsub/j_2/20250702_013047.log.json +0 -0
- pku_mmd_xsub/j_2/best_pred.pkl +3 -0
- pku_mmd_xsub/j_2/best_top1_acc_epoch_136.pth +3 -0
- pku_mmd_xsub/j_2/j_2.py +98 -0
- pku_mmd_xsub/j_3/20250702_013116.log +0 -0
- pku_mmd_xsub/j_3/20250702_013116.log.json +0 -0
- pku_mmd_xsub/j_3/best_pred.pkl +3 -0
- pku_mmd_xsub/j_3/best_top1_acc_epoch_137.pth +3 -0
- pku_mmd_xsub/j_3/j_3.py +98 -0
- pku_mmd_xsub/jm/20250702_121040.log +0 -0
- pku_mmd_xsub/jm/20250702_121040.log.json +0 -0
- pku_mmd_xsub/jm/best_pred.pkl +3 -0
- pku_mmd_xsub/jm/best_top1_acc_epoch_149.pth +3 -0
- pku_mmd_xsub/jm/jm.py +98 -0
- pku_mmd_xsub/k_1/20250702_120945.log +0 -0
- pku_mmd_xsub/k_1/20250702_120945.log.json +0 -0
- pku_mmd_xsub/k_1/best_pred.pkl +3 -0
- pku_mmd_xsub/k_1/best_top1_acc_epoch_150.pth +3 -0
- pku_mmd_xsub/k_1/k_1.py +98 -0
- pku_mmd_xsub/k_2/20250702_120831.log +0 -0
- pku_mmd_xsub/k_2/20250702_120831.log.json +0 -0
- pku_mmd_xsub/k_2/best_pred.pkl +3 -0
- pku_mmd_xsub/k_2/best_top1_acc_epoch_147.pth +3 -0
- pku_mmd_xsub/k_2/k_2.py +98 -0
pku_mmd_xsub/b_1/20250702_012820.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/b_1/20250702_012820.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/b_1/b_1.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'b'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/b_1'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/b_1/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4623033271eca566d26f073f0353e73158f1d67b5999cce0de4d3621bd28480a
|
| 3 |
+
size 954529
|
pku_mmd_xsub/b_1/best_top1_acc_epoch_134.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:80156b77c6de6fcf23339fe3ba9c96d7ae5c4092ed3569d0a4054632e47d55f5
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/b_2/20250702_012904.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/b_2/20250702_012904.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/b_2/b_2.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'b'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/b_2'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/b_2/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:27fd653ae282f9c1f119beb0d2972ac7ecef64985279484f122b58bfc06eb4a3
|
| 3 |
+
size 954783
|
pku_mmd_xsub/b_2/best_top1_acc_epoch_137.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1ab7c58bcb72226548d105511d2040e0c0d923e330da7da5fcb4207388888126
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/b_3/20250702_012939.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/b_3/20250702_012939.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/b_3/b_3.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'b'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/b_3'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['b']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/b_3/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e58f90159e8b0048eb483720366db04581e0e9e188847703b979ebabec0cb0d2
|
| 3 |
+
size 954677
|
pku_mmd_xsub/b_3/best_top1_acc_epoch_140.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:257b1b060a7ffce62e6bc1def7d44ae954baff7a8c15f3c49322896764674922
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/bm/20250702_121022.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/bm/20250702_121022.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/bm/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aa36d0ddfe4545c5831adbdcb516c0d784bbdde16b48f5e3c8845d21c762821f
|
| 3 |
+
size 954613
|
pku_mmd_xsub/bm/best_top1_acc_epoch_150.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:66ce579086538afb7a90eebd586182f6b7546e7710e3ad023ea198d4e8dfb8e6
|
| 3 |
+
size 16576377
|
pku_mmd_xsub/bm/bm.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'bm'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/bm'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['bm']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['bm']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['bm']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['bm']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['bm']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['bm']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/j_1/20250702_013020.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/j_1/20250702_013020.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/j_1/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b89de9a2daf9f78db0141371e8c15c95dd309035df79a9eedf2f19d98aedaa16
|
| 3 |
+
size 954315
|
pku_mmd_xsub/j_1/best_top1_acc_epoch_147.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d48c347b7af20a80d760ae0fe8432236e5832e8e9ca92b070cd8e86ba47c6acf
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/j_1/j_1.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'j'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/j_1'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/j_2/20250702_013047.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/j_2/20250702_013047.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/j_2/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f55342774df364b53249a74f4fd3fd0879c31721440f4dc6b57f7eb583a809f9
|
| 3 |
+
size 954009
|
pku_mmd_xsub/j_2/best_top1_acc_epoch_136.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:42fc7bcf352b4675e5fb0d0dc5d3185f8209aef3c52c1e8c22ed087cc6e3deba
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/j_2/j_2.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'j'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/j_2'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/j_3/20250702_013116.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/j_3/20250702_013116.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/j_3/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2e019bd09ca7e5691d088b36a06a7eb766f94035f548aa7dd042e618d3f4b4f0
|
| 3 |
+
size 954535
|
pku_mmd_xsub/j_3/best_top1_acc_epoch_137.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fc2139d5fa3f0dbd052d4e3f4554c0474c1d4b60ce76bdb44c60d763352c5188
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/j_3/j_3.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'j'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/j_3'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['j']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/jm/20250702_121040.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/jm/20250702_121040.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/jm/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:71801d6b71c1bd6a443981b154342a87e1a240821a555f1dcab3ff0de825bd68
|
| 3 |
+
size 954252
|
pku_mmd_xsub/jm/best_top1_acc_epoch_149.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d8586ac79145e2e03b5397110af2a84a4f4fe6035728963c6e3269727065b34a
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/jm/jm.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'jm'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/jm'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['jm']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['jm']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['jm']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['jm']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['jm']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['jm']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/k_1/20250702_120945.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/k_1/20250702_120945.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/k_1/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31b153f489afd44318e3046d5dd881bed38d11d06fe5b347c6a98322c965f41f
|
| 3 |
+
size 953839
|
pku_mmd_xsub/k_1/best_top1_acc_epoch_150.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:081186ff32a98d8019075f15f55f097d3bf0ed4d3e97d11a37aff9a701974584
|
| 3 |
+
size 16576377
|
pku_mmd_xsub/k_1/k_1.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'k'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/k_1'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|
pku_mmd_xsub/k_2/20250702_120831.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/k_2/20250702_120831.log.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pku_mmd_xsub/k_2/best_pred.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7a6111ae9f63cca3bd6d8495f73e8fb4da073f428c3ecff650bb9c31bd91012
|
| 3 |
+
size 954840
|
pku_mmd_xsub/k_2/best_top1_acc_epoch_147.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b88b71fe7b9637738d150439cfee9165e3ff1b90f512fa572cd501258c4f387
|
| 3 |
+
size 32917041
|
pku_mmd_xsub/k_2/k_2.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
modality = 'k'
|
| 2 |
+
graph = 'nturgb+d'
|
| 3 |
+
work_dir = './work_dirs/test_aclnet/pku_mmd_xsub/k_2'
|
| 4 |
+
model = dict(
|
| 5 |
+
type='RecognizerGCN',
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='GCN_Module',
|
| 8 |
+
gcn_ratio=0.125,
|
| 9 |
+
gcn_ctr='T',
|
| 10 |
+
gcn_ada='T',
|
| 11 |
+
tcn_ms_cfg=[(3, 1), (3, 2), (3, 3), (3, 4), ('max', 3), '1x1'],
|
| 12 |
+
graph_cfg=dict(
|
| 13 |
+
layout='nturgb+d',
|
| 14 |
+
mode='random',
|
| 15 |
+
num_filter=8,
|
| 16 |
+
init_off=0.04,
|
| 17 |
+
init_std=0.02)),
|
| 18 |
+
cls_head=dict(type='SimpleHead', data_cfg='pku_mmd', num_classes=51, in_channels=384))
|
| 19 |
+
dataset_type = 'PoseDataset'
|
| 20 |
+
ann_file = '/data/lhd/pyskl_data/pku/pku_mmd.pkl'
|
| 21 |
+
train_pipeline = [
|
| 22 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 23 |
+
dict(type='RandomRot', theta=0.2),
|
| 24 |
+
dict(type='Part_Drop'),
|
| 25 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 26 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 27 |
+
dict(type='FormatGCNInput'),
|
| 28 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 29 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 30 |
+
]
|
| 31 |
+
val_pipeline = [
|
| 32 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 33 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 34 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 35 |
+
dict(type='FormatGCNInput'),
|
| 36 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 37 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 38 |
+
]
|
| 39 |
+
test_pipeline = [
|
| 40 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 41 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 42 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 43 |
+
dict(type='FormatGCNInput'),
|
| 44 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 45 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 46 |
+
]
|
| 47 |
+
data = dict(
|
| 48 |
+
videos_per_gpu=16,
|
| 49 |
+
workers_per_gpu=4,
|
| 50 |
+
test_dataloader=dict(videos_per_gpu=1),
|
| 51 |
+
train=dict(
|
| 52 |
+
type='PoseDataset',
|
| 53 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 54 |
+
pipeline=[
|
| 55 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 56 |
+
dict(type='RandomRot', theta=0.2),
|
| 57 |
+
dict(type='Part_Drop'),
|
| 58 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 59 |
+
dict(type='UniformSampleDecode', clip_len=50),
|
| 60 |
+
dict(type='FormatGCNInput'),
|
| 61 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 62 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 63 |
+
],
|
| 64 |
+
split='xsub_train'),
|
| 65 |
+
val=dict(
|
| 66 |
+
type='PoseDataset',
|
| 67 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 68 |
+
pipeline=[
|
| 69 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 70 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 71 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=1),
|
| 72 |
+
dict(type='FormatGCNInput'),
|
| 73 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 74 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 75 |
+
],
|
| 76 |
+
split='xsub_val'),
|
| 77 |
+
test=dict(
|
| 78 |
+
type='PoseDataset',
|
| 79 |
+
ann_file='/data/lhd/pyskl_data/pku/pku_mmd.pkl',
|
| 80 |
+
pipeline=[
|
| 81 |
+
dict(type='PreNormalize3D', align_spine=False),
|
| 82 |
+
dict(type='GenSkeFeat', feats=['k']),
|
| 83 |
+
dict(type='UniformSampleDecode', clip_len=50, num_clips=10),
|
| 84 |
+
dict(type='FormatGCNInput'),
|
| 85 |
+
dict(type='Collect', keys=['keypoint', 'label'], meta_keys=[]),
|
| 86 |
+
dict(type='ToTensor', keys=['keypoint'])
|
| 87 |
+
],
|
| 88 |
+
split='xsub_val'))
|
| 89 |
+
optimizer = dict(
|
| 90 |
+
type='SGD', lr=0.025, momentum=0.9, weight_decay=0.0005, nesterov=True)
|
| 91 |
+
optimizer_config = dict(grad_clip=None)
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0, by_epoch=False)
|
| 93 |
+
total_epochs = 150
|
| 94 |
+
checkpoint_config = dict(interval=1)
|
| 95 |
+
evaluation = dict(interval=1, metrics=['top_k_accuracy'])
|
| 96 |
+
log_config = dict(interval=100, hooks=[dict(type='TextLoggerHook')])
|
| 97 |
+
dist_params = dict(backend='nccl')
|
| 98 |
+
gpu_ids = range(0, 1)
|