MCPcopy Create free account

hub / github.com/GuyTevet/CLoSD / functions

Functions2,247 in github.com/GuyTevet/CLoSD

↓ 6 callersMethodget_params
(self, param_list, get_name=False, pad_zeros=False)
closd/utils/smpllib/smpllib/smpl_local_robot.py:654
↓ 6 callersMethodget_scoped_name
(self, name)
closd/utils/poselib/poselib/visualization/core.py:31
↓ 6 callersMethodget_self_obs_size
(self)
closd/env/tasks/humanoid.py:477
↓ 6 callersMethodget_state
(self)
closd/env/tasks/vec_task.py:147
↓ 6 callersMethodget_task_obs_size
(self)
closd/env/tasks/humanoid_im.py:452
↓ 6 callersFunctionload_model_wo_clip
(model, state_dict)
closd/diffusion_planner/utils/model_util.py:10
↓ 6 callersMethodload_motions
(self, skeleton_trees, gender_betas, limb_weights, random_sample=True, start_idx=0, max_len=-1)
closd/utils/motion_lib_base.py:176
↓ 6 callersFunctionlog
Write the sequence of args, with no separators, to the console and output files (if you've configured an output file).
closd/diffusion_planner/diffusion/logger.py:247
↓ 6 callersFunctionplot_3d_motion
(save_path, kinematic_tree, joints, title, dataset, figsize=(3, 3), fps=120, radius=3, vis_
closd/diffusion_planner/data_loaders/humanml/utils/plot_script.py:28
↓ 6 callersFunctionquaternion_multiply
Return multiplication of two quaternions. >>> q = quaternion_multiply([4, 1, -2, 3], [8, -5, 6, 7]) >>> numpy.allclose(q, [28, -44, -14, 48])
closd/utils/smpllib/khrylib/utils/transformation.py:1478
↓ 6 callersMethodrender
(self, sync_frame_time=False)
closd/env/tasks/humanoid.py:1330
↓ 6 callersMethodreport_scalar
(self, name, value, iteration, group_name=None)
closd/diffusion_planner/train/train_platforms.py:9
↓ 6 callersMethodreset
(self)
closd/env/tasks/vec_task.py:65
↓ 6 callersFunctionrotation_matrix_to_angle_axis
This function is borrowed from https://github.com/kornia/kornia Convert 3x4 rotation matrix to Rodrigues vector Args: rotation_
closd/utils/smpllib/utils/torch_geometry_transforms.py:219
↓ 6 callersMethodsave
(self)
closd/diffusion_planner/train/training_loop.py:430
↓ 6 callersMethodset_char_color
(self, col, env_ids)
closd/env/tasks/humanoid.py:540
↓ 6 callersFunctionsmpl_to_smplh
(pose)
closd/utils/smpllib/smpllib/smpl_mujoco.py:326
↓ 6 callersMethodstep
(self, env, reward, c_reward, c_info, info)
closd/utils/smpllib/khrylib/rl/core/logger_rl.py:28
↓ 6 callersMethodstep
(opt_list)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:256
↓ 6 callersFunctiontransform_vec
(v, q, trans="root")
closd/utils/smpllib/utils/math_utils.py:119
↓ 5 callersMethod_build_key_body_ids_tensor
(self, key_body_names)
closd/env/tasks/humanoid.py:1337
↓ 5 callersMethod_compute_amp_observations_from_state
(self, root_pos, root_rot, root_vel, root_ang_vel, dof_pos, dof_vel, key_body_pos, key_body_vels, smpl_params,
closd/env/tasks/humanoid_amp.py:664
↓ 5 callersMethod_norm_disc_reward
(self)
closd/learning/amp_agent.py:901
↓ 5 callersMethod_predict_xstart_from_eps
(self, x_t, t, eps)
closd/diffusion_planner/diffusion/gaussian_diffusion.py:446
↓ 5 callersMethod_to_state_vector
(rot, rt)
closd/utils/poselib/poselib/skeleton/skeleton3d.py:547
↓ 5 callersFunctionas_minutes
(s)
closd/diffusion_planner/data_loaders/humanml/utils/utils.py:40
↓ 5 callersMethodeval_actor
(self, obs_dict)
closd/learning/amp_network_builder.py:55
↓ 5 callersMethodforward
(self, input)
closd/utils/smpllib/utils/torch_geometry_transforms.py:459
↓ 5 callersMethodget_log_prob
(self, x, action)
closd/utils/smpllib/khrylib/rl/core/policy.py:21
↓ 5 callersMethodget_motion_state
(self, motion_ids, motion_times, offset=None)
closd/utils/motion_lib_base.py:431
↓ 5 callersMethodget_num_enc_amp_obs
(self)
closd/env/tasks/humanoid_amp.py:828
↓ 5 callersFunctionmean_flat
Take the mean over all non-batch dimensions.
closd/diffusion_planner/diffusion/nn.py:87
↓ 5 callersMethodq_posterior_mean_variance
Compute the mean and variance of the diffusion posterior: q(x_{t-1} | x_t, x_0)
closd/diffusion_planner/diffusion/gaussian_diffusion.py:261
↓ 5 callersFunctionquat_identity
Construct 3D identity rotation given shape
closd/utils/poselib/poselib/core/rotation3d.py:112
↓ 5 callersFunctionrecover_root_rot_pos
(data, hml_type=None)
closd/diffusion_planner/data_loaders/humanml/scripts/motion_process.py:366
↓ 5 callersMethodrender
(self)
closd/utils/smpllib/khrylib/rl/utils/visualizer.py:71
↓ 5 callersFunctionrotation_from_quaternion
(quaternion, separate=False)
closd/utils/smpllib/utils/transformation.py:362
↓ 5 callersMethodrun
(self)
closd/learning/common_player.py:28
↓ 5 callersFunctionsafe_acos
pytorch acos nan: https://github.com/pytorch/pytorch/issues/8069
closd/utils/smpllib/utils/torch_utils.py:53
↓ 5 callersMethodset_eval
(self)
closd/learning/amp_agent.py:154
↓ 5 callersMethodset_train
(self)
closd/learning/amp_agent.py:164
↓ 5 callersMethodshow
(self)
closd/utils/poselib/poselib/visualization/plt_plotter.py:58
↓ 5 callersMethodsync_node
(self)
closd/utils/smpllib/smpllib/smpl_local_robot.py:629
↓ 5 callersMethodsync_node
(self)
closd/utils/smpllib/smpllib/smpl_local_robot.py:1571
↓ 5 callersFunctiontransform_vec
(v, q, trans="root")
closd/utils/smpllib/utils/torch_utils.py:319
↓ 5 callersMethodupdate_mdm_conditions
(self, env_ids)
closd/env/tasks/closd_t2m.py:47
↓ 4 callersMethod_build_rnn
(self, name, input, units, layers)
closd/learning/network_builder.py:90
↓ 4 callersMethod_compute_humanoid_obs
(self, env_ids=None)
closd/env/tasks/humanoid.py:1120
↓ 4 callersFunction_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
closd/diffusion_planner/utils/rotation_conversions.py:69
↓ 4 callersMethod_get_state_from_motionlib_cache
(self, motion_ids, motion_times, offset=None)
closd/env/tasks/humanoid_amp.py:413
↓ 4 callersMethod_preproc_amp_obs
(self, amp_obs)
closd/learning/amp_agent.py:904
↓ 4 callersMethod_reset_env_tensors
(self, env_ids)
closd/env/tasks/humanoid.py:567
↓ 4 callersMethod_set_env_state
( self, env_ids, root_pos, root_rot, dof_pos, root_vel,
closd/env/tasks/humanoid_amp.py:560
↓ 4 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
closd/diffusion_planner/utils/rotation_conversions.py:87
↓ 4 callersMethod_wrap_model
(self, model)
closd/diffusion_planner/diffusion/respace.py:106
↓ 4 callersFunctionadd_base_options
(parser)
closd/diffusion_planner/utils/parser_util.py:76
↓ 4 callersMethodbuild
(self, name, **kwargs)
closd/learning/network_builder.py:500
↓ 4 callersMethodckpt_file_name
(self)
closd/diffusion_planner/train/training_loop.py:369
↓ 4 callersFunctioncollate
(batch)
closd/diffusion_planner/data_loaders/tensors.py:22
↓ 4 callersFunctioncompute_metrics_lite
(pred_pos_all, gt_pos_all, root_idx = 0, use_tqdm = True, concatenate = True)
closd/utils/smpllib/smpllib/smpl_eval.py:24
↓ 4 callersMethodeval
(self)
closd/learning/common_agent.py:182
↓ 4 callersMethodfreeze
(self)
closd/utils/running_mean_std.py:45
↓ 4 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
closd/utils/poselib/poselib/core/backend/abstract.py:94
↓ 4 callersMethodfrom_skeleton_state
Construct a skeleton motion from a skeleton state. The velocities are estimated using second order guassian filter along the last axi
closd/utils/poselib/poselib/skeleton/skeleton3d.py:1000
↓ 4 callersMethodget_action_values
(self, obs)
closd/learning/common_agent.py:257
↓ 4 callersMethodget_offsets_joints_batch
(self, joints)
closd/diffusion_planner/data_loaders/humanml/common/skeleton.py:33
↓ 4 callersMethodget_task_obs_size_detail
(self)
closd/env/tasks/humanoid_im.py:488
↓ 4 callersMethodinit
(self)
closd/utils/smpllib/smpllib/smpl_local_robot.py:962
↓ 4 callersMethodinit_rnn
(self)
closd/learning/im_amp.py:100
↓ 4 callersMethodis_rnn
(self)
closd/learning/network_builder.py:435
↓ 4 callersFunctionload_saved_model
(model, model_path, use_avg: bool=False)
closd/diffusion_planner/utils/model_util.py:121
↓ 4 callersFunctionmasked_l2
(a, b, mask, loss_fn=diff_l2, epsilon=1e-8, entries_norm=True)
closd/diffusion_planner/utils/loss_util.py:13
↓ 4 callersMethodobs_to_torch
(self, obs)
closd/learning/common_player.py:148
↓ 4 callersMethodprocess_infos
(self, infos, done_indices)
closd/run.py:161
↓ 4 callersMethodq_sample
Diffuse the dataset for a given number of diffusion steps. In other words, sample from q(x_t | x_0). :param x_start: the in
closd/diffusion_planner/diffusion/gaussian_diffusion.py:239
↓ 4 callersFunctionqfix
Enforce quaternion continuity across the time dimension by selecting the representation (q or -q) with minimal distance (or, equivalently, ma
closd/diffusion_planner/data_loaders/humanml/common/quaternion.py:151
↓ 4 callersFunctionqnormalize
(q)
closd/diffusion_planner/data_loaders/humanml/common/quaternion.py:28
↓ 4 callersFunctionquaternion_about_axis
Return quaternion for rotation about axis. >>> q = quaternion_about_axis(0.123, [1, 0, 0]) >>> numpy.allclose(q, [0.99810947, 0.06146124, 0,
closd/utils/smpllib/utils/transformation.py:347
↓ 4 callersFunctionquaternion_inverse
Return inverse of quaternion. >>> q0 = random_quaternion() >>> q1 = quaternion_inverse(q0) >>> numpy.allclose(quaternion_multiply(q0, q1)
closd/utils/smpllib/khrylib/utils/transformation.py:1509
↓ 4 callersFunctionquaternion_inverse_batch
q: size(Bx4) Output: size(Bx4)
closd/utils/smpllib/utils/torch_utils.py:308
↓ 4 callersFunctionquaternion_matrix
(_q)
closd/utils/smpllib/utils/torch_utils.py:170
↓ 4 callersFunctionquaternion_to_cont6d
(quaternions)
closd/diffusion_planner/data_loaders/humanml/common/quaternion.py:316
↓ 4 callersMethodrebuild
(self)
closd/utils/smpllib/smpllib/smpl_local_robot.py:1002
↓ 4 callersMethodreset
(self)
closd/learning/replay_buffer.py:15
↓ 4 callersFunctionrotation_matrix_to_quaternion
Convert 3x4 rotation matrix to 4d quaternion vector This algorithm is based on algorithm described in https://github.com/KieranWynn/pyquatern
closd/utils/smpllib/utils/torch_geometry_transforms.py:252
↓ 4 callersMethodsample_motions
(self, n)
closd/utils/motion_lib_base.py:392
↓ 4 callersMethodsave
(self, model_dir, epoch, niter)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:766
↓ 4 callersMethodsave
(self, model_dir, epoch, niter)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:903
↓ 4 callersFunctionset_flat_params_to
(model, flat_params)
closd/utils/smpllib/khrylib/utils/torch.py:102
↓ 4 callersMethodset_params
(self, params, pad_zeros=False)
closd/utils/smpllib/smpllib/smpl_local_robot.py:770
↓ 4 callersMethodshow
(self)
closd/utils/poselib/poselib/visualization/plt_plotter.py:242
↓ 4 callersMethodstd
(self)
closd/utils/smpllib/khrylib/utils/zfilter.py:37
↓ 4 callersFunctiontransform_vec
(v, q, trans="root")
closd/utils/smpllib/khrylib/utils/math.py:111
↓ 4 callersMethodupdate
Update the plotter for any updates in the task primitives
closd/utils/poselib/poselib/visualization/core.py:70
↓ 4 callersMethodzero_grad
(opt_list)
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:246
↓ 3 callersMethod__init__
(self)
closd/utils/smpllib/utils/torch_geometry_transforms.py:456
↓ 3 callersMethod_action_to_pd_targets
(self, action)
closd/env/tasks/humanoid.py:1374
↓ 3 callersMethod_calc_amp_rewards
(self, amp_obs)
closd/learning/amp_agent.py:920
↓ 3 callersMethod_compute_velocity
(p, time_delta, guassian_filter=True)
closd/utils/poselib/poselib/skeleton/skeleton3d.py:1100
↓ 3 callersMethod_eval_critic
(self, obs_dict)
closd/learning/common_agent.py:547
← previousnext →101–200 of 2,247, ranked by callers