push

smpl_generator.py

@@ -157,6 +157,22 @@ class SMPLGenerator:
         
         if body_pose is None:
             body_pose = torch.zeros([batch_size, 69], device=self.device)
+            
+            left_shoulder_adduction = 0.5
+            right_shoulder_adduction = -0.5
+            left_shoulder_extension = 0.3
+            right_shoulder_extension = 0.3
+            
+            left_elbow_flexion = -1.2
+            right_elbow_flexion = -1.2
+            
+            body_pose[0, 16] = left_shoulder_adduction
+            body_pose[0, 17] = left_shoulder_extension
+            body_pose[0, 18] = right_shoulder_adduction
+            body_pose[0, 19] = right_shoulder_extension
+            
+            body_pose[0, 20] = left_elbow_flexion
+            body_pose[0, 21] = right_elbow_flexion
         elif isinstance(body_pose, np.ndarray):
             body_pose = torch.FloatTensor(body_pose).to(self.device)