MCPcopy Create free account

hub / github.com/MotrixLab/FineMoGen / functions

Functions416 in github.com/MotrixLab/FineMoGen

⨍ Functions 416◇ Types & classes 87
Method__getitem__
(self, idx: int)
mogen/datasets/dataset_wrappers.py:38
Method__getitem__
(self, item)
mogen/models/utils/word_vectorizer.py:72
Method__init__
(self, grad_clip=None, coalesce=True, bucket_size_mb=-1)
mogen/utils/dist_utils.py:46
Method__init__
(self, module, device_ids, dim=0, broadcas
mogen/core/distributed_wrapper.py:47
Method__init__
(self, dataloader, start=None, interval=1,
mogen/core/evaluation/eval_hooks.py:76
Method__init__
(self, data_len=0, evaluator_model=None, num_samples=300,
mogen/core/evaluation/evaluators/diversity_evaluator.py:9
Method__init__
(self, data_len=0, evaluator_model=None, top_k=3,
mogen/core/evaluation/evaluators/precision_evaluator.py:10
Method__init__
(self, data_len=0, evaluator_model=None, batch_size=None,
mogen/core/evaluation/evaluators/fid_evaluator.py:9
Method__init__
(self, data_len=0, evaluator_model=None, top_k=3,
mogen/core/evaluation/evaluators/matching_score_evaluator.py:9
Method__init__
(self, data_len=0, evaluator_model=None, num_samples=100,
mogen/core/evaluation/evaluators/multimodality_evaluator.py:10
Method__init__
(self, batch_size=None, drop_last=False, replication_times=
mogen/core/evaluation/evaluators/base_evaluator.py:9
Method__init__
(self, data_prefix: str, pipeline: list, dataset_name: Opti
mogen/datasets/text_motion_dataset.py:21
Method__init__
(self, data_prefix: str, pipeline: list, dataset_name: Opti
mogen/datasets/base_dataset.py:33
Method__init__
(self, datasets: list)
mogen/datasets/dataset_wrappers.py:16
Method__init__
(self, dataset: Dataset, times: int)
mogen/datasets/dataset_wrappers.py:32
Method__init__
(self, dataset, num_replicas=None, rank=None,
mogen/datasets/samplers/distributed_sampler.py:7
Method__init__
(self, prob=0.5)
mogen/datasets/pipelines/siamese_motion.py:42
Method__init__
(self, feet_threshold, prev_frames, n_joints, prob)
mogen/datasets/pipelines/siamese_motion.py:66
Method__init__
(self, crop_size: Optional[Union[int, None]] = None)
mogen/datasets/pipelines/transforms.py:18
Method__init__
(self, min_size: Optional[Union[int, None]] = None, max_size: Optional[Union
mogen/datasets/pipelines/transforms.py:63
Method__init__
(self, mean_path, std_path, eps=1e-9, keys=['motion'])
mogen/datasets/pipelines/transforms.py:113
Method__init__
(self, transforms)
mogen/datasets/pipelines/compose.py:17
Method__init__
(self, keys)
mogen/datasets/pipelines/formatting.py:37
Method__init__
(self, keys, order)
mogen/datasets/pipelines/formatting.py:52
Method__init__
(self, keys, meta_keys=('filename', 'ori_filename', 'ori_shape',
mogen/datasets/pipelines/formatting.py:85
Method__init__
(self, latent_dim, time_embed_dim, dropout)
mogen/models/utils/stylization_block.py:16
Method__init__
(self, diffusion)
mogen/models/utils/gaussian_diffusion.py:66
Method__init__
(self, diffusion, history_per_term=10, uniform_prob=0.001)
mogen/models/utils/gaussian_diffusion.py:134
Method__init__
( self, *, betas, model_mean_type, model_var_type, loss_type,
mogen/models/utils/gaussian_diffusion.py:336
Method__init__
(self, use_timesteps, **kwargs)
mogen/models/utils/gaussian_diffusion.py:1249
Method__init__
(self, meta_root, prefix)
mogen/models/utils/word_vectorizer.py:53
Method__init__
(self, d_model, dropout=0.1, max_len=5000)
mogen/models/utils/position_encoding.py:10
Method__init__
(self, d_model, dropout=0.1, max_len=5000)
mogen/models/transformers/mdm.py:189
Method__init__
(self, latent_dim, sequence_pos_encoder)
mogen/models/transformers/mdm.py:212
Method__init__
(self, sa_block_cfg=None, ca_block_cfg=None, ffn_cfg=None)
mogen/models/transformers/diffusion_transformer.py:33
Method__init__
(self, input_feats, max_seq_len=240, latent_dim=512,
mogen/models/transformers/diffusion_transformer.py:53
Method__init__
(self, d_model, dropout=0.0, max_len=5000)
mogen/models/transformers/intergen.py:16
Method__init__
(self, input_dim, latent_dim, ff_size, num_layers, num_heads, dropout, activation)
mogen/models/transformers/intergen.py:39
Method__init__
(self, max_seq_len=16, njoints=None, nfeats=None,
mogen/models/transformers/actor.py:131
Method__init__
(self, dataset_name="human_ml3d", latent_dim=64, input_dim=
mogen/models/transformers/finemogen.py:63
Method__init__
(self, dataset_name="human_ml3d", latent_dim=64, output_dim
mogen/models/transformers/finemogen.py:120
Method__init__
(self, ca_block_cfg=None, ffn_cfg=None)
mogen/models/transformers/finemogen.py:213
Method__init__
(self, scale_func_cfg=None, pose_encoder_cfg=None, pose_dec
mogen/models/transformers/finemogen.py:230
Method__init__
(self, sa_block_cfg=None, ca_block_cfg=None, ffn_cfg=None)
mogen/models/transformers/remodiffuse.py:32
Method__init__
(self, num_retrieval=None, topk=None, retrieval_file=None,
mogen/models/transformers/remodiffuse.py:48
Method__init__
(self, retrieval_cfg=None, scale_func_cfg=None, **kwargs)
mogen/models/transformers/remodiffuse.py:205
Method__init__
(self, **kwargs)
mogen/models/transformers/motiondiffuse.py:11
Method__init__
(self, ca_block_cfg=None, ffn_cfg=None)
mogen/models/transformers/momatmogen.py:36
Method__init__
(self, model=None, loss_recon=None, loss_reduction="frame",
mogen/models/architectures/diffusion_architecture.py:58
Method__init__
(self, encoder=None, decoder=None, loss_recon=None,
mogen/models/architectures/vae_architecture.py:57
Method__init__
(self, init_cfg=None)
mogen/models/architectures/base_architecture.py:17
Method__init__
(self, latent_dim, text_latent_dim, num_heads, dropout, time_embed_dim)
mogen/models/attentions/efficient_attention.py:52
Method__init__
(self, latent_dim, text_latent_dim, num_heads, dropout, time_embed_dim)
mogen/models/attentions/efficient_attention.py:98
Method__init__
(self, latent_dim, text_latent_dim, num_heads, num_text_heads, num_experts, topk, gate_type,
mogen/models/attentions/fine_attention.py:65
Method__init__
(self, latent_dim, text_latent_dim, num_heads, dropout, time_embed_dim)
mogen/models/attentions/semantics_modulated.py:91
Method__init__
(self, latent_dim, text_latent_dim, num_heads, dropout, time_embed_dim)
mogen/models/attentions/base_attention.py:12
Method__init__
(self, latent_dim, text_latent_dim, num_heads, dropout, time_embed_dim)
mogen/models/attentions/base_attention.py:103
Method__init__
(self, d_model, max_len=300)
mogen/models/rnns/t2m_bigru.py:53
Method__init__
(self, input_size, movement_hidden_size, movement_latent_size, motion_hidden_size, motion_lat
mogen/models/rnns/t2m_bigru.py:73
Method__init__
(self, word_size, pos_size, hidden_size, output_size)
mogen/models/rnns/t2m_bigru.py:163
Method__init__
(self, input_size, hidden_size, output_size)
mogen/models/rnns/t2m_bigru.py:210
Method__init__
(self, input_size, hidden_size, output_size)
mogen/models/rnns/t2m_bigru.py:233
Method__init__
(self, motion_encoder=None, text_encoder=None, init_cfg=None)
mogen/models/rnns/t2m_bigru.py:270
Method__init__
(self, gan_type, real_label_val=1.0, fake_label_val=0.0,
mogen/models/losses/gan_loss.py:20
Method__init__
(self, reduction='mean', loss_weight=1.0)
mogen/models/losses/mse_loss.py:38
Method__iter__
(self)
mogen/datasets/samplers/distributed_sampler.py:21
Method__len__
Return the length of current dataset.
mogen/datasets/base_dataset.py:77
Method__len__
(self)
mogen/datasets/dataset_wrappers.py:41
Method__len__
(self)
mogen/models/utils/word_vectorizer.py:69
Method__repr__
(self)
mogen/datasets/pipelines/siamese_motion.py:54
Method__repr__
(self)
mogen/datasets/pipelines/siamese_motion.py:169
Method__repr__
(self)
mogen/datasets/pipelines/transforms.py:46
Method__repr__
(self)
mogen/datasets/pipelines/transforms.py:98
Method__repr__
(self)
mogen/datasets/pipelines/compose.py:37
Method__repr__
(self)
mogen/datasets/pipelines/formatting.py:45
Method__repr__
(self)
mogen/datasets/pipelines/formatting.py:61
Method__repr__
(self)
mogen/datasets/pipelines/formatting.py:103
Method__repr__
(self)
mogen/datasets/pipelines/formatting.py:134
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
mogen/datasets/pipelines/rotation_conversions.py:121
Function_convert_weights_to_fp32
(m)
mogen/models/transformers/mdm.py:12
Method_wgan_loss
wgan loss. Args: input (Tensor): Input tensor. target (bool): Target label. Returns: Tensor: wgan
mogen/models/losses/gan_loss.py:44
Methodafter_train_iter
(self, runner)
mogen/utils/dist_utils.py:51
Functionallreduce_grads
(params, coalesce=True, bucket_size_mb=-1)
mogen/utils/dist_utils.py:31
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,
mogen/datasets/pipelines/rotation_conversions.py:419
Methodbuild_temporal_blocks
(self, sa_block_cfg, ca_block_cfg, ffn_cfg)
mogen/models/transformers/finemogen.py:244
Methodbuild_temporal_blocks
(self, sa_block_cfg, ca_block_cfg, ffn_cfg)
mogen/models/transformers/momatmogen.py:53
Methodcalc_bpd_loop
Compute the entire variational lower-bound, measured in bits-per-dim, as well as other related quantities. :param model: the
mogen/models/utils/gaussian_diffusion.py:1100
Functioncheck_input_path
Check input folder or file. Args: input_path (str): input folder or file path. allowed_suffix (List[str], optional):
mogen/utils/path_utils.py:190
Methodcompute_loss
(self, batch)
mogen/models/transformers/intergen.py:144
Methodconcat_batch_metrics
(self, batch_metrics)
mogen/core/evaluation/evaluators/matching_score_evaluator.py:49
Methodcondition_mean
(self, cond_fn, *args, **kwargs)
mogen/models/utils/gaussian_diffusion.py:1269
Methodcondition_score
(self, cond_fn, *args, **kwargs)
mogen/models/utils/gaussian_diffusion.py:1273
Functioncont6d_to_matrix_np
(cont6d)
mogen/datasets/pipelines/quaternion.py:365
Functionconvert_to_one_hot
This function converts target class indices to one-hot vectors, given the number of classes. Args: targets (Tensor): The ground truth
mogen/models/losses/utils.py:93
Methodddim_reverse_sample
Sample x_{t+1} from the model using DDIM reverse ODE.
mogen/models/utils/gaussian_diffusion.py:829
Methodencode
(self, motion, motion_mask)
mogen/models/architectures/vae_architecture.py:88
Methodencode_motion
(self, motion, motion_length=None, motion_ma
mogen/models/rnns/t2m_bigru.py:277
Methodencode_text
(self, text, clip_feat, device)
mogen/models/transformers/diffusion_transformer.py:144
Methodencode_text
(self, text, token=None, device=None, **kwargs)
mogen/models/rnns/t2m_bigru.py:284
Functioneuler2quat
Convert Euler angles to quaternions.
mogen/datasets/pipelines/quaternion.py:181
← previousnext →201–300 of 416, ranked by callers