Support HAA4D

This commit is contained in:
Yiming Dou
2022-10-01 21:42:12 +08:00
parent fd2ff6616f
commit 6ac10e22e0
12 changed files with 232 additions and 170 deletions

24
fit/configs/HAA4D.json Normal file
View File

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

View File

@ -20,19 +20,19 @@
0
],
[
1,
2,
12
],
[
2,
1,
16
],
[
4,
5,
13
],
[
5,
4,
17
],
[
@ -40,52 +40,51 @@
1
],
[
7,
8,
14
],
[
8,
7,
18
],
[
9,
20
],
[
12,
2
],
[
13,
14,
4
],
[
14,
13,
8
],
[
18,
19,
5
],
[
19,
18,
9
],
[
20,
21,
6
],
[
21,
20,
10
],
[
22,
23,
22
],
[
23,
22,
24
]
]