File size: 2,074 Bytes
5ac1897
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
""" Optimization config file. In 'optim_steps' we define each optimization steps.
Each step inherit and overwrite the parameters of the previous step."""

config = {
    'keepalive_meshviewer': False,
    'optim_steps':
        [
            {
            'description' : 'Adjust the root orientation and translation',
            'use_basic_loss': True,
            'lr': 1,
            'max_iter': 20,
            'num_steps': 10,
            'line_search_fn': 'strong_wolfe', #'strong_wolfe',
            'tolerance_change': 1e-7,# 1e-4, #0.01
            'mode' : 'root_only',

            'l_verts_loose': 300,
            'l_time_loss': 0,#5e2,

            'l_joint': 0.0,
            'l_verts': 0,
            'l_scapula_loss': 0.0,
            'l_spine_loss': 0.0,
            'l_pose_loss': 0.0,


            },
            # Adjust the upper limbs
            {
            'description' : 'Adjust the upper limbs pose',
            'lr': 0.1,
            'max_iter': 20,
            'num_steps': 10,
            'tolerance_change': 1e-7,
            'mode' : 'fixed_upper_limbs', #'fixed_root',

            'l_verts_loose': 600,
            'l_joint': 2e3,
            'l_time_loss': 0,# 5e2,
            'l_pose_loss': 1e-3,
            },
            # Adjust the whole body
            {
            'description' : 'Adjust the whole body pose with fixed root',
            'lr': 0.1,
            'max_iter': 20,
            'num_steps': 10,
            'tolerance_change': 1e-7,
            'mode' : 'fixed_root', #'fixed_root',

            'l_verts_loose': 600,
            'l_joint': 1e3,
            'l_time_loss': 0,
            'l_pose_loss': 1e-4,
            },
            #
            {
            'description' : 'Free optimization',
            'lr': 0.1,
            'max_iter': 20,
            'num_steps': 10,
            'tolerance_change': 1e-7,
            'mode' : 'free', #'fixed_root',

            'l_verts_loose': 300,
            'l_joint': 2e3,
            'l_time_loss':0,
            'l_pose_loss': 1e-4,
            },
    ]
}