MCPcopy Create free account

hub / github.com/ZhengyiLuo/SMPLSim / functions

Functions1,205 in github.com/ZhengyiLuo/SMPLSim

↓ 1 callersMethod_update
(self, vel_lines, avel_lines)
smpl_sim/poselib/visualization/skeleton_plotter_tasks.py:171
↓ 1 callersMethod_update_camera
(self)
smpl_sim/envs/nv/humanoid.py:2225
↓ 1 callersMethod_update_debug_viz
(self)
smpl_sim/envs/nv/humanoid.py:2251
↓ 1 callersMethod_update_impl
(self, task_list: List[BasePlotterTask])
smpl_sim/poselib/visualization/core.py:74
↓ 1 callersMethod_update_mean_var_count_from_moments
(self, mean, var, count, batch_mean, batch_var, batch_count)
smpl_sim/utils/running_mean_std.py:56
↓ 1 callersMethod_update_tensor_history
(self)
smpl_sim/envs/nv/humanoid.py:2109
↓ 1 callersMethod_write_states_to_file
(self, file_name)
smpl_sim/envs/nv/base_task.py:324
↓ 1 callersMethodadd_body
(self, body_node, parent_body)
smpl_sim/smpllib/smpl_local_robot.py:1568
↓ 1 callersFunctionaffine_matrix_from_points
Return affine transform matrix to register two point sets. v0 and v1 are shape (ndims, \*) arrays of at least ndims non-homogeneous coordinat
smpl_sim/utils/transformation.py:917
↓ 1 callersMethodafter_epoch
(self)
smpl_sim/agents/agent_humanoid.py:168
↓ 1 callersFunctionangle_axis_to_exp_map
(angle, axis)
smpl_sim/utils/torch_utils.py:311
↓ 1 callersFunctionangle_axis_to_exp_map
(angle, axis)
smpl_sim/utils/np_transform_utils.py:133
↓ 1 callersFunctionangle_between_vectors
Return angle between vectors. If directed is False, the input vectors are interpreted as undirected axes, i.e. the maximum angle is pi/2.
smpl_sim/utils/transformation.py:1931
↓ 1 callersFunctionappend_children
Depth first search function :param joint: joint item in the fbx :param pos: position of current element (for parenting)
smpl_sim/poselib/skeleton/backend/fbx/fbx_py27_backend.py:268
↓ 1 callersFunctionarcball_nearest_axis
Return axis, which arc is nearest to point.
smpl_sim/utils/transformation.py:1779
↓ 1 callersFunctionaxis_angle_to_quaternion
Convert rotations given as axis/angle to quaternions. Args: axis_angle: Rotations given as a vector in axis angle form,
smpl_sim/utils/pytorch3d_transforms.py:546
↓ 1 callersFunctionaxis_angle_to_quaternion
Convert rotations given as axis/angle to quaternions. Args: axis_angle: Rotations given as a vector in axis angle form,
smpl_sim/utils/rotation_conversions.py:467
↓ 1 callersMethodbuild_pd_action_scale
(self)
smpl_sim/envs/humanoid_env.py:325
↓ 1 callersFunctioncompute_height_observations
(tar_height, upright_start, humanoid_type)
smpl_sim/envs/tasks/humanoid_getup.py:20
↓ 1 callersFunctioncompute_humanoid_observations
( root_pos, root_rot, root_vel, root_ang_vel, dof_pos, dof_vel, key_body_pos,
smpl_sim/envs/nv/humanoid.py:2313
↓ 1 callersFunctioncompute_humanoid_observations_max
( body_pos, body_rot, body_vel, body_ang_vel, local_root_obs, root_height_obs )
smpl_sim/envs/nv/humanoid.py:2381
↓ 1 callersFunctioncompute_humanoid_observations_smpl
( root_pos, root_rot, root_vel, root_ang_vel, dof_pos, dof_vel, key_body_pos,
smpl_sim/envs/nv/humanoid.py:2530
↓ 1 callersFunctioncompute_humanoid_observations_smpl_max
( body_pos, body_rot, body_vel, body_ang_vel, smpl_params, limb_weight_params, loc
smpl_sim/envs/nv/humanoid.py:2606
↓ 1 callersFunctioncompute_humanoid_observations_smpl_max_v2
( body_pos, body_rot, body_vel, body_ang_vel, smpl_params, limb_weight_params, loc
smpl_sim/envs/nv/humanoid.py:2704
↓ 1 callersFunctioncompute_humanoid_reset
( reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, max_episode_
smpl_sim/envs/nv/humanoid.py:2468
↓ 1 callersFunctioncompute_humanoid_reward
(obs_buf)
smpl_sim/envs/nv/humanoid.py:2461
↓ 1 callersFunctioncompute_humanoid_self_obs_v1
(qpos, qvel, body_pos, body_rot, upright_start, root_height_obs, humanoid_type)
smpl_sim/envs/humanoid_env.py:565
↓ 1 callersFunctioncompute_humanoid_self_obs_v2
(body_pos, body_rot, body_vel, body_ang_vel, upright_start, root_height_obs, humanoid_type)
smpl_sim/envs/humanoid_env.py:637
↓ 1 callersMethodcompute_info
(self)
smpl_sim/envs/base_env.py:120
↓ 1 callersFunctioncompute_location_observations
(root_pos, root_rot, tar_pos, upright_start, humanoid_type)
smpl_sim/envs/tasks/humanoid_reach.py:21
↓ 1 callersMethodcompute_observations
(self)
smpl_sim/envs/humanoid_env.py:381
↓ 1 callersMethodcompute_observations
(self)
smpl_sim/envs/base_env.py:116
↓ 1 callersFunctioncompute_orth6d_from_rotation_matrix
(rot_mats)
smpl_sim/utils/torch_geometry_transforms.py:570
↓ 1 callersFunctioncompute_orth6d_from_rotation_matrix
(rot_mats)
smpl_sim/utils/transform_utils.py:77
↓ 1 callersFunctioncompute_penetration
(vert, info)
smpl_sim/smpllib/smpl_eval.py:207
↓ 1 callersMethodcompute_reset
(self)
smpl_sim/envs/humanoid_env.py:431
↓ 1 callersMethodcompute_reward
(self, actions)
smpl_sim/envs/humanoid_env.py:427
↓ 1 callersFunctioncompute_skate
(vert, info)
smpl_sim/smpllib/smpl_eval.py:220
↓ 1 callersFunctioncompute_speed_observations
(root_rot, tar_speed, upright_start, humanoid_type)
smpl_sim/envs/tasks/humanoid_speed.py:33
↓ 1 callersMethodcompute_task_obs
(self)
smpl_sim/envs/humanoid_task.py:23
↓ 1 callersFunctionconvert_points_from_homogeneous
r"""Function that converts points from homogeneous to Euclidean space. See :class:`~torchgeometry.ConvertPointsFromHomogeneous` for details.
smpl_sim/utils/torch_geometry_transforms.py:81
↓ 1 callersFunctionconvert_points_to_homogeneous
r"""Function that converts points from Euclidean to homogeneous space. See :class:`~torchgeometry.ConvertPointsToHomogeneous` for details. E
smpl_sim/utils/torch_geometry_transforms.py:101
↓ 1 callersFunctioncopysign
(a, b)
smpl_sim/utils/torch_utils.py:166
↓ 1 callersMethodcreate_controller
crete a torque proportional controller or a PID controller
smpl_sim/envs/smplenv.py:243
↓ 1 callersFunctioncreate_logger
(filename, file_handle=True)
smpl_sim/learning/logger_txt.py:4
↓ 1 callersMethodcreate_sim
(self, xml: str)
smpl_sim/envs/base_env.py:139
↓ 1 callersMethodcreate_sim
(self, compute_device, graphics_device, physics_engine, sim_params)
smpl_sim/envs/nv/base_task.py:284
↓ 1 callersMethodcreate_task_visualization
(self)
smpl_sim/envs/humanoid_task.py:38
↓ 1 callersMethodcreate_viewer
(self)
smpl_sim/envs/base_env.py:173
↓ 1 callersMethodcreate_viewer
(self)
smpl_sim/envs/nv/base_task.py:163
↓ 1 callersFunctiondeg2rad
r"""Function that converts angles from degrees to radians. See :class:`~torchgeometry.DegToRad` for details. Args: tensor (Tensor):
smpl_sim/utils/torch_geometry_transforms.py:58
↓ 1 callersFunctiondict_to_torch
(input_dict, dtype = None, device = None, add_dim = False)
smpl_sim/utils/torch_ext.py:191
↓ 1 callersMethoddrop_nodes_by_names
Drop a list of nodes from the skeleton and re-compute the local rotation to match the original joint position as much as possible.
smpl_sim/poselib/skeleton/skeleton3d.py:660
↓ 1 callersMethodend_episode
(self, env)
smpl_sim/learning/logger_rl.py:35
↓ 1 callersMethodend_sampling
(self)
smpl_sim/learning/logger_rl.py:41
↓ 1 callersMethodenforce_length_size
(self, size_dict)
smpl_sim/smpllib/smpl_local_robot.py:1534
↓ 1 callersFunctionestimate_advantages
(rewards, not_done, not_dead, values, gamma, tau)
smpl_sim/learning/learning_utils.py:198
↓ 1 callersFunctioneuler_from_matrix
Return Euler angles from rotation matrix for specified axis sequence. axes : One of 24 axis sequences as string or encoded tuple Note that m
smpl_sim/utils/transformation.py:1140
↓ 1 callersFunctioneuler_matrix
Return homogeneous rotation matrix from Euler angles and axis sequence. ai, aj, ak : Euler's roll, pitch and yaw angles axes : One of 24 axis
smpl_sim/utils/transformation.py:1077
↓ 1 callersMethodeval_policy
(self, epoch=0, dump=False)
smpl_sim/agents/agent_humanoid.py:227
↓ 1 callersFunctionevaluate_env
(env, time_format: str = "ms", reps=100)
examples/benchmark.py:97
↓ 1 callersFunctionevaluate_env
(env, time_format: str = "ms", reps=100)
examples/nv_benchmark.py:159
↓ 1 callersFunctionexp_map_to_angle_axis
(exp_map)
smpl_sim/utils/torch_utils.py:378
↓ 1 callersFunctionfbx_to_array
Reads an fbx file to an array. Currently reading of the frame time is not supported. 120 fps is hard coded TODO :param fbx_file_path: s
smpl_sim/poselib/skeleton/backend/fbx/fbx_read_wrapper.py:30
↓ 1 callersFunctionfbx_to_npy
This function reads in an fbx file, and saves the relevant info to a numpy array Fbx files have a series of animation curves, each of which
smpl_sim/poselib/skeleton/backend/fbx/fbx_py27_backend.py:42
↓ 1 callersMethodfix_trans_height
(pose_aa, trans, curr_gender_betas, mesh_parsers, fix_height_mode)
smpl_sim/smpllib/motion_lib_smpl.py:68
↓ 1 callersMethodfk_batch
(self, pose, trans, convert_to_mat=True, count_offset=True)
smpl_sim/smpllib/np_smpl_humanoid_batch.py:306
↓ 1 callersMethodforward
(self, *args, **kwargs)
smpl_sim/smpllib/smpl_parser.py:268
↓ 1 callersMethodforward
(self, *args, **kwargs)
smpl_sim/smpllib/smpl_parser.py:406
↓ 1 callersMethodforward
(self, *args, **kwargs)
smpl_sim/smpllib/smpl_parser.py:609
↓ 1 callersMethodforward
(self, x)
smpl_sim/learning/policy_mcp.py:28
↓ 1 callersMethodforward
(self, x)
smpl_sim/learning/policy_disc.py:18
↓ 1 callersMethodforward
(self, x)
smpl_sim/learning/policy_gaussian.py:35
↓ 1 callersMethodforward_bones
(self, bone)
smpl_sim/smpllib/skeleton_mesh_local.py:166
↓ 1 callersMethodforward_bvh
(self, bone)
smpl_sim/smpllib/skeleton_local.py:287
↓ 1 callersMethodforward_kinematics_batch
Perform forward kinematics using the given trajectory and local rotations. Arguments (where B = batch size, J = number of joints):
smpl_sim/smpllib/torch_smpl_humanoid_batch.py:172
↓ 1 callersFunctionforward_reward
(tar_speed, root_pos, prev_root_pos, dt)
smpl_sim/envs/tasks/humanoid_speed.py:9
↓ 1 callersMethodfrom_dict
Read the object from an ordered dictionary :param dict_repr: the ordered dictionary that is used to construct the object :type dict_
smpl_sim/poselib/core/backend/abstract.py:75
↓ 1 callersMethodfrom_state_vector_and_velocity
Construct a skeleton motion from a skeleton state vector, global velocity and angular velocity at each joint. :param skeleto
smpl_sim/poselib/skeleton/skeleton3d.py:960
↓ 1 callersFunctiongaussian_kernel_1d
(size, sigma)
smpl_sim/utils/smoothing_utils.py:5
↓ 1 callersMethodget_action_size
(self)
smpl_sim/envs/nv/humanoid.py:972
↓ 1 callersFunctionget_axis_params
construct arguments to `Vec` according to axis index.
smpl_sim/utils/torch_utils.py:154
↓ 1 callersMethodget_body_angular_vel
(self)
smpl_sim/envs/humanoid_env.py:543
↓ 1 callersMethodget_body_jac_by_id
(self, id, jacp=None, jacr=None)
smpl_sim/envs/humanoid_env.py:514
↓ 1 callersMethodget_body_linear_vel
(self)
smpl_sim/envs/humanoid_env.py:539
↓ 1 callersMethodget_body_xpos_by_id
(self, body_id)
smpl_sim/envs/humanoid_env.py:421
↓ 1 callersMethodget_body_xquat
(self)
smpl_sim/envs/humanoid_env.py:424
↓ 1 callersFunctionget_frobenious_norm
(x, y)
smpl_sim/smpllib/smpl_eval.py:262
↓ 1 callersFunctionget_geom_dict
( smpl_verts, smpl_jts, skin_weights, joint_names, scale_dict={}, )
smpl_sim/smpllib/smpl_local_robot.py:146
↓ 1 callersFunctionget_heading_new
(q)
smpl_sim/utils/math_utils.py:201
↓ 1 callersFunctionget_heading_q
(q)
smpl_sim/utils/math_utils.py:150
↓ 1 callersFunctionget_joint_geometries
( smpl_verts, smpl_jts, skin_weights, joint_names, geom_dir, scale_dict={}, suffix
smpl_sim/smpllib/smpl_local_robot.py:82
↓ 1 callersMethodget_motion_num_steps
(self, motion_ids=None)
smpl_sim/smpllib/motion_lib_base.py:307
↓ 1 callersMethodget_motion_state_intervaled
(self, motion_ids, motion_times, offset=None)
smpl_sim/smpllib/motion_lib_base.py:313
↓ 1 callersMethodget_new_a_scale
(self)
smpl_sim/smpllib/smpl_mujoco_new.py:363
↓ 1 callersMethodget_new_jkd
(self)
smpl_sim/smpllib/smpl_mujoco_new.py:358
↓ 1 callersMethodget_new_jkp
(self)
smpl_sim/smpllib/smpl_mujoco_new.py:353
↓ 1 callersMethodget_new_torque_limit
(self)
smpl_sim/smpllib/smpl_mujoco_new.py:368
↓ 1 callersMethodget_nn_weights
(self)
smpl_sim/agents/agent_humanoid.py:113
↓ 1 callersMethodget_num_actors_per_env
(self)
smpl_sim/envs/nv/humanoid.py:978
← previousnext →301–400 of 1,205, ranked by callers