push pose

smpl_generator.py

@@ -158,21 +158,40 @@ class SMPLGenerator:
         if body_pose is None:
             body_pose = torch.zeros([batch_size, 69], device=self.device)
             
-            left_shoulder_adduction = 0.6
-            right_shoulder_adduction = -0.6
-            left_shoulder_extension = 0.4
-            right_shoulder_extension = 0.4
+            left_shoulder_adduction = 0.25
+            right_shoulder_adduction = -0.25
             
-            left_elbow_flexion = -1.3
-            right_elbow_flexion = -1.3
+            left_shoulder_extension = -0.10
+            right_shoulder_extension = -0.10
+            
+            left_shoulder_twist = 0.05
+            right_shoulder_twist = -0.05
+            
+            left_elbow_flexion = 0.18
+            right_elbow_flexion = 0.18
+            
+            left_foot_outward = 0.18
+            right_foot_outward = -0.18
+            
+            left_ankle_dorsiflex = 0.05
+            right_ankle_dorsiflex = 0.05
             
             body_pose[0, 16] = left_shoulder_adduction
             body_pose[0, 17] = left_shoulder_extension
+            body_pose[0, 22] = left_shoulder_twist
+            
             body_pose[0, 18] = right_shoulder_adduction
             body_pose[0, 19] = right_shoulder_extension
+            body_pose[0, 23] = right_shoulder_twist
             
             body_pose[0, 20] = left_elbow_flexion
             body_pose[0, 21] = right_elbow_flexion
+            
+            body_pose[0, 26] = left_ankle_dorsiflex
+            body_pose[0, 27] = left_foot_outward
+            
+            body_pose[0, 24] = right_ankle_dorsiflex
+            body_pose[0, 25] = right_foot_outward
         elif isinstance(body_pose, np.ndarray):
             body_pose = torch.FloatTensor(body_pose).to(self.device)