MCPcopy Create free account

hub / github.com/GuyTevet/CLoSD / functions

Functions2,247 in github.com/GuyTevet/CLoSD

↓ 2 callersFunction_check_longest_curve
(curve, max_curve_key_count)
closd/utils/poselib/poselib/skeleton/backend/fbx/fbx_py27_backend.py:235
↓ 2 callersMethod_combine_rewards
(self, task_rewards, amp_rewards)
closd/learning/amp_agent.py:909
↓ 2 callersMethod_compute_amp_observations
(self, env_ids=None)
closd/env/tasks/humanoid_amp.py:627
↓ 2 callersMethod_compute_angular_velocity
(r, time_delta: float, guassian_filter=True)
closd/utils/poselib/poselib/skeleton/skeleton3d.py:1110
↓ 2 callersMethod_compute_norms
(self, grad_scale=1.0)
closd/diffusion_planner/diffusion/fp16_util.py:216
↓ 2 callersMethod_compute_observations
(self, env_ids=None)
closd/env/tasks/humanoid.py:1115
↓ 2 callersMethod_create_smpl_humanoid_xml
(self, num_humanoids, smpl_robot, queue, pid)
closd/env/tasks/humanoid.py:671
↓ 2 callersMethod_critic_loss
(self, value_preds_batch, values, curr_e_clip, return_batch, clip_value)
closd/learning/common_agent.py:571
↓ 2 callersMethod_draw
(self)
closd/utils/poselib/poselib/visualization/plt_plotter.py:199
↓ 2 callersMethod_draw
(self)
closd/utils/poselib/poselib/visualization/plt_plotter.py:397
↓ 2 callersMethod_eval_disc
(self, amp_obs)
closd/learning/amp_agent.py:916
↓ 2 callersMethod_fetch_amp_obs_demo
(self, num_samples)
closd/learning/amp_agent.py:869
↓ 2 callersMethod_get_lines_and_dots
Get all the lines and dots needed to draw the skeleton state
closd/utils/poselib/poselib/visualization/skeleton_plotter_tasks.py:47
↓ 2 callersMethod_get_num_bodies
(self)
closd/utils/motion_lib_base.py:555
↓ 2 callersMethod_get_pairwise_average_translation
(self)
closd/utils/poselib/poselib/skeleton/skeleton3d.py:643
↓ 2 callersMethod_get_state_from_gen_cache
(self, motion_ids)
closd/env/tasks/closd.py:651
↓ 2 callersMethod_get_vel_and_avel
Get all the velocity and angular velocity lines
closd/utils/poselib/poselib/visualization/skeleton_plotter_tasks.py:160
↓ 2 callersFunction_index_from_letter
(letter: str)
closd/utils/pytorch3d_transforms.py:358
↓ 2 callersFunction_index_from_letter
(letter: str)
closd/utils/smpllib/utils/rotation_conversions.py:235
↓ 2 callersFunction_index_from_letter
(letter: str)
closd/diffusion_planner/utils/rotation_conversions.py:209
↓ 2 callersMethod_init_lim
(self)
closd/utils/poselib/poselib/visualization/plt_plotter.py:75
↓ 2 callersMethod_init_lim
(self)
closd/utils/poselib/poselib/visualization/plt_plotter.py:259
↓ 2 callersMethod_lines_extract_xy_impl
(index, lines_task)
closd/utils/poselib/poselib/visualization/plt_plotter.py:99
↓ 2 callersMethod_lines_extract_xyz_impl
(index, lines_task)
closd/utils/poselib/poselib/visualization/plt_plotter.py:290
↓ 2 callersMethod_max
(self, x, y)
closd/utils/poselib/poselib/visualization/plt_plotter.py:68
↓ 2 callersMethod_min
(self, x, y)
closd/utils/poselib/poselib/visualization/plt_plotter.py:61
↓ 2 callersMethod_read_pixels_as_in_window
(self)
closd/utils/smpllib/khrylib/rl/envs/common/mjviewer.py:216
↓ 2 callersMethod_refresh_sim_tensors
(self)
closd/env/tasks/humanoid.py:1104
↓ 2 callersMethod_remapped_to
(self, joint_mapping: Dict[str, str], target_skeleton_tree: SkeletonTree)
closd/utils/poselib/poselib/skeleton/skeleton3d.py:696
↓ 2 callersMethod_reset_default
(self, env_ids)
closd/env/tasks/humanoid_amp.py:350
↓ 2 callersMethod_reset_envs
(self, env_ids)
closd/env/tasks/humanoid.py:552
↓ 2 callersMethod_reset_pose_buffer
(self, env_ids=None)
closd/env/tasks/closd.py:407
↓ 2 callersMethod_reset_ref_state_init
(self, env_ids)
closd/env/tasks/humanoid_amp.py:449
↓ 2 callersMethod_reset_sample_idx
(self)
closd/learning/replay_buffer.py:71
↓ 2 callersMethod_sample_time
(self, motion_ids)
closd/env/tasks/humanoid_im.py:628
↓ 2 callersMethod_sample_time
(self, motion_ids)
closd/env/tasks/humanoid_amp.py:357
↓ 2 callersMethod_shuffle_idx_buf
(self)
closd/learning/amp_datasets.py:96
↓ 2 callersMethod_trail_extract_xy_impl
(index, trail_task)
closd/utils/poselib/poselib/visualization/plt_plotter.py:103
↓ 2 callersMethod_trail_extract_xyz_impl
(index, trail_task)
closd/utils/poselib/poselib/visualization/plt_plotter.py:294
↓ 2 callersMethod_truncate
(self, s)
closd/diffusion_planner/diffusion/logger.py:80
↓ 2 callersMethod_update_marker
(self)
closd/env/tasks/closd.py:555
↓ 2 callersMethod_zero_out_far
(self, _ref_rb_pos_subset, _ref_body_vel_subset, _root_pos, _body_pos_subset, _body_vel_subset)
closd/env/tasks/closd.py:564
↓ 2 callersFunctionadd_data_options
(parser)
closd/diffusion_planner/utils/parser_util.py:159
↓ 2 callersFunctionadd_diffusion_options
(parser)
closd/diffusion_planner/utils/parser_util.py:87
↓ 2 callersFunctionadd_model_options
(parser)
closd/diffusion_planner/utils/parser_util.py:96
↓ 2 callersFunctionadd_sampling_options
(parser)
closd/diffusion_planner/utils/parser_util.py:217
↓ 2 callersMethodafter_init
(self, algo)
closd/run.py:155
↓ 2 callersMethodafter_print_stats
(self, frame, epoch_num, total_time)
closd/run.py:175
↓ 2 callersFunctionagt_color
(aidx)
closd/utils/draw_utils.py:17
↓ 2 callersFunctionapply_rules
(args)
closd/diffusion_planner/utils/parser_util.py:48
↓ 2 callersFunctionapprox_standard_normal_cdf
A fast approximation of the cumulative distribution function of the standard normal.
closd/diffusion_planner/diffusion/losses.py:42
↓ 2 callersFunctionarcball_map_to_sphere
Return unit sphere coordinates from window coordinates.
closd/utils/smpllib/utils/transformation.py:1751
↓ 2 callersFunctionarcball_map_to_sphere
Return unit sphere coordinates from window coordinates.
closd/utils/smpllib/khrylib/utils/transformation.py:1738
↓ 2 callersMethodbackward
(self)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:966
↓ 2 callersMethodbackward_G
(self)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:452
↓ 2 callersMethodbatch_forward
(self, x, reverse=False)
closd/utils/smpllib/khrylib/models/rnn.py:45
↓ 2 callersMethodbound_loss
(self, mu)
closd/learning/common_agent.py:507
↓ 2 callersFunctioncalc_heading
(q)
closd/utils/torch_utils.py:201
↓ 2 callersFunctioncalculate_activation_statistics
Params: -- activation: num_samples x dim_feat Returns: -- mu: dim_feat -- sigma: dim_feat x dim_feat
closd/diffusion_planner/data_loaders/humanml/utils/metrics.py:61
↓ 2 callersFunctioncalculate_top_k
(mat, top_k)
closd/diffusion_planner/data_loaders/humanml/utils/metrics.py:23
↓ 2 callersMethodchange_char_color
(self)
closd/env/tasks/humanoid.py:521
↓ 2 callersMethodclip_policy_grad
(self)
closd/utils/smpllib/khrylib/rl/agents/agent_ppo.py:53
↓ 2 callersMethodclose
(self)
closd/utils/smpllib/khrylib/rl/envs/common/mujoco_env.py:139
↓ 2 callersFunctioncollate_tensors
(batch)
closd/diffusion_planner/data_loaders/tensors.py:9
↓ 2 callersFunctioncompute_error_accel
Computes acceleration error: 1/(n-2) \sum_{i=1}^{n-1} X_{i-1} - 2X_i + X_{i+1} Note that for each frame that is not visible, three en
closd/utils/smpllib/smpllib/smpl_eval.py:251
↓ 2 callersFunctioncompute_error_vel
(joints_gt, joints_pred, vis=None)
closd/utils/smpllib/smpllib/smpl_eval.py:288
↓ 2 callersFunctioncompute_flat_grad
(output, inputs, filter_input_ids=set(), retain_graph=False, create_graph=False)
closd/utils/smpllib/khrylib/utils/torch.py:126
↓ 2 callersFunctioncompute_humanoid_im_reset
(reset_buf, progress_buf, contact_buf, contact_body_ids, rigid_body_pos, ref_body_pos, pass_time, enable_early
closd/env/tasks/humanoid_im.py:1512
↓ 2 callersFunctioncompute_imitation_observations_v6
(root_pos, root_rot, body_pos, body_rot, body_vel, body_ang_vel, ref_body_pos, ref_body_rot, ref_body_vel, ref
closd/env/tasks/humanoid_im.py:1240
↓ 2 callersFunctioncompute_rotation_matrix_from_ortho6d
(ortho6d)
closd/utils/smpllib/utils/torch_geometry_transforms.py:554
↓ 2 callersFunctioncompute_rotation_matrix_from_ortho6d
(ortho6d)
closd/utils/smpllib/utils/transform_utils.py:23
↓ 2 callersMethodcondition_score
Compute what the p_mean_variance output would have been, should the model's score function be conditioned by cond_fn. See co
closd/diffusion_planner/diffusion/gaussian_diffusion.py:504
↓ 2 callersMethodcondition_score_with_grad
Compute what the p_mean_variance output would have been, should the model's score function be conditioned by cond_fn. See co
closd/diffusion_planner/diffusion/gaussian_diffusion.py:528
↓ 2 callersFunctionconfigure
If comm is provided, average all numerical stats across that comm
closd/diffusion_planner/diffusion/logger.py:442
↓ 2 callersMethodconstruct_tree
( self, template_fname=TEMPLATE_FILE, offset=np.array([0, 0, 0]),
closd/utils/smpllib/khrylib/mocap/skeleton_mesh_local.py:211
↓ 2 callersFunctioncont6d_to_matrix_np
(cont6d)
closd/diffusion_planner/data_loaders/humanml/common/quaternion.py:341
↓ 2 callersFunctioncreate_collection
(col_name)
closd/blender/blender_utils.py:19
↓ 2 callersFunctioncross_product
( u, v)
closd/utils/smpllib/utils/torch_geometry_transforms.py:574
↓ 2 callersFunctioncross_product
(u, v)
closd/utils/smpllib/utils/transform_utils.py:64
↓ 2 callersMethoddemap_params
(self, params)
closd/utils/smpllib/smpllib/smpl_local_robot.py:2331
↓ 2 callersFunctiondof_to_obs_smpl
(pose)
closd/env/tasks/humanoid.py:1419
↓ 2 callersMethoddumpkvs
(self)
closd/diffusion_planner/diffusion/logger.py:355
↓ 2 callersFunctionedit_args
()
closd/diffusion_planner/utils/parser_util.py:342
↓ 2 callersMethodenv_reset
(self, env_ids=None)
closd/learning/common_agent.py:979
↓ 2 callersMethodenv_reset
(self, env_ids=None)
closd/learning/common_player.py:191
↓ 2 callersMethodenv_reset
(self, env_ids=None)
closd/learning/common_player.py:384
↓ 2 callersFunctioneuclidean_distance_matrix
Params: -- matrix1: N1 x D -- matrix2: N2 x D Returns: -- dist: N1 x N2 dist[i, j] == distance(matrix
closd/diffusion_planner/data_loaders/humanml/utils/metrics.py:7
↓ 2 callersFunctioneuclidean_to_rotation_matrix
Get the rotation matrix on the top-left corner of a Euclidean transformation matrix
closd/utils/poselib/poselib/core/rotation3d.py:431
↓ 2 callersFunctioneuclidean_to_transform
Construct a transform from a Euclidean transformation matrix
closd/utils/poselib/poselib/core/rotation3d.py:466
↓ 2 callersFunctioneuclidean_translation
Get the translation vector located at the last column of the matrix
closd/utils/poselib/poselib/core/rotation3d.py:446
↓ 2 callersFunctioneuler_matrix
Return homogeneous rotation matrix from Euler angles and axis sequence. ai, aj, ak : Euler's roll, pitch and yaw angles axes : One of 24 axis
closd/utils/smpllib/khrylib/utils/transformation.py:1078
↓ 2 callersMethodevaluate
(self)
closd/diffusion_planner/train/training_loop.py:269
↓ 2 callersMethodfind_resume_checkpoint
look for all file in save directory in the pattent of model{number}.pt and return the one with the highest step number. TODO: Imp
closd/diffusion_planner/train/training_loop.py:413
↓ 2 callersFunctionfoot_detect
(positions, thres)
closd/diffusion_planner/data_loaders/humanml/scripts/motion_process.py:47
↓ 2 callersMethodforward
(self, x)
closd/utils/smpllib/khrylib/rl/core/critic.py:15
↓ 2 callersMethodforward
(self, x)
closd/utils/smpllib/khrylib/models/mlp.py:24
↓ 2 callersMethodforward
(self, batch_data)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:54
↓ 2 callersMethodforward
(self, batch_data)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:941
↓ 2 callersMethodforward_bvh
(self, bone)
closd/utils/smpllib/khrylib/mocap/skeleton_bvh.py:197
↓ 2 callersMethodforward_kinematics_cont6d
(self, cont6d_params, root_pos, skel_joints=None, do_root_R=True)
closd/diffusion_planner/data_loaders/humanml/common/skeleton.py:176
← previousnext →301–400 of 2,247, ranked by callers