MCPcopy Create free account

hub / github.com/GuyTevet/motion-diffusion-model / functions

Functions683 in github.com/GuyTevet/motion-diffusion-model

Method__init__
(self, mode, model, diffusion, dataiterator, device, cond_mode, dataset, num_samples)
eval/a2m/stgcn_eval.py:35
Method__init__
(self, mode, model, diffusion, dataiterator, device, unconstrained, num_samples: int=-1)
eval/a2m/gru_eval.py:20
Method__init__
(self, in_channels, out_channels, kernel_size,
eval/a2m/recognition/models/stgcn.py:155
Method__init__
(self, layout='openpose', strategy='uniform', kintree_path=
eval/a2m/recognition/models/stgcnutils/graph.py:26
Method__init__
(self, in_channels, out_channels, kernel_size,
eval/a2m/recognition/models/stgcnutils/tgcn.py:34
Method__init__
(self, input_size, hidden_size, hidden_layer, device, output_size=12, use_noise=None)
eval/a2m/action2motion/models.py:7
Method__init__
(self, device)
eval/a2m/action2motion/evaluate.py:10
Method__init__
(self, dataname, parameters, device, seed=None)
eval/a2m/stgcn/evaluate.py:11
Method__init__
(self, npy_path, sample_idx, rep_idx, device=0, cuda=True)
visualize/vis_utils.py:9
Method__init__
(self, num_frames, device_id, cuda=True)
visualize/simplify_loc2rot.py:15
Method__init__
(self, smplxmodel, step_size=1e-2, batch_size=1,
visualize/joints2smpl/src/smplify.py:47
Method__init__
(self, dtype=torch.float32, **kwargs)
visualize/joints2smpl/src/prior.py:53
Method__init__
(self, prior_folder='prior', num_gaussians=6, dtype=DEFAULT_DTYPE, epsilon=1e-16,
visualize/joints2smpl/src/prior.py:101
Method__init__
(self, datapath="dataset/uestc", method_name="vibe", view="all", **kargs)
data_loaders/a2m/uestc.py:54
Method__init__
(self, datapath="dataset/HumanAct12Poses", split="train", **kargs)
data_loaders/a2m/humanact12poses.py:10
Method__init__
(self, num_frames=1, sampling="conseq", sampling_step=1, split="train", pose_rep="rot6d", tra
data_loaders/a2m/dataset.py:11
Method__init__
(self, meta_root, prefix)
data_loaders/humanml/utils/word_vectorizer.py:47
Method__init__
(self, log_dir)
data_loaders/humanml/networks/trainers.py:17
Method__init__
(self, args, movement_enc, movement_dec)
data_loaders/humanml/networks/trainers.py:26
Method__init__
(self, args, text_enc, seq_pri, seq_dec, att_layer, mov_dec, mov_enc=None, seq_post=None)
data_loaders/humanml/networks/trainers.py:213
Method__init__
(self, args, estimator)
data_loaders/humanml/networks/trainers.py:750
Method__init__
(self, args, text_encoder, motion_encoder, movement_encoder)
data_loaders/humanml/networks/trainers.py:881
Method__init__
(self, margin=3.0)
data_loaders/humanml/networks/modules.py:16
Method__init__
(self, d_model, max_len=300)
data_loaders/humanml/networks/modules.py:64
Method__init__
(self, input_size, hidden_size, output_size)
data_loaders/humanml/networks/modules.py:80
Method__init__
(self, input_size, hidden_size, output_size)
data_loaders/humanml/networks/modules.py:102
Method__init__
(self, text_size, input_size, output_size, hidden_size, n_layers)
data_loaders/humanml/networks/modules.py:124
Method__init__
(self, text_size, input_size, output_size, hidden_size, n_layers)
data_loaders/humanml/networks/modules.py:188
Method__init__
(self, word_size, pos_size, hidden_size, device)
data_loaders/humanml/networks/modules.py:268
Method__init__
(self, word_size, pos_size, hidden_size, output_size, device)
data_loaders/humanml/networks/modules.py:312
Method__init__
(self, input_size, hidden_size, output_size, device)
data_loaders/humanml/networks/modules.py:354
Method__init__
(self, word_size, pos_size, hidden_size, output_size)
data_loaders/humanml/networks/modules.py:390
Method__init__
(self, opt)
data_loaders/humanml/networks/evaluator_wrapper.py:29
Method__init__
(self, dataset_name, device)
data_loaders/humanml/networks/evaluator_wrapper.py:123
Method__init__
(self, opt, mean, std, split_file, w_vectorizer)
data_loaders/humanml/data/dataset.py:24
Method__init__
(self, opt, mean, std, split_file, w_vectorizer)
data_loaders/humanml/data/dataset.py:208
Method__init__
(self, opt, mean, std, split_file, w_vectorizer)
data_loaders/humanml/data/dataset.py:382
Method__init__
(self, opt, mean, std, split_file)
data_loaders/humanml/data/dataset.py:536
Method__init__
(self, opt, mean, std, text_file, w_vectorizer)
data_loaders/humanml/data/dataset.py:609
Method__init__
(self, opt, mean, std, split_file)
data_loaders/humanml/data/dataset.py:674
Method__init__
(self, mode, datapath='./dataset/humanml_opt.txt', split="train", **kwargs)
data_loaders/humanml/data/dataset.py:754
Method__init__
(self, mode, datapath='./dataset/kit_opt.txt', split="train", **kwargs)
data_loaders/humanml/data/dataset.py:822
Method__init__
(self, offset, kinematic_tree, device)
data_loaders/humanml/common/skeleton.py:5
Method__init__
(self, opt, dataset, w_vectorizer, mm_num_samples, mm_num_repeats)
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:54
Method__init__
(self, args, model, diffusion, dataloader, mm_num_samples, mm_num_repeats, max_motion_length, num_samples_limi
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:150
Method__init__
(self, opt, motion_dataset, w_vectorizer)
data_loaders/humanml/motion_loaders/model_motion_loaders.py:15
Method__init__
(self, save_dir, *args, **kwargs)
train/train_platforms.py:5
Method__init__
(self, save_dir)
train/train_platforms.py:23
Method__init__
(self, save_dir)
train/train_platforms.py:45
Method__init__
(self, save_dir, config=None, *args, **kwargs)
train/train_platforms.py:62
Method__init__
(self, args, train_platform, model, diffusion, data)
train/training_loop.py:38
Method__iter__
(self)
eval/a2m/stgcn_eval.py:74
Method__iter__
(self)
eval/a2m/gru_eval.py:52
Method__len__
(self)
data_loaders/a2m/dataset.py:228
Method__len__
(self)
data_loaders/humanml/utils/word_vectorizer.py:61
Method__len__
(self)
data_loaders/humanml/data/dataset.py:135
Method__len__
(self)
data_loaders/humanml/data/dataset.py:312
Method__len__
(self)
data_loaders/humanml/data/dataset.py:467
Method__len__
(self)
data_loaders/humanml/data/dataset.py:591
Method__len__
(self)
data_loaders/humanml/data/dataset.py:645
Method__len__
(self)
data_loaders/humanml/data/dataset.py:738
Method__len__
(self)
data_loaders/humanml/data/dataset.py:817
Method__len__
(self)
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:124
Method__len__
(self)
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:260
Method__len__
(self)
data_loaders/humanml/motion_loaders/model_motion_loaders.py:20
Method__str__
(self)
eval/unconstrained/models/stgcnutils/graph.py:42
Method__str__
(self)
eval/a2m/recognition/models/stgcnutils/graph.py:42
Method_apply
(self, fn)
model/mdm.py:286
Function_axis_angle_rotation
Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. A
utils/rotation_conversions.py:123
Function_find_free_port
()
utils/dist_util.py:70
Method_get_videopath
(self, action, view, subject, side)
data_loaders/a2m/uestc.py:211
Method_load_joints3D
(self, ind, frame_ix)
data_loaders/a2m/humanact12poses.py:36
Method_load_rotvec
(self, ind, frame_ix)
data_loaders/a2m/humanact12poses.py:39
Method_scale_timesteps
(self, t)
diffusion/respace.py:113
Functionangle_l2
(angle1, angle2)
utils/loss_util.py:5
Functionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
diffusion/nn.py:43
Functionaxis_angle_to_matrix
Convert rotations given as axis/angle to rotation matrices. Args: axis_angle: Rotations given as a vector in axis angle form,
utils/rotation_conversions.py:418
Methodbackward
(self, loss: th.Tensor)
diffusion/fp16_util.py:176
Methodbert_encode_text
(self, raw_text)
model/mdm.py:180
Functionbody_fitting_loss
Loss function for body fitting
visualize/joints2smpl/src/customloss.py:55
Methodcalc_bpd_loop
Compute the entire variational lower-bound, measured in bits-per-dim, as well as other related quantities. :param model: the
diffusion/gaussian_diffusion.py:1544
Functioncalculate_R_precision
(embedding1, embedding2, top_k, sum_all=False)
data_loaders/humanml/utils/metrics.py:37
Functioncalculate_activation_statistics
Params: -- activation: num_samples x dim_feat Returns: -- mu: dim_feat -- sigma: dim_feat x dim_feat
data_loaders/humanml/utils/metrics.py:60
Functioncalculate_diversity
(activation, diversity_times)
data_loaders/humanml/utils/metrics.py:73
Functioncalculate_frechet_distance
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2
data_loaders/humanml/utils/metrics.py:95
Functioncalculate_matching_score
(embedding1, embedding2, sum_all=False)
data_loaders/humanml/utils/metrics.py:47
Functioncamera_fitting_loss
Loss function for camera optimization.
visualize/joints2smpl/src/customloss.py:91
Functioncheckpoint
Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas
diffusion/nn.py:131
Functioncleanexit
()
utils/misc.py:56
Methodclip_encode_text
(self, raw_text)
model/mdm.py:163
Methodclip_norm
(network_list)
data_loaders/humanml/networks/trainers.py:45
Methodclose
(self)
diffusion/logger.py:93
Methodclose
(self)
diffusion/logger.py:109
Methodclose
(self)
diffusion/logger.py:146
Methodclose
(self)
diffusion/logger.py:185
Methodclose
(self)
train/train_platforms.py:40
Methodclose
(self)
train/train_platforms.py:52
Methodclose
(self)
train/train_platforms.py:87
Methodclosure
()
visualize/joints2smpl/src/smplify.py:168
Functioncollate_fn
(batch)
data_loaders/humanml/data/dataset.py:17
← previousnext →401–500 of 683, ranked by callers