MCPcopy Create free account

hub / github.com/InternRobotics/UniHSI / functions

Functions660 in github.com/InternRobotics/UniHSI

⨍ Functions 660◇ Types & classes 60
↓ 2 callersMethodget_action_values
(self, obs_dict, rand_action_probs)
unihsi/learning/amp_agent.py:170
↓ 2 callersFunctionget_attr_val_from_sample
Retrieves param value for the given prop and attr from the sample.
unihsi/env/tasks/base_task.py:420
↓ 2 callersMethodkeep_nodes_by_names
( self, node_names: List[str], pairwise_translation=None )
unihsi/poselib/poselib/skeleton/skeleton3d.py:283
↓ 2 callersMethodobs_to_torch
(self, obs)
unihsi/learning/common_player.py:157
↓ 2 callersMethodprocess_infos
(self, infos, done_indices)
unihsi/run.py:97
↓ 2 callersFunctionquat_angle_axis
The (angle, axis) representation of the rotation. The axis is normalized to unit length. The angle is guaranteed to be between [0, pi].
unihsi/poselib/poselib/core/rotation3d.py:255
↓ 2 callersFunctionquat_conjugate
quaternion with its imaginary part negated
unihsi/poselib/poselib/core/rotation3d.py:82
↓ 2 callersFunctionquat_identity_like
Construct identity 3D rotation with the same shape
unihsi/poselib/poselib/core/rotation3d.py:247
↓ 2 callersMethodreset
(self, env_ids=None)
unihsi/run.py:142
↓ 2 callersMethodreset
(self)
unihsi/learning/replay_buffer.py:43
↓ 2 callersMethodretarget_to
Retarget the skeleton state to a target skeleton tree. This is a naive retarget implementation with rough approximations. The functi
unihsi/poselib/poselib/skeleton/skeleton3d.py:814
↓ 2 callersMethodsample_motions
(self, n)
unihsi/utils/motion_lib.py:139
↓ 2 callersMethodsample_time
(self, motion_ids, truncate_time=None)
unihsi/utils/motion_lib.py:147
↓ 2 callersMethodset_eval
(self)
unihsi/learning/amp_agent.py:63
↓ 2 callersMethodshow
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:80
↓ 2 callersMethodto_dict
(self)
unihsi/poselib/poselib/skeleton/skeleton3d.py:166
↓ 2 callersFunctiontransform_inverse
Inverse transformation
unihsi/poselib/poselib/core/rotation3d.py:328
↓ 2 callersMethodupdate
(self, frame_index=None, reset_trail=False, skeleton_motion=None)
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:158
↓ 1 callersFunctionVaryPoint
(data, axis, degree)
unihsi/env/tasks/unihsi_partnet_train.py:93
↓ 1 callersFunctionVaryPoint
(data, axis, degree)
unihsi/env/tasks/unihsi_partnet.py:93
↓ 1 callersFunctionVaryPoint
(data, axis, degree)
unihsi/env/tasks/unihsi_partnet_test.py:92
↓ 1 callersMethod__init__
( self, task_name: str, dots: np.ndarray, color: str = "blue", marker_
unihsi/poselib/poselib/visualization/simple_plotter_tasks.py:111
↓ 1 callersMethod__init__
(self, task: "BasePlotterTask")
unihsi/poselib/poselib/visualization/plt_plotter.py:53
↓ 1 callersMethod__init__
( self, task_name: str, skeleton_state, joints_color: str = "red", lin
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:44
↓ 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
unihsi/poselib/poselib/skeleton/skeleton3d.py:99
↓ 1 callersMethod__init__
(self, network)
unihsi/learning/amp_models.py:33
↓ 1 callersMethod_action_to_pd_targets
(self, action)
unihsi/env/tasks/humanoid.py:508
↓ 1 callersMethod_amp_debug
(self, info)
unihsi/learning/amp_players.py:75
↓ 1 callersMethod_amp_debug
(self, info)
unihsi/learning/amp_agent.py:654
↓ 1 callersMethod_build_amp_buffers
(self)
unihsi/learning/amp_agent.py:538
↓ 1 callersMethod_build_amp_obs_demo_buf
(self, num_samples)
unihsi/env/tasks/humanoid_amp.py:139
↓ 1 callersMethod_build_contact_body_ids_tensor
(self, contact_body_names)
unihsi/env/tasks/humanoid.py:495
↓ 1 callersMethod_build_env
(self, env_id, env_ptr, humanoid_asset)
unihsi/env/tasks/humanoid.py:315
↓ 1 callersMethod_build_heightmap_enc
(self, input_shape)
unihsi/learning/amp_network_builder.py:253
↓ 1 callersMethod_build_key_body_ids_tensor
(self, key_body_names)
unihsi/env/tasks/humanoid.py:482
↓ 1 callersMethod_build_net
(self, config)
unihsi/learning/common_player.py:196
↓ 1 callersMethod_build_net_config
(self)
unihsi/learning/common_agent.py:504
↓ 1 callersMethod_build_net_config
(self)
unihsi/learning/common_player.py:214
↓ 1 callersMethod_build_pd_action_offset_scale
(self)
unihsi/env/tasks/humanoid.py:343
↓ 1 callersMethod_build_rand_action_probs
(self)
unihsi/learning/amp_agent.py:460
↓ 1 callersMethod_build_strike_body_ids_tensor
(self, env_ptr, actor_handle, body_names)
unihsi/env/tasks/unihsi_partnet_train.py:531
↓ 1 callersMethod_build_strike_body_ids_tensor
(self, env_ptr, actor_handle, body_names)
unihsi/env/tasks/unihsi_scannet_test.py:434
↓ 1 callersMethod_build_strike_body_ids_tensor
(self, env_ptr, actor_handle, body_names)
unihsi/env/tasks/unihsi_partnet.py:531
↓ 1 callersMethod_build_strike_body_ids_tensor
(self, env_ptr, actor_handle, body_names)
unihsi/env/tasks/unihsi_partnet_test.py:545
↓ 1 callersMethod_build_strike_body_ids_tensor
(self, env_ptr, actor_handle, body_names)
unihsi/env/tasks/unihsi_scannet.py:415
↓ 1 callersMethod_build_termination_heights
(self)
unihsi/env/tasks/humanoid.py:232
↓ 1 callersMethod_calc_advs
(self, batch_dict)
unihsi/learning/common_agent.py:567
↓ 1 callersMethod_calc_amp_rewards
(self, amp_obs)
unihsi/learning/amp_players.py:98
↓ 1 callersMethod_calc_disc_rewards
(self, amp_obs)
unihsi/learning/amp_players.py:105
↓ 1 callersMethod_calc_disc_rewards
(self, amp_obs)
unihsi/learning/amp_agent.py:606
↓ 1 callersMethod_calc_frame_blend
(self, time, len, num_frames, dt)
unihsi/utils/motion_lib.py:334
↓ 1 callersMethod_combine_rewards
(self, task_rewards, amp_rewards)
unihsi/learning/amp_agent.py:576
↓ 1 callersMethod_compute_angular_velocity
(r, time_delta: float, guassian_filter=True)
unihsi/poselib/poselib/skeleton/skeleton3d.py:1261
↓ 1 callersMethod_compute_disc_acc
(self, disc_agent_logit, disc_demo_logit)
unihsi/learning/amp_agent.py:527
↓ 1 callersMethod_compute_motion_dof_vels
(self, motion)
unihsi/utils/motion_lib.py:350
↓ 1 callersMethod_compute_reset
(self)
unihsi/env/tasks/humanoid.py:397
↓ 1 callersMethod_compute_reward
(self, actions)
unihsi/env/tasks/humanoid.py:393
↓ 1 callersMethod_compute_task_obs
(self, env_ids=None)
unihsi/env/tasks/humanoid_amp_task.py:93
↓ 1 callersMethod_compute_velocity
(p, time_delta, guassian_filter=True)
unihsi/poselib/poselib/skeleton/skeleton3d.py:1251
↓ 1 callersMethod_create_envs
(self, num_envs, spacing, num_per_row)
unihsi/env/tasks/humanoid.py:250
↓ 1 callersMethod_create_ground_plane
(self)
unihsi/env/tasks/humanoid.py:192
↓ 1 callersMethod_create_impl
(self, task_list: List[BasePlotterTask])
unihsi/poselib/poselib/visualization/core.py:98
↓ 1 callersMethod_create_mesh_ground
(self)
unihsi/env/tasks/unihsi_partnet_train.py:186
↓ 1 callersMethod_create_mesh_ground
(self)
unihsi/env/tasks/unihsi_scannet_test.py:198
↓ 1 callersMethod_create_mesh_ground
(self)
unihsi/env/tasks/unihsi_partnet.py:186
↓ 1 callersMethod_create_mesh_ground
(self)
unihsi/env/tasks/unihsi_partnet_test.py:200
↓ 1 callersMethod_create_mesh_ground
(self)
unihsi/env/tasks/unihsi_scannet.py:179
↓ 1 callersMethod_disc_loss
(self, disc_agent_logit, disc_demo_logit, obs_demo)
unihsi/learning/amp_agent.py:478
↓ 1 callersMethod_disc_loss_neg
(self, disc_logits)
unihsi/learning/amp_agent.py:517
↓ 1 callersMethod_disc_loss_pos
(self, disc_logits)
unihsi/learning/amp_agent.py:522
↓ 1 callersMethod_draw_task
(self)
unihsi/env/tasks/humanoid_amp_task.py:99
↓ 1 callersMethod_fetch_motion_files
(self, motion_file)
unihsi/utils/motion_lib.py:309
↓ 1 callersFunction_get_animation_curve
(joint, fbx_scene)
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:177
↓ 1 callersFunction_get_frame_count
(fbx_scene)
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:152
↓ 1 callersMethod_get_humanoid_collision_filter
(self)
unihsi/env/tasks/humanoid.py:390
↓ 1 callersMethod_get_mean_rewards
(self)
unihsi/learning/common_agent.py:497
↓ 1 callersMethod_get_num_bodies
(self)
unihsi/utils/motion_lib.py:345
↓ 1 callersMethod_get_obj_parts
(self, object, contact_parts, label_mapping, label, pcd, stand_point)
unihsi/env/tasks/unihsi_partnet_train.py:464
↓ 1 callersMethod_get_obj_parts
(self, contact_parts, label_mapping, label, pcd, stand_point)
unihsi/env/tasks/unihsi_scannet_test.py:381
↓ 1 callersMethod_get_obj_parts
(self, object, contact_parts, label_mapping, label, pcd, stand_point)
unihsi/env/tasks/unihsi_partnet.py:464
↓ 1 callersMethod_get_obj_parts
(self, object, contact_parts, label_mapping, label, pcd, stand_point)
unihsi/env/tasks/unihsi_partnet_test.py:478
↓ 1 callersMethod_get_obj_parts
(self, contact_parts, label_mapping, label, pcd, stand_point)
unihsi/env/tasks/unihsi_scannet.py:362
↓ 1 callersMethod_get_part_labels
(self, partnet_id)
unihsi/env/tasks/unihsi_partnet_train.py:440
↓ 1 callersMethod_get_part_labels
(self, scene_id, obj_id, part_ids)
unihsi/env/tasks/unihsi_scannet_test.py:358
↓ 1 callersMethod_get_part_labels
(self, partnet_id)
unihsi/env/tasks/unihsi_partnet.py:440
↓ 1 callersMethod_get_part_labels
(self, partnet_id)
unihsi/env/tasks/unihsi_partnet_test.py:454
↓ 1 callersMethod_get_part_labels
(self, scene_id, obj_id, part_ids)
unihsi/env/tasks/unihsi_scannet.py:339
↓ 1 callersMethod_get_path
(self)
unihsi/utils/logger.py:67
↓ 1 callersMethod_get_pcd_parts
(self, pcd_list)
unihsi/env/tasks/unihsi_partnet_train.py:393
↓ 1 callersMethod_get_pcd_parts
(self, pcd_list)
unihsi/env/tasks/unihsi_scannet_test.py:311
↓ 1 callersMethod_get_pcd_parts
(self, pcd_list)
unihsi/env/tasks/unihsi_partnet.py:393
↓ 1 callersMethod_get_pcd_parts
(self, pcd_list)
unihsi/env/tasks/unihsi_partnet_test.py:407
↓ 1 callersMethod_get_pcd_parts
(self, pcd_list)
unihsi/env/tasks/unihsi_scannet.py:292
↓ 1 callersFunction_get_skeleton
(root_joint)
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:233
↓ 1 callersMethod_init_amp_demo_buf
(self)
unihsi/learning/amp_agent.py:556
↓ 1 callersMethod_init_amp_obs
(self, env_ids)
unihsi/env/tasks/humanoid_amp.py:278
↓ 1 callersMethod_init_amp_obs_default
(self, env_ids)
unihsi/env/tasks/humanoid_amp.py:290
↓ 1 callersMethod_init_amp_obs_ref
(self, env_ids, motion_ids, motion_times)
unihsi/env/tasks/humanoid_amp.py:295
↓ 1 callersMethod_init_camera
(self)
unihsi/env/tasks/humanoid.py:512
↓ 1 callersMethod_init_data_buf
(self, data_dict)
unihsi/learning/replay_buffer.py:105
← previousnext →101–200 of 660, ranked by callers