MCPcopy Create free account

hub / github.com/TeleHuman/TextOp / functions

Functions1,187 in github.com/TeleHuman/TextOp

⨍ Functions 1,187◇ Types & classes 186
↓ 1 callersMethodensure_sequence_capacity
Ensure motion sequence has enough capacity
TextOpDeploy/src/textop_ctrl/scripts/motion_watcher.py:187
↓ 1 callersFunctioneuler_to_rot
(roll, pitch, yaw)
TextOpDeploy/src/unitree_mujoco/terrain_tool/terrain_generator.py:32
↓ 1 callersMethodeval_mode
(self)
deps/rsl_rl-modular-normed/rsl_rl/runners/on_policy_runner.py:479
↓ 1 callersMethodexport
(self, path, filename)
TextOpTracker/source/textop_tracker/textop_tracker/utils/exporter.py:59
↓ 1 callersFunctionfoot_detect
(positions, thres=0.002)
dataset/process_retarget_data.py:16
↓ 1 callersMethodforward
(self, x)
TextOpRobotMDAR/robotmdar/diffusion/nn.py:14
↓ 1 callersMethodforward_kinematics_batch
Perform forward kinematics using the given trajectory and local rotations
TextOpRobotMDAR/robotmdar/skeleton/forward_kinematics.py:285
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
TextOpRobotMDAR/robotmdar/model/operator/cross_attention.py:259
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
TextOpRobotMDAR/robotmdar/model/operator/cross_attention.py:323
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
TextOpRobotMDAR/robotmdar/model/operator/cross_attention.py:274
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
TextOpRobotMDAR/robotmdar/model/operator/cross_attention.py:347
↓ 1 callersMethodgenerator
(self)
deps/rsl_rl-modular-normed/rsl_rl/storage/rollout_storage.py:170
↓ 1 callersFunctionget_action_scale
(joint_name)
TextOpTracker/scripts/deploy_mujoco.py:154
↓ 1 callersFunctionget_crc
TextOpDeploy/src/unitree_mujoco/example/ros2/src/motor_crc.cpp:4
↓ 1 callersMethodget_dir
(self)
TextOpRobotMDAR/robotmdar/diffusion/logger.py:388
↓ 1 callersMethodget_foot_id
Get foot id
TextOpRobotMDAR/robotmdar/skeleton/forward_kinematics.py:418
↓ 1 callersMethodget_inference_policy
(self, device=None)
deps/rsl_rl-modular-normed/rsl_rl/runners/on_policy_runner.py:457
↓ 1 callersMethodget_intrinsic_reward
(self, rnd_state)
deps/rsl_rl-modular-normed/rsl_rl/modules/rnd.py:112
↓ 1 callersFunctionget_joint_default_pos
(joint_name)
TextOpTracker/scripts/deploy_mujoco.py:180
↓ 1 callersMethodget_motion_data_batch
( self, motion_idx: int, time_steps_start: torch.Tensor, time_steps_end: torch.Tensor)
TextOpTracker/source/textop_tracker/textop_tracker/tasks/tracking/mdp/commands_multi.py:140
↓ 1 callersFunctionget_new_coordinate
(jts)
TextOpRobotMDAR/robotmdar/dtype/motion.py:565
↓ 1 callersMethodget_next_state
Gets the next state of the motion.
TextOpTracker/scripts/pklpack_to_npz.py:194
↓ 1 callersMethodget_next_state
Gets the next state of the motion.
TextOpTracker/scripts/csv_to_npz.py:188
↓ 1 callersFunctionget_rank_without_mpi_import
()
TextOpRobotMDAR/robotmdar/diffusion/logger.py:403
↓ 1 callersFunctionget_text_embedding
Encode text using CLIP model.
TextOpDeploy/src/textop_ctrl/scripts/rmdar.py:45
↓ 1 callersFunctionget_text_embedding
Encode text using CLIP model.
TextOpRobotMDAR/robotmdar/eval/freq_dar.py:32
↓ 1 callersMethodhas_motion_data
New method to check if motion data is available
TextOpDeploy/src/textop_ctrl/include/textop_ctrl/motion_loader.hpp:38
↓ 1 callersMethodhave_overlap
(self, seg1, seg2)
TextOpRobotMDAR/robotmdar/dataloader/data.py:394
↓ 1 callersFunctionhelper
※ This function must be called at the beginning of main() function
TextOpDeploy/src/unitree_mujoco/simulate/src/param.h:57
↓ 1 callersFunctionhook_modnorm
(mlp)
deps/rsl_rl-modular-normed/rsl_rl/modular_norm/mlp.py:47
↓ 1 callersFunctionhydra_main
(cfg: DictConfig)
TextOpRobotMDAR/robotmdar/eval/freq_dar.py:295
↓ 1 callersFunctioninit_cmd_hg
TextOpDeploy/src/textop_ctrl/src/command_helper.cpp:60
↓ 1 callersMethodinit_storage
( self, training_type, num_envs, num_transitions_per_env, actor_obs_shape, critic_obs_shape, actions_s
deps/rsl_rl-modular-normed/rsl_rl/algorithms/ppo.py:118
↓ 1 callersFunctioninterpolate_motion_data_slerp
(root_trans_offset, root_rot, dof, original_fps=30, target_fps=50)
dataset/process_retarget_data.py:49
↓ 1 callersFunctioninterpolate_slerp
(quaternions, target_time)
dataset/process_retarget_data.py:65
↓ 1 callersMethodisAxis
* Returns true if this event is the result of an axis movement. */
TextOpDeploy/src/unitree_mujoco/simulate/src/joystick/joystick.h:74
↓ 1 callersMethodisButton
* Returns true if this event is the result of a button press. */
TextOpDeploy/src/unitree_mujoco/simulate/src/joystick/joystick.h:66
↓ 1 callersFunctionjetson_compatible_torch_compile
(vae, denoiser, dar_cfg)
TextOpDeploy/src/textop_ctrl/scripts/rmdar.py:115
↓ 1 callersFunctionlist_to_csv_str
(arr, *, decimals: int = 3, delimiter: str = ",")
TextOpTracker/source/textop_tracker/textop_tracker/utils/exporter.py:84
↓ 1 callersFunctionload_amass
(amass_dir)
dataset/pack_dataset.py:85
↓ 1 callersFunctionload_babel
(BABEL_DIR)
dataset/pack_dataset.py:76
↓ 1 callersMethodload_dar
(self)
TextOpDeploy/src/textop_ctrl/scripts/rmdar.py:344
↓ 1 callersMethodload_from_yaml
TextOpDeploy/src/unitree_mujoco/simulate/src/param.h:29
↓ 1 callersMethodload_model
(self, ckpt_path: Path)
TextOpRobotMDAR/robotmdar/train/manager.py:662
↓ 1 callersMethodload_model
(self, ckpt_path: Path)
TextOpRobotMDAR/robotmdar/train/manager.py:880
↓ 1 callersFunctionload_mujoco
()
TextOpDeploy/src/textop_ctrl/scripts/show_realrobot.py:28
↓ 1 callersMethodload_mujoco
Initialize MuJoCo visualization
TextOpDeploy/src/textop_ctrl/scripts/motion_watcher.py:45
↓ 1 callersMethodload_npz_data
Load motion data from NPZ file
TextOpDeploy/src/textop_ctrl/scripts/npz_motion_publisher.py:73
↓ 1 callersFunctionload_pkl
(pkl_path)
dataset/motion_package.py:9
↓ 1 callersFunctionlogkv
Log a value of some diagnostic Call this once for each diagnostic quantity, each iteration If called many times, last value will be used.
TextOpRobotMDAR/robotmdar/diffusion/logger.py:212
↓ 1 callersMethodlogkv
(self, key, val)
TextOpRobotMDAR/robotmdar/diffusion/logger.py:347
↓ 1 callersFunctionmain
()
TextOpDeploy/src/textop_ctrl/scripts/make_zero_npz.py:96
↓ 1 callersFunctionmain
Main function to run MotionDAR node with Hydra configuration
TextOpDeploy/src/textop_ctrl/scripts/rmdar.py:604
↓ 1 callersFunctionmain
(args=None)
TextOpDeploy/src/textop_ctrl/scripts/npz_motion_publisher.py:312
↓ 1 callersFunctionmain
(args=None)
TextOpDeploy/src/textop_ctrl/scripts/motion_watcher.py:303
↓ 1 callersFunctionmain
()
dataset/motion_package.py:46
↓ 1 callersFunctionmain
()
dataset/smplx_to_robot_dataset.py:171
↓ 1 callersFunctionmain
()
dataset/pack_dataset.py:158
↓ 1 callersFunctionmain
()
TextOpTracker/scripts/replay_npz.py:103
↓ 1 callersFunctionmain
Main function.
TextOpTracker/scripts/pklpack_to_npz.py:319
↓ 1 callersFunctionmain
Main function.
TextOpTracker/scripts/csv_to_npz.py:320
↓ 1 callersFunctionmain
Main function.
TextOpTracker/scripts/pkl_to_npz.py:57
↓ 1 callersFunctionmain
Play with RSL-RL agent.
TextOpTracker/scripts/rsl_rl/play.py:76
↓ 1 callersFunctionmain
Train with RSL-RL agent.
TextOpTracker/scripts/rsl_rl/train.py:75
↓ 1 callersFunctionmain
Export policy to ONNX format.
TextOpTracker/scripts/rsl_rl/export.py:73
↓ 1 callersFunctionmain
(cfg: DictConfig)
TextOpRobotMDAR/robotmdar/cli.py:24
↓ 1 callersFunctionmain
Main function to run the frequency benchmark.
TextOpRobotMDAR/robotmdar/eval/freq_dar.py:242
↓ 1 callersFunctionmake_output_format
(format, ev_dir, log_suffix="")
TextOpRobotMDAR/robotmdar/diffusion/logger.py:191
↓ 1 callersMethodmask_cond
(self, cond, force_mask=False)
TextOpRobotMDAR/robotmdar/model/mld_denoiser.py:53
↓ 1 callersMethodmask_cond
(self, cond, force_mask=False)
TextOpRobotMDAR/robotmdar/model/mld_denoiser.py:152
↓ 1 callersFunctionmaster_params_to_model_params
Copy the master parameter data back into the model parameters.
TextOpRobotMDAR/robotmdar/diffusion/fp16_util.py:65
↓ 1 callersFunctionmaster_params_to_state_dict
( model, param_groups_and_shapes, master_params, use_fp16 )
TextOpRobotMDAR/robotmdar/diffusion/fp16_util.py:95
↓ 1 callersFunctionmatrix_to_quaternion
w x y z Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3,
deps/isaac_utils/isaac_utils/rotations.py:575
↓ 1 callersFunctionmerge_datasets
(amass_data, babel_data_all, output_dir)
dataset/pack_dataset.py:107
↓ 1 callersFunctionmerge_motion_files
(pkl_paths)
dataset/motion_package.py:28
↓ 1 callersFunctionmodel_grads_to_master_grads
Copy the gradients from the model parameters into the master parameters from make_master_params().
TextOpRobotMDAR/robotmdar/diffusion/fp16_util.py:52
↓ 1 callersFunctionmotion_dict_to_feature_v1
(motion_dict: MotionDict)
TextOpRobotMDAR/robotmdar/dtype/motion.py:138
↓ 1 callersFunctionmotion_dict_to_feature_v2
(motion_dict: MotionDict)
TextOpRobotMDAR/robotmdar/dtype/motion.py:296
↓ 1 callersFunctionmotion_feature_to_dict_v1
(motion_feature: MotionFeatureV1, abs_pose: Optional[AbsolutePose] = None)
TextOpRobotMDAR/robotmdar/dtype/motion.py:181
↓ 1 callersFunctionmotion_feature_to_dict_v2
(motion_feature: MotionFeatureV2, abs_pose: Optional[AbsolutePose] = None)
TextOpRobotMDAR/robotmdar/dtype/motion.py:323
↓ 1 callersFunctionmpi_weighted_mean
Copied from: https://github.com/openai/baselines/blob/ea25b9e8b234e6ee1bca43083f8f3cf974143998/baselines/common/mpi_util.py#L110 Perform a we
TextOpRobotMDAR/robotmdar/diffusion/logger.py:412
↓ 1 callersFunctionmy_quat_rotate
(q, v)
TextOpRobotMDAR/robotmdar/dtype/rotation.py:221
↓ 1 callersFunctionmy_quat_rotate
(q, v)
deps/isaac_utils/isaac_utils/rotations.py:235
↓ 1 callersFunctionnormalize_angle
(x)
TextOpRobotMDAR/robotmdar/dtype/rotation.py:148
↓ 1 callersFunctionnormalize_angle
(x)
deps/isaac_utils/isaac_utils/rotations.py:163
↓ 1 callersMethodp_sample_loop_progressive
Generate samples from the model and yield intermediate samples from each timestep of diffusion. Arguments are the same as p_
TextOpRobotMDAR/robotmdar/diffusion/gaussian_diffusion.py:666
↓ 1 callersFunctionparam_grad_or_zeros
(param)
TextOpRobotMDAR/robotmdar/diffusion/fp16_util.py:141
↓ 1 callersFunctionpd_control
Calculates torques from position commands
TextOpTracker/scripts/deploy_mujoco.py:686
↓ 1 callersFunctionperturb_feature_v3
Add random perturbation to motion features with component-specific strategies. Args: motion_feature: Input motion feature tensor [B,
TextOpRobotMDAR/robotmdar/dtype/motion.py:1261
↓ 1 callersMethodplms_sample
Sample x_{t-1} from the model using Pseudo Linear Multistep. Same usage as p_sample().
TextOpRobotMDAR/robotmdar/diffusion/gaussian_diffusion.py:1148
↓ 1 callersMethodplms_sample_loop_progressive
Use PLMS to sample from the model and yield intermediate samples from each timestep of PLMS. Same usage as p_sample_loop_pro
TextOpRobotMDAR/robotmdar/diffusion/gaussian_diffusion.py:1274
↓ 1 callersMethodpredict
TextOpDeploy/src/textop_ctrl/src/onnx_policy.cpp:107
↓ 1 callersFunctionprocess_babel_json
(babel_json_path)
dataset/pack_dataset.py:15
↓ 1 callersFunctionprocess_directory
(source_dir, target_dir, file_extension='.pkl')
dataset/process_retarget_data.py:112
↓ 1 callersMethodprocess_env_step
(self, rewards, dones, infos)
deps/rsl_rl-modular-normed/rsl_rl/algorithms/ppo.py:152
↓ 1 callersFunctionprocess_pkl_file
(input_path, output_path)
dataset/process_retarget_data.py:80
↓ 1 callersFunctionprofile_kv
(scopename)
TextOpRobotMDAR/robotmdar/diffusion/logger.py:294
↓ 1 callersMethodq_mean_variance
Get the distribution q(x_t | x_0). :param x_start: the [N x C x ...] tensor of noiseless inputs. :param t: the number of dif
TextOpRobotMDAR/robotmdar/diffusion/gaussian_diffusion.py:242
↓ 1 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].
deps/isaac_utils/isaac_utils/rotations.py:107
↓ 1 callersFunctionquat_apply
(a: Tensor, b: Tensor, w_last: bool)
deps/isaac_utils/isaac_utils/rotations.py:22
← previousnext →401–500 of 1,187, ranked by callers