MCPcopy Create free account

hub / github.com/ZhengyiLuo/SMPLSim / functions

Functions1,205 in github.com/ZhengyiLuo/SMPLSim

↓ 2 callersFunctionmeasure_function
(env, func_name, reps=1000, **kwargs)
examples/nv_benchmark.py:149
↓ 2 callersFunctionnormalize
(x, eps: float = 1e-9)
smpl_sim/utils/torch_utils.py:63
↓ 2 callersFunctionnormalize
(v, eps = 1e-9)
smpl_sim/utils/np_transform_utils.py:10
↓ 2 callersFunctionnormalize_angle
(x)
smpl_sim/utils/torch_utils.py:124
↓ 2 callersFunctionnormalize_vector
( v, return_mag =False)
smpl_sim/utils/torch_geometry_transforms.py:618
↓ 2 callersFunctionnormalize_vector
(v, return_mag=False)
smpl_sim/utils/transform_utils.py:131
↓ 2 callersFunctionp_mpjpe
Pose error: MPJPE after rigid alignment (scale, rotation, and translation), often referred to as "Protocol #2" in many papers.
smpl_sim/smpllib/smpl_eval.py:98
↓ 2 callersFunctionparse_sim_params
(args, cfg, cfg_train, flags)
smpl_sim/envs/nv/utils/config.py:169
↓ 2 callersFunctionplot_skeleton_state
Visualize a skeleton state :param skeleton_state: :param task_name: :type skeleton_state: SkeletonState :type task_name: string,
smpl_sim/poselib/visualization/common.py:15
↓ 2 callersMethodppo_loss
(self, states, actions, advantages, fixed_log_probs, ind)
smpl_sim/agents/agent_ppo.py:96
↓ 2 callersMethodpreprocess_actions
(self, actions)
smpl_sim/agents/agent.py:153
↓ 2 callersFunctionquat_conjugate
(a)
smpl_sim/utils/torch_utils.py:104
↓ 2 callersFunctionquat_conjugate
quaternion with its imaginary part negated
smpl_sim/utils/poselib_rotation3d.py:60
↓ 2 callersFunctionquat_conjugate
quaternion with its imaginary part negated
smpl_sim/poselib/core/rotation3d.py:60
↓ 2 callersFunctionquat_from_angle_axis
(angle, axis)
smpl_sim/utils/np_transform_utils.py:16
↓ 2 callersFunctionquat_from_rotation_matrix
Construct a 3D rotation from a valid 3x3 rotation matrices. Reference can be found here: http://www.cg.info.hiroshima-cu.ac.jp/~miyazaki/
smpl_sim/utils/poselib_rotation3d.py:147
↓ 2 callersFunctionquat_identity_like
Construct identity 3D rotation with the same shape
smpl_sim/poselib/core/rotation3d.py:223
↓ 2 callersFunctionquat_inverse
The inverse of the rotation
smpl_sim/utils/poselib_rotation3d.py:215
↓ 2 callersFunctionquat_normalize
Construct 3D rotation from quaternion (the quaternion needs not to be normalized).
smpl_sim/utils/pytorch3d_transforms.py:674
↓ 2 callersFunctionquaternion_to_matrix
Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten
smpl_sim/utils/pytorch3d_transforms.py:48
↓ 2 callersFunctionquaternion_to_matrix
Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten
smpl_sim/utils/rotation_conversions.py:46
↓ 2 callersMethodreset
(self, env_ids=None)
smpl_sim/envs/nv/humanoid.py:998
↓ 2 callersFunctionrodrigues
Rodrigues' rotation formula that turns axis-angle vector into rotation matrix in a batch-ed manner. Parameter: ---------- r: Axi
smpl_sim/smpllib/np_smpl_humanoid_batch.py:49
↓ 2 callersFunctionrotation_matrix_to_quaternion
Convert 3x4 rotation matrix to 4d quaternion vector This algorithm is based on algorithm described in https://github.com/KieranWynn/pyquatern
smpl_sim/utils/torch_geometry_transforms.py:252
↓ 2 callersMethodsample
(self, batch_size=None)
smpl_sim/learning/memory.py:16
↓ 2 callersMethodseed
(self, seed)
smpl_sim/agents/agent_humanoid.py:252
↓ 2 callersMethodselect_action
(self, x, mean_action=False)
smpl_sim/learning/policy.py:12
↓ 2 callersMethodset_full_state_weights
(self, state)
smpl_sim/agents/agent_humanoid.py:131
↓ 2 callersFunctionsmoothing_factor
(t_e, cutoff)
smpl_sim/utils/one_euor_filter.py:4
↓ 2 callersFunctionsmpl_to_smplh
(pose)
smpl_sim/smpllib/smpl_mujoco_new.py:19
↓ 2 callersFunctionsmplh_to_smpl
(pose)
smpl_sim/smpllib/smpl_mujoco_new.py:14
↓ 2 callersMethodsync_node
(self)
smpl_sim/smpllib/smpl_local_robot.py:967
↓ 2 callersFunctionthunk
()
examples/benchmark.py:71
↓ 2 callersFunctiontorch_rand_float
(lower, upper, shape, device)
smpl_sim/utils/torch_utils.py:213
↓ 2 callersMethodtrans_value
transform states before going into value net
smpl_sim/agents/agent.py:164
↓ 2 callersFunctiontransform_inverse
Inverse transformation
smpl_sim/poselib/core/rotation3d.py:301
↓ 2 callersMethodupdate
(self, frame_index=None, reset_trail=False, skeleton_motion=None)
smpl_sim/poselib/visualization/skeleton_plotter_tasks.py:136
↓ 2 callersFunctionupdate_joint_limits
(joint_range)
smpl_sim/smpllib/smpl_local_robot.py:176
↓ 2 callersFunctionupdate_joint_limits_upright
(joint_range)
smpl_sim/smpllib/smpl_local_robot.py:247
↓ 2 callersMethodupdate_model
(self, betas, gender, dt = 1/30)
smpl_sim/smpllib/torch_smpl_humanoid_batch.py:82
↓ 2 callersMethodupdate_model
(self, betas, gender)
smpl_sim/smpllib/np_smpl_humanoid_batch.py:157
↓ 2 callersMethodupdate_sampling_prob
(self, termination_history)
smpl_sim/smpllib/motion_lib_base.py:263
↓ 2 callersFunctionvertizalize_smpl_root
(poses, root_vec=[np.pi / 2, 0, 0])
smpl_sim/utils/transform_utils.py:143
↓ 2 callersMethodwrite_xml
(self, fname)
smpl_sim/smpllib/smpl_local_robot.py:1716
↓ 1 callersMethod__init__
(self, insize, epsilon=1e-05, per_channel=False,
smpl_sim/utils/running_mean_std.py:11
↓ 1 callersMethod__init__
( self, task_name: str, dots: np.ndarray, color: str = "blue", marker_
smpl_sim/poselib/visualization/simple_plotter_tasks.py:89
↓ 1 callersMethod__init__
(self, task: "BasePlotterTask")
smpl_sim/poselib/visualization/plt_plotter.py:31
↓ 1 callersMethod__init__
( self, task_name: str, skeleton_state, joints_color: str = "red", lin
smpl_sim/poselib/visualization/skeleton_plotter_tasks.py:22
↓ 1 callersMethod__init__
:param node_names: a list of names for each tree node :type node_names: List[str] :param parent_indices: an int32-typed tenso
smpl_sim/poselib/skeleton/skeleton3d.py:76
↓ 1 callersMethod__init__
(self, input_dim, hidden_dims=(128, 128), activation='tanh')
smpl_sim/learning/mlp.py:36
↓ 1 callersMethod__init__
(self, probs=None, logits=None)
smpl_sim/learning/distributions.py:30
↓ 1 callersFunction_axis_angle_rotation
Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. A
smpl_sim/utils/pytorch3d_transforms.py:234
↓ 1 callersFunction_axis_angle_rotation
Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. A
smpl_sim/utils/rotation_conversions.py:155
↓ 1 callersMethod_build_contact_body_ids_tensor
(self, contact_body_names)
smpl_sim/envs/nv/humanoid.py:2192
↓ 1 callersMethod_build_env
(self, env_id, env_ptr, humanoid_asset)
smpl_sim/envs/nv/humanoid.py:1489
↓ 1 callersMethod_build_key_body_ids_tensor
(self, key_body_names)
smpl_sim/envs/nv/humanoid.py:2166
↓ 1 callersMethod_build_pd_action_offset_scale
(self)
smpl_sim/envs/nv/humanoid.py:1732
↓ 1 callersMethod_build_termination_heights
(self)
smpl_sim/envs/nv/humanoid.py:1169
↓ 1 callersMethod_clear_recorded_states
(self)
smpl_sim/envs/nv/base_task.py:318
↓ 1 callersMethod_compute_angular_velocity
(rotations, time_delta: float, guassian_filter=True)
smpl_sim/smpllib/torch_smpl_humanoid_batch.py:217
↓ 1 callersMethod_compute_angular_velocity
(r, time_delta: float, guassian_filter=True)
smpl_sim/poselib/skeleton/skeleton3d.py:1110
↓ 1 callersMethod_compute_desired_accel
( self, qpos_err: np.ndarray, qvel_err: np.ndarray, k_p: np.ndarray, k
smpl_sim/envs/controllers.py:165
↓ 1 callersMethod_compute_humanoid_obs
(self, env_ids=None)
smpl_sim/envs/nv/humanoid.py:1853
↓ 1 callersMethod_compute_reset
(self)
smpl_sim/envs/nv/humanoid.py:1824
↓ 1 callersMethod_compute_reward
(self, actions)
smpl_sim/envs/nv/humanoid.py:1820
↓ 1 callersFunction_compute_rotation_matrix
(angle_axis, theta2, eps=1e-6)
smpl_sim/utils/torch_geometry_transforms.py:138
↓ 1 callersFunction_compute_rotation_matrix_taylor
(angle_axis)
smpl_sim/utils/torch_geometry_transforms.py:162
↓ 1 callersMethod_compute_torque
( self, setpoint: np.ndarray, mj_model: mujoco.MjModel, mj_data: mujoco.MjData )
smpl_sim/envs/controllers.py:142
↓ 1 callersMethod_compute_velocity
(p, time_delta, guassian_filter=True)
smpl_sim/smpllib/torch_smpl_humanoid_batch.py:205
↓ 1 callersMethod_compute_velocity
(p, time_delta, guassian_filter=True)
smpl_sim/poselib/skeleton/skeleton3d.py:1100
↓ 1 callersFunction_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
smpl_sim/utils/pytorch3d_transforms.py:110
↓ 1 callersFunction_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
smpl_sim/utils/rotation_conversions.py:77
↓ 1 callersMethod_create_envs
(self, num_envs, spacing, num_per_row)
smpl_sim/envs/nv/humanoid.py:1249
↓ 1 callersMethod_create_ground_plane
(self)
smpl_sim/envs/nv/humanoid.py:1103
↓ 1 callersMethod_create_humanoid_robot
(self, cfg)
smpl_sim/envs/humanoid_env.py:219
↓ 1 callersMethod_create_impl
(self, task_list: List[BasePlotterTask])
smpl_sim/poselib/visualization/core.py:77
↓ 1 callersMethod_create_renderer
(self)
smpl_sim/envs/base_env.py:168
↓ 1 callersFunction_get_animation_curve
(joint, fbx_scene)
smpl_sim/poselib/skeleton/backend/fbx/fbx_py27_backend.py:205
↓ 1 callersFunction_get_frame_count
(fbx_scene)
smpl_sim/poselib/skeleton/backend/fbx/fbx_py27_backend.py:180
↓ 1 callersMethod_get_qv_from_motion
(self, expert_data)
smpl_sim/envs/smplenv.py:370
↓ 1 callersFunction_get_skeleton
(root_joint)
smpl_sim/poselib/skeleton/backend/fbx/fbx_py27_backend.py:261
↓ 1 callersFunction_import_module
Try import all public attributes from module into global namespace. Existing attributes with name clashes are renamed with prefix. Attributes
smpl_sim/utils/transformation.py:2017
↓ 1 callersMethod_init_camera
(self)
smpl_sim/envs/nv/humanoid.py:2211
↓ 1 callersMethod_init_tensor_history
(self, env_ids)
smpl_sim/envs/nv/humanoid.py:2087
↓ 1 callersMethod_load_amass_gender_betas
(self)
smpl_sim/envs/nv/humanoid.py:1231
↓ 1 callersMethod_physics_step
(self)
smpl_sim/envs/nv/base_task.py:973
↓ 1 callersMethod_record_states
(self)
smpl_sim/envs/nv/base_task.py:321
↓ 1 callersMethod_render_frame
(self)
smpl_sim/envs/base_env.py:148
↓ 1 callersMethod_reset_actors
(self, env_ids)
smpl_sim/envs/nv/humanoid.py:2014
↓ 1 callersMethod_reset_env_tensors
(self, env_ids)
smpl_sim/envs/nv/humanoid.py:1072
↓ 1 callersMethod_set_aspect_equal_2d
(self, zero_centered=True)
smpl_sim/poselib/visualization/plt_plotter.py:177
↓ 1 callersMethod_set_aspect_equal_3d
(self)
smpl_sim/poselib/visualization/plt_plotter.py:376
↓ 1 callersMethod_set_lim
(self)
smpl_sim/poselib/visualization/plt_plotter.py:87
↓ 1 callersMethod_set_lim
(self)
smpl_sim/poselib/visualization/plt_plotter.py:275
↓ 1 callersMethod_setup_action_space
(self)
smpl_sim/agents/agent.py:44
↓ 1 callersMethod_setup_character_props
(self, key_bodies)
smpl_sim/envs/nv/humanoid.py:1116
↓ 1 callersMethod_setup_tensors
(self)
smpl_sim/envs/nv/humanoid.py:192
↓ 1 callersFunction_sigmoids
Returns 1 when `x` == 0, between 0 and 1 otherwise. Args: x: A scalar or numpy array. value_at_1: A float between 0 and 1 specifying the ou
smpl_sim/utils/tolerance.py:25
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
smpl_sim/utils/rotation_conversions.py:95
↓ 1 callersMethod_update
(self, lines, dots)
smpl_sim/poselib/visualization/skeleton_plotter_tasks.py:64
← previousnext →201–300 of 1,205, ranked by callers