support Human3.6M

This commit is contained in:
Iridoudou
2021-08-19 12:40:38 +08:00
parent 5ebe4a42a0
commit e0ee13d6d6

View File

@ -5,77 +5,75 @@
"TRAIN": { "TRAIN": {
"LEARNING_RATE": 2e-2, "LEARNING_RATE": 2e-2,
"MAX_EPOCH": 1000, "MAX_EPOCH": 1000,
"WRITE": 1 "WRITE": 1,
"OPTIMIZE_SCALE":1,
"OPTIMIZE_SHAPE":0
}, },
"USE_GPU": 1, "USE_GPU": 1,
"DATASET": { "DATASET": {
"NAME": "UTD-MHAD", "NAME": "UTD-MHAD",
"PATH": "../UTD-MHAD/Skeleton/Skeleton/", "PATH": "../human3.6M/output",
"TARGET_PATH": "", "TARGET_PATH": "",
"DATA_MAP": [ "DATA_MAP": [
[
12,
1
],
[ [
0, 0,
3 0
],
[
16,
4
],
[
18,
5
],
[
20,
6
],
[
22,
7
],
[
17,
8
],
[
19,
9
],
[
21,
10
],
[
23,
11
],
[
1,
12
],
[
4,
13
],
[
7,
14
], ],
[ [
2, 2,
16 1
], ],
[ [
5, 5,
17 2
], ],
[ [
8, 8,
18 3
],
[
1,
4
],
[
4,
5
],
[
7,
6
],
[
12,
8
],
[
13,
11
],
[
18,
12
],
[
20,
13
],
[
14,
14
],
[
19,
15
],
[
21,
16
],
[
9,
17
] ]
] ]
}, },