MCPcopy Create free account

hub / github.com/NVlabs/GR00T-WholeBodyControl / functions

Functions6,086 in github.com/NVlabs/GR00T-WholeBodyControl

↓ 8 callersFunctiontensor_to_dict
Construct an ordered dictionary from the object :rtype: OrderedDict
gear_sonic/utils/motion_lib/skeleton.py:139
↓ 8 callersFunctiontorch_safe_atan2
(y, x, eps: float = 1e-8)
gear_sonic/trl/utils/torch_transform.py:108
↓ 8 callersFunctionunitree_hg_msg_dds__LowCmd_
()
external_dependencies/unitree_sdk2_python/unitree_sdk2py/idl/default.py:219
↓ 8 callersMethodupdate_scene
Updates the scene with latest sim data Args: data: structure storing current simulation state scene_option: we use th
decoupled_wbc/dexmg/gr00trobocasa/robocasa/utils/usd/exporter.py:158
↓ 8 callersFunctionzero_loss
Return a zero loss tensor with requires_grad=True.
gear_sonic/trl/losses/token_losses.py:425
↓ 7 callersMethodClose
(self)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/core/channel.py:73
↓ 7 callersMethodGetNumSmplJoints
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/motion_data_reader.hpp:93
↓ 7 callersMethodGetNumSmplPoses
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/motion_data_reader.hpp:94
↓ 7 callersMethodInit
(self)
external_dependencies/unitree_sdk2_python/example/g1/low_level/g1_low_level_example.py:89
↓ 7 callersMethodRead
(self, timeout: float = None)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/core/channel.py:177
↓ 7 callersMethodSetFsmId
(self, fsm_id: int)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/g1/loco/g1_loco_client.py:34
↓ 7 callersMethodSwitchGait
(self, t: int)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/b2/sport/sport_client.py:131
↓ 7 callersMethodSwitchSet
(self, on: bool)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/go2/obstacles_avoid/obstacles_avoid_client.py:24
↓ 7 callersFunction_build_vel_frame_mask
Build a frame mask for velocity (finite-difference) tensors. Velocity at frame i uses frames i and i+1, so frame i is valid only if both fram
gear_sonic/trl/losses/token_losses.py:480
↓ 7 callersFunction_controller_inputs
(snapshot: Any)
gear_sonic/utils/teleop/isaac_teleop_client.py:62
↓ 7 callersMethod_run_in_tmux
Run a command in a new tmux window or pane
decoupled_wbc/scripts/deploy_g1.py:56
↓ 7 callersMethod_snapshot_side
(self, raw: dict[str, Any] | None, side: str)
gear_sonic/utils/teleop/isaac_teleop_client.py:215
↓ 7 callersFunctionangle_axis_to_quaternion
r"""Convert an angle axis to a quaternion. The quaternion vector has components in (x, y, z, w) or (w, x, y, z) format. Adapted from ceres C
gear_sonic/trl/utils/kornia_transform.py:849
↓ 7 callersMethodcache_forward_kinematics
Perform forward kinematics to update the pose of every joint and frame in the Pinocchio data structures for the given configuration `
gear_sonic/data/robot_model/robot_model.py:213
↓ 7 callersFunctioncalc_heading_quat_d
Calculate heading quaternion from full quaternion (wxyz format) - double version
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/math_utils.hpp:300
↓ 7 callersMethodchange_state
Cycle through states: IDLE -> RECORDING -> NEED_TO_SAVE -> IDLE.
decoupled_wbc/control/utils/episode_state.py:17
↓ 7 callersFunctioncompares_unordered
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:22975
↓ 7 callersFunctioncompares_unordered
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/nlohmann/json.hpp:22975
↓ 7 callersMethoddump
! @brief internal implementation of the serialization function This function is called by the public member function dump and organizes t
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:18115
↓ 7 callersMethodextract_root
Extracts root motion features from input. Args: motion (torch.Tensor): Motion features of shape [..., D] Returns:
motionbricks/motionbricks/motionlib/core/motion_reps/motion_reps_base/seperate_root_local_body.py:210
↓ 7 callersFunctionfind
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:19290
↓ 7 callersMethodgetVertexCount
(self, excludeMaskedVerts=False)
decoupled_wbc/control/teleop/gui/core/module3d.py:570
↓ 7 callersMethodget_action
Compute and return the next action at the specified time, based on current observation and planner command. Args: time: O
decoupled_wbc/control/base/policy.py:30
↓ 7 callersMethodget_hand_actuated_joint_indices
Get the indices of hand actuated joints in the full configuration. Ordering is that of the actuated joints as defined in the suppleme
gear_sonic/data/robot_model/robot_model.py:163
↓ 7 callersMethodget_num_codes
(self, include_aug_tokens: bool)
motionbricks/motionbricks/motion_backbone/neural_modules/pose_backbone.py:177
↓ 7 callersMethodget_num_heads
brief: num heads for the tokenizers
motionbricks/motionbricks/motion_backbone/neural_modules/pose_backbone.py:203
↓ 7 callersFunctionget_robot_type_and_model
Get the robot type from the robot name.
decoupled_wbc/control/robot_model/instantiation/__init__.py:4
↓ 7 callersMethodget_state
Get current state.
decoupled_wbc/control/utils/episode_state.py:30
↓ 7 callersMethodglobal_to_local
Convert features from global to local representation. Args: features: Input features in global representation is_norm
motionbricks/motionbricks/motionlib/core/motion_reps/motion_reps_base/dual_root_local_body.py:236
↓ 7 callersFunctionif
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:22946
↓ 7 callersFunctionif
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/nlohmann/json.hpp:22946
↓ 7 callersFunctionmatrix_to_quaternion
Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Returns:
motionbricks/motionbricks/motionlib/core/utils/rotations.py:111
↓ 7 callersFunctionprint_command
(char, info)
decoupled_wbc/control/main/teleop/run_sync_sim_data_collection.py:38
↓ 7 callersMethodpublish
(self, key: str)
decoupled_wbc/tests/control/main/teleop/test_g1_control_loop.py:32
↓ 7 callersFunctionrandomize_materials_rgba
( rng: np.random.Generator, mjcf_obj: MJCFObject, gradient: Gradient, linear: bool )
decoupled_wbc/dexmg/gr00trobocasa/robocasa/utils/visuals_utls.py:20
↓ 7 callersMethodrender
(self)
decoupled_wbc/control/envs/robocasa/sync_env.py:238
↓ 7 callersMethodsample
(self, fixtures=None, reference=None, on_top=True)
decoupled_wbc/dexmg/gr00trobocasa/robocasa/utils/placement_samplers.py:708
↓ 7 callersFunctionstringToPoseArray
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/bindings/py_bindings.cpp:71
↓ 7 callersMethodtimer
Returns a context manager for timing an operation.
gear_sonic/utils/data_collection/telemetry.py:45
↓ 7 callersMethodwidth
Return the width of the Image in pixels.
decoupled_wbc/control/teleop/gui/library/image.py:132
↓ 6 callersMethodGetInputDimension
* @brief Get input dimension * @return Input dimension size */
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/encoder.hpp:317
↓ 6 callersMethodGetTensorShape
gear_sonic_deploy/src/TRTInference/InferenceEngine.cpp:766
↓ 6 callersMethodInit
(self)
external_dependencies/unitree_sdk2_python/example/b2w/low_level/b2w_stand_example.py:56
↓ 6 callersMethodNeedsByteSwap
@return True if the payload byte order differs from the native CPU order.
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/input_interface/zmq_packed_message_subscriber.hpp:170
↓ 6 callersMethodRecoveryStand
(self)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/b2/sport/sport_client.py:97
↓ 6 callersMethodSetVolume
(self, level: int)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/b2/vui/vui_client.py:47
↓ 6 callersMethodStandUp
(self)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/b2/sport/sport_client.py:83
↓ 6 callersMethodStop
Stop the background thread and close the socket.
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/input_interface/zmq_packed_message_subscriber.hpp:241
↓ 6 callersMethodTriggerSafetyReset
Called by the manager to request a safety reset on this interface.
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/input_interface/input_interface.hpp:437
↓ 6 callersMethod__init__
(self, name, joints=None)
decoupled_wbc/dexmg/gr00trobocasa/robocasa/models/objects/xml_objects.py:200
↓ 6 callersMethod_get_objects
(self)
decoupled_wbc/dexmg/gr00trobocasa/robocasa/environments/locomanipulation/locomanip_basic.py:207
↓ 6 callersMethod_get_tracker_data
Poll current tracking data and return it as a dictionary. Returns: Dict with keys ``left_controller``, ``right_controller``, ``he
gear_sonic/utils/teleop/isaac_teleop_client.py:172
↓ 6 callersMethod_setup_references
(self)
decoupled_wbc/dexmg/gr00trobocasa/robocasa/environments/locomanipulation/base.py:731
↓ 6 callersMethodcache_forward_kinematics
Perform forward kinematics using the reduced configuration. :param q_reduced: Configuration in reduced space
decoupled_wbc/control/robot_model/robot_model.py:738
↓ 6 callersFunctioncanonicalPath
Return canonical name for location specified by path. Useful for comparing paths.
decoupled_wbc/control/teleop/gui/library/getpath.py:113
↓ 6 callersMethodchange_first_heading
Transform motion to be relative to the first frame. Args: motion: Input motion features is_normalized: Whether input
motionbricks/motionbricks/motionlib/core/motion_reps/motion_reps_base/dual_root_local_body.py:316
↓ 6 callersMethodclose
Close OpenXR session, stop DeviceIO + CloudXR runtime.
gear_sonic/utils/teleop/isaac_teleop_client.py:303
↓ 6 callersFunctioncont6d_to_matrix
(cont6d)
motionbricks/motionbricks/motionlib/core/utils/rotations.py:44
↓ 6 callersMethodconvert_gripper_q
(self, q: np.ndarray, side: str = "left")
decoupled_wbc/control/envs/robocasa/utils/robocasa_env.py:336
↓ 6 callersFunctiondecoder_output_to_egocentric_transforms
Convert decoder output to joint positions and 6D rotations (ortho6d). Always returns 6D rotations. For geodesic loss, convert to matrices in the
gear_sonic/trl/losses/token_losses.py:47
↓ 6 callersMethoddevice
Return the device of the humanoid tensors.
gear_sonic/utils/motion_lib/torch_humanoid_batch.py:242
↓ 6 callersMethodfinish
gear_sonic_deploy/src/TRTInference/picosha2.h:205
↓ 6 callersMethodgetFaceCount
(self, excludeMaskedFaces=False)
decoupled_wbc/control/teleop/gui/core/module3d.py:694
↓ 6 callersMethodget_object_root_pos
(self, motion_ids, motion_steps)
gear_sonic/utils/motion_lib/motion_lib_base.py:673
↓ 6 callersMethodget_root_pos_w
(self, motion_ids, motion_steps)
gear_sonic/utils/motion_lib/motion_lib_base.py:610
↓ 6 callersMethodget_state
Get current state.
gear_sonic/utils/data_collection/episode_state.py:30
↓ 6 callersMethodhands
(self)
decoupled_wbc/control/envs/g1/g1_env.py:105
↓ 6 callersFunctionis_string
@brief return whether value is a string @sa https://json.nlohmann.me/api/basic_json/is_string/
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:20673
↓ 6 callersFunctionis_string
@brief return whether value is a string @sa https://json.nlohmann.me/api/basic_json/is_string/
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/nlohmann/json.hpp:20673
↓ 6 callersFunctionjson_pointer
@brief create JSON pointer @sa https://json.nlohmann.me/api/json_pointer/json_pointer/
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:13867
↓ 6 callersFunctionjson_pointer
@brief create JSON pointer @sa https://json.nlohmann.me/api/json_pointer/json_pointer/
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/nlohmann/json.hpp:13867
↓ 6 callersFunctionload_motion_rep
(conf: DictConfig)
motionbricks/motionbricks/helper/pl_util.py:5
↓ 6 callersMethodload_motions
( self, random_sample=True, start_idx=0, max_len=-1, target_heading=No
gear_sonic/utils/motion_lib/motion_lib_base.py:1006
↓ 6 callersFunctionnormalize
L2-normalize a tensor along its last dimension. Args: x: Input tensor of any shape. eps: Minimum norm value to clamp to, preventi
gear_sonic/utils/torch_utils.py:109
↓ 6 callersFunctionnormalize_vec
(x: torch.Tensor, dim: int = -1, eps: float = 1e-9)
motionbricks/motionbricks/motionlib/core/utils/torch_utils.py:7
↓ 6 callersMethodnull
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:7298
↓ 6 callersMethodnull
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/nlohmann/json.hpp:7298
↓ 6 callersMethodpublish
(self, msg: dict)
decoupled_wbc/control/utils/ros_utils.py:93
↓ 6 callersFunctionpush_back
@brief append an unescaped token at the end of the reference pointer @sa https://json.nlohmann.me/api/json_pointer/push_back/
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/aarch64/nlohmann/json.hpp:13988
↓ 6 callersFunctionpush_back
@brief append an unescaped token at the end of the reference pointer @sa https://json.nlohmann.me/api/json_pointer/push_back/
external_dependencies/XRoboToolkit-PC-Service-Pybind_X86_and_ARM64/include/nlohmann/json.hpp:13988
↓ 6 callersFunctionquat_conjugate
(a)
motionbricks/motionbricks/motionlib/core/utils/rotations.py:29
↓ 6 callersMethodrun
(self)
decoupled_wbc/control/teleop/device/manus.py:111
↓ 6 callersFunctionsave_array_as_csv
Save numpy array as CSV file
gear_sonic_deploy/reference/convert_motions.py:141
↓ 6 callersMethodset_goal
Set the command from the planner that the policy should follow. Args: goal: Dictionary containing high-level commands or goals fr
decoupled_wbc/control/base/policy.py:13
↓ 6 callersMethodset_pose
( env: "LocoManipulationEnv", x_range: [tuple[float, float]], y_range: [tuple[float, f
decoupled_wbc/dexmg/gr00trobocasa/robocasa/environments/locomanipulation/base.py:107
↓ 6 callersMethodstart
Main simulation loop
gear_sonic/utils/mujoco_sim/base_sim.py:598
↓ 6 callersFunctiontest
(args: argparse.Namespace = None)
motionbricks/motionbricks/exp_setup/experiment.py:33
↓ 6 callersMethodunnormalize
(self, data: torch.Tensor, index=None)
motionbricks/motionbricks/motionlib/core/utils/stats.py:60
↓ 6 callersMethodupdate_key
Write ``data`` into the buffer at the current step for ``key``. Args: key: Previously registered buffer key. data: Te
gear_sonic/trl/modules/data_utils.py:74
↓ 5 callersMethodBodyAngVelocities
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/motion_data_reader.hpp:77
↓ 5 callersMethodBodyLinVelocities
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/motion_data_reader.hpp:76
↓ 5 callersMethodEnqueue
gear_sonic_deploy/src/TRTInference/InferenceEngine.cpp:761
↓ 5 callersMethodFinalize
Finalize recording - closes files and writes metadata
gear_sonic_deploy/src/g1/g1_deploy_onnx_ref/include/motion_data_reader.hpp:517
↓ 5 callersMethodGetServerApiVersion
(self)
external_dependencies/unitree_sdk2_python/unitree_sdk2py/rpc/client.py:34
↓ 5 callersMethodInit
(self)
external_dependencies/unitree_sdk2_python/example/go2/low_level/go2_stand_example.py:55
← previousnext →301–400 of 6,086, ranked by callers