MCPcopy Create free account

hub / github.com/InternRobotics/UniHSI / functions

Functions660 in github.com/InternRobotics/UniHSI

⨍ Functions 660◇ Types & classes 60
↓ 1 callersMethod_init_train
(self)
unihsi/learning/common_agent.py:523
↓ 1 callersMethod_load_config_params
(self, config)
unihsi/learning/common_agent.py:500
↓ 1 callersMethod_load_mesh
(self)
unihsi/env/tasks/unihsi_partnet_train.py:249
↓ 1 callersMethod_load_mesh
(self)
unihsi/env/tasks/unihsi_scannet_test.py:262
↓ 1 callersMethod_load_mesh
(self)
unihsi/env/tasks/unihsi_partnet.py:249
↓ 1 callersMethod_load_mesh
(self)
unihsi/env/tasks/unihsi_partnet_test.py:263
↓ 1 callersMethod_load_mesh
(self)
unihsi/env/tasks/unihsi_scannet.py:243
↓ 1 callersMethod_load_motion
(self, motion_file)
unihsi/env/tasks/humanoid_amp.py:161
↓ 1 callersMethod_load_motions
(self, motion_file)
unihsi/utils/motion_lib.py:229
↓ 1 callersMethod_load_pcd
(self, min_mesh_dict)
unihsi/env/tasks/unihsi_partnet_train.py:338
↓ 1 callersMethod_load_pcd
(self, min_mesh_dict)
unihsi/env/tasks/unihsi_partnet.py:338
↓ 1 callersMethod_load_pcd
(self, min_mesh_dict)
unihsi/env/tasks/unihsi_partnet_test.py:352
↓ 1 callersMethod_local_rotation_to_dof
(self, local_rot)
unihsi/utils/motion_lib.py:367
↓ 1 callersMethod_local_rotation_to_dof_vel
(self, local_rot0, local_rot1, dt)
unihsi/utils/motion_lib.py:397
↓ 1 callersMethod_log_train_info
(self, train_info, frame)
unihsi/learning/common_agent.py:582
↓ 1 callersMethod_physics_step
(self)
unihsi/env/tasks/base_task.py:410
↓ 1 callersMethod_post_step
(self, info)
unihsi/learning/common_player.py:211
↓ 1 callersMethod_preproc_amp_obs
(self, amp_obs)
unihsi/learning/amp_players.py:89
↓ 1 callersMethod_record_train_batch_info
(self, batch_dict, train_info)
unihsi/learning/common_agent.py:579
↓ 1 callersMethod_record_train_batch_info
(self, batch_dict, train_info)
unihsi/learning/amp_agent.py:632
↓ 1 callersMethod_reset_actors
(self, env_ids)
unihsi/env/tasks/unihsi_partnet_train.py:544
↓ 1 callersMethod_reset_actors
(self, env_ids)
unihsi/env/tasks/humanoid.py:440
↓ 1 callersMethod_reset_actors
(self, env_ids)
unihsi/env/tasks/unihsi_scannet_test.py:448
↓ 1 callersMethod_reset_actors
(self, env_ids)
unihsi/env/tasks/unihsi_partnet.py:544
↓ 1 callersMethod_reset_actors
(self, env_ids)
unihsi/env/tasks/unihsi_partnet_test.py:558
↓ 1 callersMethod_reset_actors
(self, env_ids)
unihsi/env/tasks/unihsi_scannet.py:429
↓ 1 callersMethod_reset_env_tensors
(self, env_ids)
unihsi/env/tasks/humanoid.py:178
↓ 1 callersMethod_reset_envs
(self, env_ids)
unihsi/env/tasks/humanoid.py:170
↓ 1 callersMethod_reset_hybrid_state_init
(self, env_ids)
unihsi/env/tasks/humanoid_amp.py:263
↓ 1 callersMethod_reset_target
(self, env_ids, success)
unihsi/env/tasks/unihsi_partnet_train.py:550
↓ 1 callersMethod_reset_target
(self, env_ids, success, big_force)
unihsi/env/tasks/unihsi_scannet_test.py:454
↓ 1 callersMethod_reset_target
(self, env_ids, success)
unihsi/env/tasks/unihsi_partnet.py:550
↓ 1 callersMethod_reset_target
(self, env_ids, success, big_force)
unihsi/env/tasks/unihsi_partnet_test.py:564
↓ 1 callersMethod_reset_target
(self, env_ids, success)
unihsi/env/tasks/unihsi_scannet.py:435
↓ 1 callersMethod_reset_task
(self, env_ids)
unihsi/env/tasks/humanoid_amp_task.py:76
↓ 1 callersMethod_set_aspect_equal_2d
(self, zero_centered=True)
unihsi/poselib/poselib/visualization/plt_plotter.py:199
↓ 1 callersMethod_set_aspect_equal_3d
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:398
↓ 1 callersMethod_set_env_state
(self, env_ids, root_pos, root_rot, dof_pos, root_vel, root_ang_vel, dof_vel)
unihsi/env/tasks/humanoid_amp.py:313
↓ 1 callersMethod_set_lim
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:109
↓ 1 callersMethod_set_lim
(self)
unihsi/poselib/poselib/visualization/plt_plotter.py:297
↓ 1 callersMethod_set_path
(self, path)
unihsi/utils/logger.py:64
↓ 1 callersMethod_setup_action_space
(self)
unihsi/learning/common_agent.py:514
↓ 1 callersMethod_setup_action_space
(self)
unihsi/learning/common_player.py:223
↓ 1 callersMethod_setup_character_props
(self, key_bodies)
unihsi/env/tasks/humanoid.py:201
↓ 1 callersMethod_shuffle_idx_buf
(self)
unihsi/learning/amp_datasets.py:57
↓ 1 callersMethod_store_replay_amp_obs
(self, amp_obs)
unihsi/learning/amp_agent.py:615
↓ 1 callersMethod_update
(self, lines, dots)
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:86
↓ 1 callersMethod_update
(self, vel_lines, avel_lines)
unihsi/poselib/poselib/visualization/skeleton_plotter_tasks.py:193
↓ 1 callersMethod_update_amp_demos
(self)
unihsi/learning/amp_agent.py:566
↓ 1 callersMethod_update_debug_viz
(self)
unihsi/env/tasks/humanoid.py:543
↓ 1 callersMethod_update_hist_amp_obs
(self, env_ids=None)
unihsi/env/tasks/humanoid_amp.py:323
↓ 1 callersMethod_update_impl
(self, task_list: List[BasePlotterTask])
unihsi/poselib/poselib/visualization/core.py:95
↓ 1 callersMethod_update_task
(self)
unihsi/env/tasks/humanoid_amp_task.py:68
↓ 1 callersMethodafter_init
(self, algo)
unihsi/run.py:91
↓ 1 callersMethodafter_print_stats
(self, frame, epoch_num, total_time)
unihsi/run.py:111
↓ 1 callersFunctionangle_axis_to_exp_map
(angle, axis)
unihsi/utils/torch_utils.py:58
↓ 1 callersFunctionappend_children
Depth first search function :param joint: joint item in the fbx :param pos: position of current element (for parenting)
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:240
↓ 1 callersFunctionbuild_alg_runner
(algo_observer)
unihsi/run.py:173
↓ 1 callersMethodbuild_amp_obs_demo
(self, motion_ids, motion_times0)
unihsi/env/tasks/humanoid_amp.py:121
↓ 1 callersFunctioncompute_contact_reward
(target, root_state, pcd_buffer, joint_pos_buffer, prev_root_pos, dt, contact_type
unihsi/env/tasks/unihsi_partnet_train.py:804
↓ 1 callersFunctioncompute_contact_reward
(target, root_state, pcd_buffer, joint_pos_buffer, prev_root_pos, dt, contact_type
unihsi/env/tasks/unihsi_scannet_test.py:755
↓ 1 callersFunctioncompute_contact_reward
(target, root_state, pcd_buffer, joint_pos_buffer, prev_root_pos, dt, contact_type
unihsi/env/tasks/unihsi_partnet.py:803
↓ 1 callersFunctioncompute_contact_reward
(target, root_state, pcd_buffer, joint_pos_buffer, prev_root_pos, dt, contact_type
unihsi/env/tasks/unihsi_partnet_test.py:836
↓ 1 callersFunctioncompute_contact_reward
(target, root_state, pcd_buffer, joint_pos_buffer, prev_root_pos, dt, contact_type
unihsi/env/tasks/unihsi_scannet.py:717
↓ 1 callersFunctioncompute_humanoid_observations_max
(body_pos, body_rot, body_vel, body_ang_vel, local_root_obs, root_height_obs)
unihsi/env/tasks/humanoid.py:621
↓ 1 callersFunctioncompute_humanoid_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, strike_body_ids, max_episode_length,
unihsi/env/tasks/unihsi_partnet_train.py:910
↓ 1 callersFunctioncompute_humanoid_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, max_episod
unihsi/env/tasks/humanoid.py:674
↓ 1 callersFunctioncompute_humanoid_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, strike_body_ids, max_episode_length,
unihsi/env/tasks/unihsi_scannet_test.py:859
↓ 1 callersFunctioncompute_humanoid_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, strike_body_ids, max_episode_length,
unihsi/env/tasks/unihsi_partnet.py:909
↓ 1 callersFunctioncompute_humanoid_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, strike_body_ids, max_episode_length,
unihsi/env/tasks/unihsi_partnet_test.py:942
↓ 1 callersFunctioncompute_humanoid_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, strike_body_ids, max_episode_length,
unihsi/env/tasks/unihsi_scannet.py:821
↓ 1 callersFunctioncompute_humanoid_reward
(obs_buf)
unihsi/env/tasks/humanoid.py:668
↓ 1 callersFunctioncompute_strike_observations
(root_states, tar_pos, joint_pos_buffer, pcd_buffer, tar_rot, tar_vel, tar_ang_vel, contact_type, contact_vali
unihsi/env/tasks/unihsi_partnet_train.py:755
↓ 1 callersFunctioncompute_strike_observations
(root_states, tar_pos, joint_pos_buffer, pcd_buffer, tar_rot, tar_vel, tar_ang_vel, contact_type, contact_vali
unihsi/env/tasks/unihsi_scannet_test.py:695
↓ 1 callersFunctioncompute_strike_observations
(root_states, tar_pos, joint_pos_buffer, pcd_buffer, tar_rot, tar_vel, tar_ang_vel, contact_type, contact_vali
unihsi/env/tasks/unihsi_partnet.py:754
↓ 1 callersFunctioncompute_strike_observations
(root_states, tar_pos, joint_pos_buffer, pcd_buffer, tar_rot, tar_vel, tar_ang_vel, contact_type, contact_vali
unihsi/env/tasks/unihsi_partnet_test.py:787
↓ 1 callersFunctioncompute_strike_observations
(root_states, tar_pos, joint_pos_buffer, pcd_buffer, tar_rot, tar_vel, tar_ang_vel, contact_type, contact_vali
unihsi/env/tasks/unihsi_scannet.py:657
↓ 1 callersFunctionconfigure_gym
(gym, config)
unihsi/utils/gym_util.py:51
↓ 1 callersFunctioncopy_file
(source_path, destination_path)
data/cp_scannet_test.py:9
↓ 1 callersFunctioncreate_rlgpu_env
(**kwargs)
unihsi/run.py:55
↓ 1 callersMethodcreate_sim
(self, compute_device, graphics_device, physics_engine, sim_params)
unihsi/env/tasks/base_task.py:118
↓ 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.
unihsi/poselib/poselib/skeleton/skeleton3d.py:743
↓ 1 callersFunctionexp_map_to_angle_axis
(exp_map)
unihsi/utils/torch_utils.py:97
↓ 1 callersFunctionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
unihsi/env/tasks/unihsi_partnet_train.py:31
↓ 1 callersFunctionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
unihsi/env/tasks/unihsi_scannet_test.py:35
↓ 1 callersFunctionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
unihsi/env/tasks/unihsi_partnet.py:31
↓ 1 callersFunctionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
unihsi/env/tasks/unihsi_partnet_test.py:30
↓ 1 callersFunctionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
unihsi/env/tasks/unihsi_scannet.py:33
↓ 1 callersFunctionfbx_to_array
Reads an fbx file to an array. :param fbx_file_path: str, file path to fbx :return: tuple with joint_names, parents, transforms, frame t
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_read_wrapper.py:25
↓ 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
unihsi/poselib/poselib/skeleton/backend/fbx/fbx_backend.py:48
↓ 1 callersMethodfetch_amp_obs_demo
(self, num_samples)
unihsi/env/tasks/humanoid_amp.py:100
↓ 1 callersMethodfrom_file
Read the object from a file (either .npy or .json) :param path: path of the file :type path: string :param args, kwargs: the
unihsi/poselib/poselib/core/backend/abstract.py:120
↓ 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
unihsi/poselib/poselib/skeleton/skeleton3d.py:1081
↓ 1 callersMethodget_action
(self, obs_dict, is_determenistic = False)
unihsi/learning/common_player.py:175
↓ 1 callersMethodget_action_size
(self)
unihsi/env/tasks/humanoid.py:138
↓ 1 callersFunctionget_args
(benchmark=False)
unihsi/utils/config.py:186
↓ 1 callersMethodget_disc_logit_weights
(self)
unihsi/learning/amp_network_builder.py:114
↓ 1 callersMethodget_disc_weights
(self)
unihsi/learning/amp_network_builder.py:117
↓ 1 callersFunctionget_height_map
Load region meshes of a scenes Args: points: scene point cloud HEIGHT_MAP_DIM: height map dimension Return: Ret
unihsi/env/tasks/unihsi_partnet_train.py:54
↓ 1 callersFunctionget_height_map
Load region meshes of a scenes Args: points: scene point cloud HEIGHT_MAP_DIM: height map dimension Return: Ret
unihsi/env/tasks/unihsi_scannet_test.py:58
← previousnext →201–300 of 660, ranked by callers