Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/MotrixLab/MotionDiffuse
/ types & classes
Types & classes
40 in github.com/MotrixLab/MotionDiffuse
⨍
Functions
231
◇
Types & classes
40
↓ 6 callers
Class
Skeleton
text2motion/utils/skeleton.py:4
↓ 5 callers
Class
StylizationBlock
text2motion/models/transformer.py:60
↓ 3 callers
Class
DDPMTrainer
text2motion/trainers/ddpm_trainer.py:29
↓ 3 callers
Class
MotionTransformer
text2motion/models/transformer.py:288
↓ 2 callers
Class
FFN
text2motion/models/transformer.py:157
↓ 2 callers
Class
PositionalEncoding
text2motion/datasets/evaluator_models.py:62
↓ 2 callers
Class
WordVectorizer
text2motion/utils/word_vectorizer.py:46
↓ 1 callers
Class
DistributedSampler
text2motion/datasets/dataloader.py:16
↓ 1 callers
Class
EvaluationDataset
text2motion/datasets/evaluator.py:16
↓ 1 callers
Class
EvaluatorModelWrapper
text2motion/datasets/evaluator.py:376
↓ 1 callers
Class
GaussianDiffusion
Utilities for training and sampling diffusion models. Ported directly from here, and then adapted over time to further experimentation.
text2motion/models/gaussian_diffusion.py:312
↓ 1 callers
Class
LinearTemporalCrossAttention
text2motion/models/transformer.py:122
↓ 1 callers
Class
LinearTemporalDiffusionTransformerDecoderLayer
text2motion/models/transformer.py:173
↓ 1 callers
Class
LinearTemporalSelfAttention
text2motion/models/transformer.py:89
↓ 1 callers
Class
LossSecondMomentResampler
text2motion/models/gaussian_diffusion.py:123
↓ 1 callers
Class
MMGeneratedDataset
text2motion/datasets/evaluator.py:299
↓ 1 callers
Class
MotionEncoderBiGRUCo
text2motion/datasets/evaluator_models.py:353
↓ 1 callers
Class
MovementConvEncoder
text2motion/datasets/evaluator_models.py:79
↓ 1 callers
Class
TemporalCrossAttention
text2motion/models/transformer.py:229
↓ 1 callers
Class
TemporalDiffusionTransformerDecoderLayer
text2motion/models/transformer.py:264
↓ 1 callers
Class
TemporalSelfAttention
text2motion/models/transformer.py:196
↓ 1 callers
Class
Text2MotionDataset
Dataset for Text2Motion generation task.
text2motion/datasets/dataset.py:11
↓ 1 callers
Class
Text2MotionDatasetV2
text2motion/datasets/evaluator.py:139
↓ 1 callers
Class
TextEncoderBiGRUCo
text2motion/datasets/evaluator_models.py:311
↓ 1 callers
Class
TrainCompOptions
text2motion/options/train_options.py:4
↓ 1 callers
Class
UniformSampler
text2motion/models/gaussian_diffusion.py:65
Class
AttLayer
text2motion/datasets/evaluator_models.py:232
Class
BaseOptions
text2motion/options/base_options.py:8
Class
ContrastiveLoss
Contrastive loss function. Based on: http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf
text2motion/datasets/evaluator_models.py:11
Class
LossAwareSampler
text2motion/models/gaussian_diffusion.py:74
Class
LossType
text2motion/models/gaussian_diffusion.py:300
Class
ModelMeanType
Which type of output the model predicts.
text2motion/models/gaussian_diffusion.py:276
Class
ModelVarType
What is used as the model's output variance. The LEARNED_RANGE option has been added to allow the model to predict values between FIXED_
text2motion/models/gaussian_diffusion.py:286
Class
MotionLenEstimatorBiGRU
text2motion/datasets/evaluator_models.py:389
Class
MovementConvDecoder
text2motion/datasets/evaluator_models.py:101
Class
ScheduleSampler
A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unbias
text2motion/models/gaussian_diffusion.py:30
Class
TestOptions
text2motion/options/evaluate_options.py:4
Class
TextDecoder
text2motion/datasets/evaluator_models.py:187
Class
TextEncoderBiGRU
text2motion/datasets/evaluator_models.py:267
Class
TextVAEDecoder
text2motion/datasets/evaluator_models.py:123