MCPcopy Create free account

hub / github.com/InternRobotics/UniHSI / functions

Functions660 in github.com/InternRobotics/UniHSI

↓ 66 callersFunctionquat_rotate
Rotate a 3D vector with the 3D rotation
unihsi/poselib/poselib/core/rotation3d.py:230
↓ 58 callersMethodclone
(self)
unihsi/poselib/poselib/skeleton/skeleton3d.py:1031
↓ 53 callersMethodnumpy
(self)
unihsi/utils/motion_lib.py:49
↓ 34 callersMethodformat
(self, record)
unihsi/utils/logger.py:25
↓ 28 callersMethodupdate
Update the plotter for any updates in the task primitives
unihsi/poselib/poselib/visualization/core.py:91
↓ 13 callersMethodfrom_rotation_and_root_translation
Construct a skeleton state from rotation and root translation :param skeleton_tree: the skeleton tree :type skeleton_tree: S
unihsi/poselib/poselib/skeleton/skeleton3d.py:652
↓ 13 callersFunctiontensor_to_dict
Construct an ordered dictionary from the object :rtype: OrderedDict
unihsi/poselib/poselib/core/tensor_utils.py:32
↓ 12 callersMethodload
(self, params)
unihsi/learning/amp_network_builder.py:60
↓ 11 callersFunctionquat_from_angle_axis
Create a 3D rotation from angle and axis of rotation. The rotation is counter-clockwise along the axis. The rotation can be interpreted as
unihsi/poselib/poselib/core/rotation3d.py:147
↓ 11 callersFunctionquat_mul
quaternion multiplication
unihsi/poselib/poselib/core/rotation3d.py:37
↓ 11 callersMethodstep
(self, actions)
unihsi/env/tasks/vec_task.py:37
↓ 8 callersMethodfrom_dict
(cls, dict_repr, *args, **kwargs)
unihsi/poselib/poselib/skeleton/skeleton3d.py:159
↓ 8 callersFunctionquat_inverse
The inverse of the rotation
unihsi/poselib/poselib/core/rotation3d.py:239
↓ 8 callersFunctionquat_mul_norm
Combine two set of 3D rotations together using \**\* operator. The shape needs to be broadcastable
unihsi/poselib/poselib/core/rotation3d.py:221
↓ 8 callersFunctiontransform_translation
Get translation from transform
unihsi/poselib/poselib/core/rotation3d.py:322
↓ 8 callersMethodupdate
(self, skeleton_state)
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:65
↓ 7 callersFunctiontransform_from_rotation_translation
Construct a transform from a quaternion and 3D translation. Only one of them can be None.
unihsi/poselib/poselib/core/rotation3d.py:289
↓ 7 callersFunctiontransform_rotation
Get rotation from transform
unihsi/poselib/poselib/core/rotation3d.py:316
↓ 6 callersMethodget_buffer_size
(self)
unihsi/learning/replay_buffer.py:49
↓ 6 callersMethodget_scoped_name
(self, name)
unihsi/poselib/poselib/visualization/core.py:52
↓ 6 callersFunctionquat_normalize
Construct 3D rotation from quaternion (the quaternion needs not to be normalized).
unihsi/poselib/poselib/core/rotation3d.py:117
↓ 5 callersFunctioncopy_file
(source_path, destination_path)
data/cp_partnet_train.py:49
↓ 5 callersFunctioncopy_file
(source_path, destination_path)
data/cp_partnet_test.py:48
↓ 5 callersMethodcreate
Create more task primitives from a task for the plotter
unihsi/poselib/poselib/visualization/core.py:85
↓ 5 callersMethodeval_disc
(self, amp_obs)
unihsi/learning/amp_network_builder.py:109
↓ 5 callersFunctionquat_identity
Construct 3D identity rotation given shape
unihsi/poselib/poselib/core/rotation3d.py:136
↓ 4 callersMethod_preproc_amp_obs
(self, amp_obs)
unihsi/learning/amp_agent.py:571
↓ 4 callersMethod_preproc_obs
(self, obs_batch)
unihsi/learning/common_player.py:164
↓ 4 callersMethod_to_state_vector
(rot, rt)
unihsi/poselib/poselib/skeleton/skeleton3d.py:620
↓ 4 callersFunctionbuild_amp_observations
(root_pos, root_rot, root_vel, root_ang_vel, dof_pos, dof_vel, key_body_pos, local
unihsi/env/tasks/humanoid_amp.py:391
↓ 4 callersMethodfrom_skeleton_state
Construct a skeleton motion from a skeleton state. The velocities are estimated using second order gaussian filter along the last axi
unihsi/poselib/poselib/skeleton/skeleton3d.py:1118
↓ 4 callersMethodget_num_amp_obs
(self)
unihsi/env/tasks/humanoid_amp.py:97
↓ 4 callersMethodindex
get the index of the node :param node_name: the name of the node :type node_name: string :rtype: int
unihsi/poselib/poselib/skeleton/skeleton3d.py:231
↓ 4 callersMethodset_train
(self)
unihsi/learning/amp_agent.py:69
↓ 3 callersMethod__init__
(self, **kwargs)
unihsi/learning/amp_network_builder.py:40
↓ 3 callersMethod_indent
(self, s)
unihsi/poselib/poselib/skeleton/skeleton3d.py:138
↓ 3 callersMethod_max
(self, x, y)
unihsi/poselib/poselib/visualization/plt_plotter.py:274
↓ 3 callersMethod_min
(self, x, y)
unihsi/poselib/poselib/visualization/plt_plotter.py:267
↓ 3 callersMethod_transfer_to
(self, new_skeleton_tree: SkeletonTree)
unihsi/poselib/poselib/skeleton/skeleton3d.py:734
↓ 3 callersMethod_update_lim
(self, xs, ys)
unihsi/poselib/poselib/visualization/plt_plotter.py:103
↓ 3 callersMethod_update_lim
(self, xs, ys, zs)
unihsi/poselib/poselib/visualization/plt_plotter.py:289
↓ 3 callersMethodenv_reset
(self, env_ids=None)
unihsi/learning/common_agent.py:482
↓ 3 callersMethodenv_step
(self, env, actions)
unihsi/learning/common_player.py:179
↓ 3 callersMethodeval_critic
(self, obs)
unihsi/learning/amp_network_builder.py:102
↓ 3 callersMethodget_motion_state
(self, motion_ids, motion_times)
unihsi/utils/motion_lib.py:162
↓ 3 callersMethodget_state
(self)
unihsi/env/tasks/vec_task.py:123
↓ 3 callersMethodget_total_count
(self)
unihsi/learning/replay_buffer.py:52
↓ 3 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/
unihsi/poselib/poselib/core/rotation3d.py:171
↓ 3 callersMethodrender
(self, sync_frame_time=False)
unihsi/env/tasks/humanoid.py:475
↓ 3 callersMethodsample
(self, n)
unihsi/learning/replay_buffer.py:79
↓ 3 callersMethodshow
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:264
↓ 3 callersMethodstore
(self, data_dict)
unihsi/learning/replay_buffer.py:55
↓ 3 callersFunctiontransform_mul
Combine two transformation together
unihsi/poselib/poselib/core/rotation3d.py:347
↓ 2 callersMethod__init__
(self, task, rl_device, sync_frame_time=False, clip_observations=5.0, clip_actions=1.0)
unihsi/env/tasks/vec_task_wrappers.py:35
↓ 2 callersMethod__init__
(self, task, rl_device, clip_observations=5.0, clip_actions=1.0)
unihsi/env/tasks/vec_task.py:18
↓ 2 callersMethod_actor_loss
(self, old_action_log_probs_batch, action_log_probs, advantage, curr_e_clip)
unihsi/learning/common_agent.py:536
↓ 2 callersMethod_build_disc
(self, input_shape)
unihsi/learning/amp_network_builder.py:126
↓ 2 callersMethod_calc_amp_rewards
(self, amp_obs)
unihsi/learning/amp_agent.py:599
↓ 2 callersFunction_check_longest_curve
(curve, max_curve_key_count)
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:207
↓ 2 callersMethod_compute_amp_observations
(self, env_ids=None)
unihsi/env/tasks/humanoid_amp.py:332
↓ 2 callersMethod_compute_humanoid_obs
(self, env_ids=None)
unihsi/env/tasks/humanoid.py:424
↓ 2 callersMethod_compute_observations
(self, env_ids=None)
unihsi/env/tasks/humanoid.py:414
↓ 2 callersMethod_critic_loss
(self, value_preds_batch, values, curr_e_clip, return_batch, clip_value)
unihsi/learning/common_agent.py:552
↓ 2 callersMethod_draw
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:221
↓ 2 callersMethod_draw
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:419
↓ 2 callersMethod_eval_critic
(self, obs_dict)
unihsi/learning/common_agent.py:526
↓ 2 callersMethod_eval_disc
(self, amp_obs)
unihsi/learning/amp_players.py:94
↓ 2 callersMethod_eval_disc
(self, amp_obs)
unihsi/learning/amp_agent.py:583
↓ 2 callersMethod_fetch_amp_obs_demo
(self, num_samples)
unihsi/learning/amp_agent.py:534
↓ 2 callersMethod_get_lines_and_dots
Get all the lines and dots needed to draw the skeleton state
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:69
↓ 2 callersMethod_get_pairwise_average_translation
(self)
unihsi/poselib/poselib/skeleton/skeleton3d.py:722
↓ 2 callersMethod_get_vel_and_avel
Get all the velocity and angular velocity lines
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:182
↓ 2 callersMethod_init_lim
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:97
↓ 2 callersMethod_init_lim
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:281
↓ 2 callersMethod_lines_extract_xy_impl
(index, lines_task)
unihsi/poselib/poselib/visualization/plt_plotter.py:121
↓ 2 callersMethod_lines_extract_xyz_impl
(index, lines_task)
unihsi/poselib/poselib/visualization/plt_plotter.py:312
↓ 2 callersMethod_max
(self, x, y)
unihsi/poselib/poselib/visualization/plt_plotter.py:90
↓ 2 callersMethod_min
(self, x, y)
unihsi/poselib/poselib/visualization/plt_plotter.py:83
↓ 2 callersFunction_recursive_to_list
Takes some iterable that might contain iterables and converts it to a list of lists [of lists... etc] Mainly used for converting the st
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:258
↓ 2 callersMethod_refresh_sim_tensors
(self)
unihsi/env/tasks/humanoid.py:404
↓ 2 callersMethod_remapped_to
( self, joint_mapping: Dict[str, str], target_skeleton_tree: SkeletonTree )
unihsi/poselib/poselib/skeleton/skeleton3d.py:785
↓ 2 callersMethod_reset_default
(self, env_ids)
unihsi/env/tasks/humanoid_amp.py:191
↓ 2 callersMethod_reset_ref_state_init
(self, env_ids)
unihsi/env/tasks/humanoid_amp.py:198
↓ 2 callersMethod_reset_sample_idx
(self)
unihsi/learning/replay_buffer.py:99
↓ 2 callersMethod_trail_extract_xy_impl
(index, trail_task)
unihsi/poselib/poselib/visualization/plt_plotter.py:125
↓ 2 callersMethod_trail_extract_xyz_impl
(index, trail_task)
unihsi/poselib/poselib/visualization/plt_plotter.py:316
↓ 2 callersMethodbound_loss
(self, mu)
unihsi/learning/common_agent.py:487
↓ 2 callersMethodbuild
(self, config)
unihsi/learning/amp_models.py:37
↓ 2 callersFunctioncalc_heading
(q)
unihsi/utils/torch_utils.py:146
↓ 2 callersMethodcreate_sim
(self)
unihsi/env/tasks/humanoid.py:145
↓ 2 callersMethoddim
(self)
unihsi/poselib/poselib/visualization/simple_plotter_tasks.py:162
↓ 2 callersMethoddiscount_values
(self, mb_fdones, mb_values, mb_rewards, mb_next_values)
unihsi/learning/common_agent.py:468
↓ 2 callersFunctiondof_to_obs
(pose, dof_obs_size, dof_offsets)
unihsi/env/tasks/humanoid.py:552
↓ 2 callersMethoddrop_nodes_by_names
( self, node_names: List[str], pairwise_translation=None )
unihsi/poselib/poselib/skeleton/skeleton3d.py:240
↓ 2 callersMethodenv_reset
(self, env_ids=None)
unihsi/learning/common_player.py:207
↓ 2 callersFunctioneuclidean_to_rotation_matrix
Get the rotation matrix on the top-left corner of a Euclidean transformation matrix
unihsi/poselib/poselib/core/rotation3d.py:427
↓ 2 callersFunctioneuclidean_to_transform
Construct a transform from a Euclidean transformation matrix
unihsi/poselib/poselib/core/rotation3d.py:462
↓ 2 callersFunctioneuclidean_translation
Get the translation vector located at the last column of the matrix
unihsi/poselib/poselib/core/rotation3d.py:442
↓ 2 callersMethodeval_actor
(self, obs)
unihsi/learning/amp_network_builder.py:79
↓ 2 callersMethodfrom_dict
Read the object from an ordered dictionary :param dict_repr: the ordered dictionary that is used to construct the object :type dict_
unihsi/poselib/poselib/core/backend/abstract.py:101
next →1–100 of 660, ranked by callers