MCPcopy Create free account

hub / github.com/LeCAR-Lab/BFM-Zero / functions

Functions1,070 in github.com/LeCAR-Lab/BFM-Zero

↓ 3 callersFunctioncopysign
(a, b)
humanoidverse/utils/torch_utils.py:140
↓ 3 callersMethoddevice
(self)
humanoidverse/agents/fb/agent.py:79
↓ 3 callersFunctionexport_meta_policy_as_onnx
(inference_model, path, exported_policy_name, example_obs_dict, z_dim, history: bool = False, use_29dof: bool
humanoidverse/utils/helpers.py:240
↓ 3 callersFunctionextract_and_concat_vec3
(arr: torch.Tensor)
humanoidverse/agents/evaluations/humanoidverse_isaac.py:148
↓ 3 callersFunctionextract_model
(agent_or_model)
humanoidverse/agents/evaluations/base.py:20
↓ 3 callersFunctionfind_start_stop_traj
(end: torch.Tensor, at_capacity: bool = True, cursor=None | int)
humanoidverse/agents/buffers/trajectory.py:276
↓ 3 callersMethodfk_batch
(self, pose, trans, convert_to_mat=True, return_full = False, dt=1/30)
humanoidverse/utils/motion_lib/torch_humanoid_batch.py:171
↓ 3 callersFunctionget_backward_observation
(env, motion_id, use_root_height_obs: bool = False, velocity_multiplier: float = 1.0)
humanoidverse/utils/helpers.py:157
↓ 3 callersMethodinitialise
(self, observation: np.ndarray | Dict[str, np.ndarray], info: Dict)
humanoidverse/agents/evaluations/humanoidverse_isaac.py:37
↓ 3 callersFunctionlayernorm
(input_dim, num_parallel=1)
humanoidverse/agents/nn_models.py:181
↓ 3 callersMethodload
(cls, path: str, device: str | None = None, strict: bool = True)
humanoidverse/agents/base_model.py:90
↓ 3 callersMethodload
(cls, path: str, device: str | None = None)
humanoidverse/agents/buffers/transition.py:206
↓ 3 callersFunctionload_model_from_checkpoint_dir
(checkpoint_dir: str, device: str = "cpu")
humanoidverse/agents/load_utils.py:37
↓ 3 callersFunctionnormalize
(x, eps: float = 1e-9)
humanoidverse/simulator/genesis/tmp_gs_utils.py:94
↓ 3 callersFunctionquat_identity_like
Construct identity quaternion with same shape.
humanoidverse/utils/torch_utils.py:615
↓ 3 callersFunctionquaternion_raw_multiply
Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part
humanoidverse/utils/motion_lib/motion_utils/rotation_conversions.py:357
↓ 3 callersMethodquery
(self, key: str, filter_by_length: bool = True)
humanoidverse/agents/envs/utils/history_handler.py:35
↓ 3 callersMethodrefresh_sim_tensors
(self)
humanoidverse/simulator/mujoco/mujoco.py:227
↓ 3 callersMethodrender
( self, camera: str | None = None, scene_option: Optional[mujoco.MjvOption] = None,
humanoidverse/envs/g1_env_helper/base.py:72
↓ 3 callersMethodreward_inference
(self, task: str)
humanoidverse/envs/g1_env_helper/bench/reward_eval_hv.py:75
↓ 3 callersFunctionrot2eul
(R: np.ndarray)
humanoidverse/envs/g1_env_helper/rewards.py:61
↓ 3 callersMethodsave
(self, output_folder: str)
humanoidverse/agents/fb/agent.py:411
↓ 3 callersMethodset_actor_root_state_tensor
(self, set_env_ids, root_states)
humanoidverse/simulator/mujoco/mujoco.py:240
↓ 3 callersMethodset_task
(self, task: g1_rewards.RewardFunction | str | None)
humanoidverse/envs/g1_env_helper/robot.py:271
↓ 3 callersFunctionslerp
Spherical linear interpolation between two quaternions.
humanoidverse/utils/torch_utils.py:382
↓ 3 callersFunctiontask_to_xml
Resolve task name to the single G1 scene XML path (package data dir).
humanoidverse/envs/g1_env_helper/robot.py:43
↓ 3 callersFunctionto_torch
(x, dtype=torch.float, device='cuda:0', requires_grad=False)
humanoidverse/utils/torch_utils.py:18
↓ 3 callersFunctionto_torch
(tensor)
humanoidverse/utils/motion_lib/motion_lib_base.py:42
↓ 3 callersMethodupdate_fb
( self, obs: torch.Tensor | dict[str, torch.Tensor], action: torch.Tensor, dis
humanoidverse/agents/fb/agent.py:216
↓ 3 callersMethodupdate_view_location
Updates the camera view pose based on the current viewer origin and the eye and lookat positions. Args: eye: The eye position of
humanoidverse/simulator/isaacsim/isaaclab_viewpoint_camera_controller.py:165
↓ 2 callersFunction_angle_from_tan
Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:
humanoidverse/utils/motion_lib/motion_utils/rotation_conversions.py:211
↓ 2 callersFunction_build_batch_forward
(cfg, obs_space, z_dim, action_dim, output_dim=None)
humanoidverse/agents/nn_models.py:138
↓ 2 callersMethod_calc_frame_blend
(self, time, len, num_frames, dt)
humanoidverse/utils/motion_lib/motion_lib_base.py:597
↓ 2 callersMethod_compute_angular_velocity
(r, time_delta: float, guassian_filter=True)
humanoidverse/utils/motion_lib/torch_humanoid_batch.py:290
↓ 2 callersMethod_compute_observations
Computes observations
humanoidverse/envs/legged_base_task/legged_robot_base.py:594
↓ 2 callersMethod_compute_torques
Compute torques from actions. Actions can be interpreted as position or velocity targets given to a PD controller, or directly as scaled
humanoidverse/envs/legged_base_task/legged_robot_base.py:637
↓ 2 callersMethod_compute_velocity
(p, time_delta, guassian_filter=True)
humanoidverse/utils/motion_lib/torch_humanoid_batch.py:280
↓ 2 callersMethod_create_trimesh
Adds a triangle mesh terrain to the simulation, sets parameters based on the cfg.
humanoidverse/simulator/isaacgym/isaacgym.py:131
↓ 2 callersMethod_draw_debug_vis
Draws visualizations for dubugging (slows down simulation a lot). Default behaviour: draws height measurement points
humanoidverse/envs/legged_base_task/legged_robot_base.py:675
↓ 2 callersMethod_get_g1env_observation
Turn current Isaac sim state into a G1Env-like observation.
humanoidverse/agents/envs/humanoidverse_isaac.py:404
↓ 2 callersMethod_get_obs
(self, data: mujoco.MjData, info: dict[str, Any], rng: np.random.RandomState, last_action: np.ndarray)
humanoidverse/envs/g1_env_helper/robot_29dof.py:278
↓ 2 callersMethod_get_obs
(self, data: mujoco.MjData, info: dict[str, Any], rng: np.random.RandomState, last_action: np.ndarray)
humanoidverse/envs/g1_env_helper/robot.py:224
↓ 2 callersMethod_get_pairwise_average_translation
(self)
humanoidverse/utils/motion_lib/skeleton.py:839
↓ 2 callersFunction_index_from_letter
(letter: str)
humanoidverse/utils/motion_lib/motion_utils/rotation_conversions.py:242
↓ 2 callersMethod_post_compute_observations_callback
(self)
humanoidverse/envs/legged_base_task/legged_robot_base.py:373
↓ 2 callersMethod_prepare_for_train
(self)
humanoidverse/agents/fb/model.py:95
↓ 2 callersMethod_remapped_to
( self, joint_mapping: Dict[str, str], target_skeleton_tree: SkeletonTree )
humanoidverse/utils/motion_lib/skeleton.py:902
↓ 2 callersMethod_resample_motion_time_and_ids
(self, env_ids)
humanoidverse/envs/legged_robot_motions/legged_robot_motions.py:161
↓ 2 callersMethod_reset_dofs
Resets DOF position and velocities of selected environmments Positions are randomly selected within 0.5:1.5 x default positions. Velo
humanoidverse/envs/legged_base_task/legged_robot_base.py:943
↓ 2 callersMethod_reset_root_states
Resets ROOT states position and velocities of selected environmments if target_root_states is not None, reset to target_root_states
humanoidverse/envs/legged_base_task/legged_robot_base.py:961
↓ 2 callersMethod_sample_tracking_z
(self, replay_buffer, batch_dim, traj_length)
humanoidverse/agents/fb/agent.py:345
↓ 2 callersMethod_sim2real_costs
(self, data, action, last_action, contact)
humanoidverse/envs/g1_env_helper/robot.py:283
↓ 2 callersMethodactor
(self, obs: torch.Tensor | dict[str, torch.Tensor], z: torch.Tensor, std: float)
humanoidverse/agents/fb/model.py:115
↓ 2 callersFunctionadd_arrow_from_xpos_to_direction
(renderer, xpos, vector, rgba)
humanoidverse/envs/g1_env_helper/rewards.py:55
↓ 2 callersMethodadd_keyboard_callback
(self, key, callback)
humanoidverse/simulator/isaacsim/isaacsim.py:447
↓ 2 callersMethodadd_terrain_to_map
(self, terrain, row, col)
humanoidverse/envs/env_utils/terrain.py:96
↓ 2 callersFunctionadd_visual_arrow
Adds an arrow to an mjvScene.
humanoidverse/envs/g1_env_helper/rewards.py:23
↓ 2 callersMethodbuild
(self, obs_space: spaces.Space)
humanoidverse/agents/normalizers.py:48
↓ 2 callersMethodbuild
(self, obs_space, action_dim: int)
humanoidverse/agents/fb_cpr_aux/agent.py:37
↓ 2 callersFunctioncalc_heading
Calculate heading direction from quaternion.
humanoidverse/utils/torch_utils.py:446
↓ 2 callersMethodclear_lines
(self)
humanoidverse/simulator/genesis/genesis.py:414
↓ 2 callersMethodclose
(self)
humanoidverse/envs/gymnasium_wrapper.py:125
↓ 2 callersMethodcopy_to_mujoco
(self)
humanoidverse/simulator/genesis/genesis_mjdebug.py:442
↓ 2 callersMethodcreate_single_env
( self, wrappers: tp.Optional[tp.List[tp.Callable]] = None, shared_motion_lib: tp.Optional[MotionBuffe
humanoidverse/utils/g1_env_config.py:173
↓ 2 callersFunctiondistance_proximity
(next_obs: torch.Tensor, tracking_target: torch.Tensor, bound: float = 2.0, margin: float = 2)
humanoidverse/envs/g1_env_helper/bench/tracking_eval_hv.py:276
↓ 2 callersFunctiondistance_proximity
(next_obs: torch.Tensor, tracking_target: torch.Tensor, bound: float = 2.0, margin: float = 2)
humanoidverse/envs/g1_env_helper/bench/tracking_eval.py:221
↓ 2 callersFunctiondistance_proximity
(next_obs: torch.Tensor, tracking_target: torch.Tensor, bound: float = 2.0, margin: float = 2, prefix="")
humanoidverse/agents/evaluations/humanoidverse_isaac.py:576
↓ 2 callersMethoddraw_sphere
(self, pos, radius, color, env_id, pos_id=0)
humanoidverse/simulator/genesis/genesis.py:418
↓ 2 callersMethoddrop_nodes_by_names
( self, node_names: List[str], pairwise_translation=None )
humanoidverse/utils/motion_lib/skeleton.py:353
↓ 2 callersFunctiondtype_numpytotorch
(np_dtype: Any)
humanoidverse/agents/buffers/transition.py:257
↓ 2 callersFunctionemd_numpy
(next_obs: torch.Tensor, tracking_target: torch.Tensor)
humanoidverse/envs/g1_env_helper/bench/tracking_eval_hv.py:264
↓ 2 callersFunctionemd_numpy
(next_obs: torch.Tensor, tracking_target: torch.Tensor)
humanoidverse/envs/g1_env_helper/bench/tracking_eval.py:209
↓ 2 callersFunctionemd_numpy
(next_obs: torch.Tensor, tracking_target: torch.Tensor, prefix="")
humanoidverse/agents/evaluations/humanoidverse_isaac.py:564
↓ 2 callersMethodencode_expert
( self, next_obs: torch.Tensor | dict[str, torch.Tensor], )
humanoidverse/agents/fb_cpr/agent.py:153
↓ 2 callersMethodepisodic_domain_randomization
(self, env_ids)
humanoidverse/simulator/mujoco/mujoco.py:62
↓ 2 callersMethodexport
(self, path)
humanoidverse/utils/helpers.py:149
↓ 2 callersFunctionextract_values
(d: Dict, idxs: List | torch.Tensor | np.ndarray)
humanoidverse/agents/buffers/transition.py:229
↓ 2 callersFunctionget_euler_xyz_in_tensor
Convert quaternion to Euler angles as tensor (assumes w_last=True format).
humanoidverse/utils/torch_utils.py:526
↓ 2 callersMethodget_full_buffer
(self)
humanoidverse/agents/buffers/transition.py:119
↓ 2 callersFunctionget_g1_robot_xml_root
Path to humanoidverse/data/robots/g1 (scene xmls, meshes, goal_frames). Resolved from package location.
humanoidverse/utils/g1_env_config.py:17
↓ 2 callersMethodget_motion_length
(self, motion_ids=None)
humanoidverse/utils/motion_lib/motion_lib_base.py:590
↓ 2 callersFunctionjson_to_space
Convert JSON data back to a Gymnasium space.
humanoidverse/agents/envs/utils/gym_spaces.py:39
↓ 2 callersMethodkeep_nodes_by_names
( self, node_names: List[str], pairwise_translation=None )
humanoidverse/utils/motion_lib/skeleton.py:396
↓ 2 callersMethodload_motions_for_evaluation
(self, start_idx = 0)
humanoidverse/utils/motion_lib/motion_lib_base.py:313
↓ 2 callersFunctionmake_from_name
( name: str | None = None, )
humanoidverse/envs/g1_env_helper/robot.py:289
↓ 2 callersMethodmake_terrain
(self, terrain_type, difficulty)
humanoidverse/envs/env_utils/terrain.py:84
↓ 2 callersMethodmesh_fk
Load the mesh from the XML file and merge them into the humanoid based on the current pose.
humanoidverse/utils/motion_lib/torch_humanoid_batch.py:368
↓ 2 callersFunctionquat_angle_axis
Convert quaternion to angle-axis representation.
humanoidverse/utils/torch_utils.py:300
↓ 2 callersFunctionquaternion_to_matrix
Convert quaternions to rotation matrices (assumes wxyz format).
humanoidverse/utils/torch_utils.py:653
↓ 2 callersFunctionquaternion_to_matrix
Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten
humanoidverse/utils/motion_lib/motion_utils/rotation_conversions.py:46
↓ 2 callersMethodrefresh_sim_tensors
(self)
humanoidverse/simulator/isaacgym/isaacgym.py:461
↓ 2 callersMethodrefresh_sim_tensors
(self)
humanoidverse/simulator/isaacgym/isaacgym_hoi.py:543
↓ 2 callersMethodrender
(self)
humanoidverse/agents/envs/humanoidverse_isaac.py:523
↓ 2 callersMethodresample_motion
(self)
humanoidverse/envs/legged_robot_motions/legged_robot_motions.py:186
↓ 2 callersMethodreset_parameters
(self)
humanoidverse/agents/nn_models.py:590
↓ 2 callersMethodretarget_to
Retarget the skeleton state to a target skeleton tree. This is a naive retarget implementation with rough approximations. The functi
humanoidverse/utils/motion_lib/skeleton.py:931
↓ 2 callersMethodrun
(self, *, timestep, agent_or_model, replay_buffer, logger, **kwargs)
humanoidverse/agents/evaluations/g1env.py:89
↓ 2 callersMethodsample_mixed_z
(self, train_goal: torch.Tensor, expert_encodings: torch.Tensor, *args, **kwargs)
humanoidverse/agents/fb_cpr/agent.py:127
↓ 2 callersMethodsave
(self, time: int, replay_buffer: Dict[str, tp.Any])
humanoidverse/train.py:578
↓ 2 callersMethodset_dof_state_tensor
(self, set_env_ids, dof_states)
humanoidverse/simulator/mujoco/mujoco.py:252
← previousnext →101–200 of 1,070, ranked by callers