support NTU

This commit is contained in:
Iridoudou
2021-09-03 11:06:26 +08:00
parent 0702081dbe
commit 16387a7afe
8 changed files with 236 additions and 23 deletions

94
fit/configs/NTU.json Normal file
View File

@ -0,0 +1,94 @@
{
"MODEL": {
"GENDER": "neutral"
},
"TRAIN": {
"LEARNING_RATE": 5e-2,
"MAX_EPOCH": 1000,
"WRITE": 10,
"OPTIMIZE_SCALE":0,
"OPTIMIZE_SHAPE":1
},
"USE_GPU": 1,
"DATASET": {
"NAME": "NTU",
"PATH": "../NTU RGB+D/skeleton_npy",
"TARGET_PATH": "",
"DATA_MAP": [
[
0,
0
],
[
1,
12
],
[
2,
16
],
[
4,
13
],
[
5,
17
],
[
6,
1
],
[
7,
14
],
[
8,
18
],
[
9,
20
],
[
12,
2
],
[
13,
4
],
[
14,
8
],
[
18,
5
],
[
19,
9
],
[
20,
6
],
[
21,
10
],
[
22,
22
],
[
23,
24
]
]
},
"DEBUG": 0
}