Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/GuyTevet/motion-diffusion-model
/ types & classes
Types & classes
98 in github.com/GuyTevet/motion-diffusion-model
⨍
Functions
683
◇
Types & classes
98
↓ 10 callers
Class
st_gcn
r"""Applies a spatial temporal graph convolution over an input graph sequence. Args: in_channels (int): Number of channels in the input se
eval/a2m/recognition/models/stgcn.py:134
↓ 6 callers
Class
ClassifierFreeSampleModel
utils/sampler_util.py:10
↓ 6 callers
Class
Skeleton
data_loaders/humanml/common/skeleton.py:4
↓ 6 callers
Class
st_gcn
r"""Applies a spatial temporal graph convolution over an input graph sequence. Args: in_channels (int): Number of channels in the input se
eval/unconstrained/models/stgcn.py:136
↓ 4 callers
Class
Logger
data_loaders/humanml/networks/trainers.py:16
↓ 3 callers
Class
MovementConvEncoder
data_loaders/humanml/networks/modules.py:79
↓ 3 callers
Class
Rotation2xyz
model/rotation2xyz.py:11
↓ 3 callers
Class
WordVectorizer
data_loaders/humanml/utils/word_vectorizer.py:46
↓ 3 callers
Class
joints2smpl
visualize/simplify_loc2rot.py:13
↓ 2 callers
Class
AutoRegressiveSampler
utils/sampler_util.py:41
↓ 2 callers
Class
ConvTemporalGraphical
r"""The basic module for applying a graph convolution. Args: in_channels (int): Number of channels in the input sequence data out_
eval/a2m/recognition/models/stgcnutils/tgcn.py:7
↓ 2 callers
Class
EvaluatorMDMWrapper
data_loaders/humanml/networks/evaluator_wrapper.py:121
↓ 2 callers
Class
HumanOutputFormat
diffusion/logger.py:36
↓ 2 callers
Class
MMGeneratedDataset
data_loaders/humanml/motion_loaders/model_motion_loaders.py:14
↓ 2 callers
Class
MaxMixturePrior
visualize/joints2smpl/src/prior.py:99
↓ 2 callers
Class
ModelOutput
sample/predict.py:26
↓ 2 callers
Class
MotionEncoderBiGRUCo
data_loaders/humanml/networks/modules.py:353
↓ 2 callers
Class
PositionalEncoding
data_loaders/humanml/networks/modules.py:62
↓ 2 callers
Class
SMPLify3D
Implementation of SMPLify, use 3D joints.
visualize/joints2smpl/src/smplify.py:44
↓ 2 callers
Class
STGCN
r"""Spatial temporal graph convolutional networks. Args: in_channels (int): Number of channels in the input data num_class (int):
eval/unconstrained/models/stgcn.py:11
↓ 2 callers
Class
STGCN
r"""Spatial temporal graph convolutional networks. Args: in_channels (int): Number of channels in the input data num_class (int):
eval/a2m/recognition/models/stgcn.py:11
↓ 2 callers
Class
Text2MotionDatasetV2
data_loaders/humanml/data/dataset.py:207
↓ 2 callers
Class
TextEncoderBiGRUCo
data_loaders/humanml/networks/modules.py:311
↓ 1 callers
Class
A2MEvaluation
eval/a2m/action2motion/evaluate.py:9
↓ 1 callers
Class
AttLayer
data_loaders/humanml/networks/modules.py:232
↓ 1 callers
Class
BERT
model/BERT/BERT_encoder.py:12
↓ 1 callers
Class
CSVOutputFormat
diffusion/logger.py:113
↓ 1 callers
Class
CompMDMGeneratedDataset
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:148
↓ 1 callers
Class
CompTrainerV6
data_loaders/humanml/networks/trainers.py:211
↓ 1 callers
Class
CompV6GeneratedDataset
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:52
↓ 1 callers
Class
ContrastiveLoss
Contrastive loss function. Based on: http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf
data_loaders/humanml/networks/modules.py:11
↓ 1 callers
Class
EmbedAction
model/mdm.py:389
↓ 1 callers
Class
EmbedTargetLocMulti
model/mdm.py:449
↓ 1 callers
Class
EmbedTargetLocSingle
model/mdm.py:399
↓ 1 callers
Class
EmbedTargetLocSplit
model/mdm.py:422
↓ 1 callers
Class
GaussianDiffusion
Utilities for training and sampling diffusion models. Ported directly from here, and then adapted over time to further experimentation.
diffusion/gaussian_diffusion.py:105
↓ 1 callers
Class
Graph
The Graph to model the skeletons extracted by the openpose Args: strategy (string): must be one of the follow candidates - unifor
eval/unconstrained/models/stgcnutils/graph.py:7
↓ 1 callers
Class
Graph
The Graph to model the skeletons extracted by the openpose Args: strategy (string): must be one of the follow candidates - unifor
eval/a2m/recognition/models/stgcnutils/graph.py:7
↓ 1 callers
Class
GroupNorm32
diffusion/nn.py:18
↓ 1 callers
Class
InputProcess
model/mdm.py:333
↓ 1 callers
Class
JSONOutputFormat
diffusion/logger.py:98
↓ 1 callers
Class
L2Prior
visualize/joints2smpl/src/prior.py:91
↓ 1 callers
Class
Logger
diffusion/logger.py:332
↓ 1 callers
Class
LossSecondMomentResampler
diffusion/resample.py:124
↓ 1 callers
Class
MDM
model/mdm.py:11
↓ 1 callers
Class
MixedPrecisionTrainer
diffusion/fp16_util.py:148
↓ 1 callers
Class
MotionDiscriminator
eval/a2m/action2motion/models.py:6
↓ 1 callers
Class
MotionDiscriminatorForFID
eval/a2m/action2motion/models.py:44
↓ 1 callers
Class
MotionLenEstimatorBiGRU
data_loaders/humanml/networks/modules.py:389
↓ 1 callers
Class
MovementConvDecoder
data_loaders/humanml/networks/modules.py:101
↓ 1 callers
Class
OutputProcess
model/mdm.py:360
↓ 1 callers
Class
PositionalEncoding
model/mdm.py:296
↓ 1 callers
Class
SMPL
Extension of the official SMPL implementation to support more joints
model/smpl.py:64
↓ 1 callers
Class
SMPLifyAnglePrior
visualize/joints2smpl/src/prior.py:52
↓ 1 callers
Class
SpacedDiffusion
A diffusion process which can skip steps in a base diffusion process. :param use_timesteps: a collection (sequence or set) of timesteps from
diffusion/respace.py:65
↓ 1 callers
Class
TensorBoardOutputFormat
Dumps key/value pairs into TensorBoard's numeric format.
diffusion/logger.py:150
↓ 1 callers
Class
TextDecoder
data_loaders/humanml/networks/modules.py:187
↓ 1 callers
Class
TextEncoderBiGRU
data_loaders/humanml/networks/modules.py:267
↓ 1 callers
Class
TextOnlyDataset
data_loaders/humanml/data/dataset.py:673
↓ 1 callers
Class
TextVAEDecoder
data_loaders/humanml/networks/modules.py:123
↓ 1 callers
Class
TimestepEmbedder
model/mdm.py:316
↓ 1 callers
Class
TrainLoop
train/training_loop.py:37
↓ 1 callers
Class
UESTC
data_loaders/a2m/uestc.py:51
↓ 1 callers
Class
UniformSampler
diffusion/resample.py:61
↓ 1 callers
Class
WeightedSum
utils/misc.py:5
↓ 1 callers
Class
_WrappedModel
diffusion/respace.py:118
Class
CheckpointFunction
diffusion/nn.py:148
Class
ClassifierFreeSampleModel
model/cfg_sampler.py:8
Class
ClearmlPlatform
train/train_platforms.py:22
Class
Dataset
data_loaders/a2m/dataset.py:10
Class
DecompTrainerV3
data_loaders/humanml/networks/trainers.py:25
Class
Evaluation
eval/a2m/stgcn/evaluate.py:10
Class
EvaluatorModelWrapper
data_loaders/humanml/networks/evaluator_wrapper.py:27
Class
HumanAct12Poses
data_loaders/a2m/humanact12poses.py:7
Class
HumanML3D
data_loaders/humanml/data/dataset.py:753
Class
KIT
data_loaders/humanml/data/dataset.py:821
Class
KVWriter
diffusion/logger.py:26
Class
LengthEstTrainer
data_loaders/humanml/networks/trainers.py:748
Class
LossAwareSampler
diffusion/resample.py:70
Class
LossType
diffusion/gaussian_diffusion.py:93
Class
ModelMeanType
Which type of output the model predicts.
diffusion/gaussian_diffusion.py:69
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_
diffusion/gaussian_diffusion.py:79
Class
MotionDatasetV2
data_loaders/humanml/data/dataset.py:535
Class
NewDataloader
eval/a2m/stgcn_eval.py:34
Class
NewDataloader
eval/a2m/gru_eval.py:19
Class
NoPlatform
train/train_platforms.py:56
Class
Predictor
sample/predict.py:53
Class
RawTextDataset
data_loaders/humanml/data/dataset.py:608
Class
ScheduleSampler
A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unbia
diffusion/resample.py:23
Class
SeqWriter
diffusion/logger.py:31
Class
SiLU
diffusion/nn.py:13
Class
TensorboardPlatform
train/train_platforms.py:44
Class
Text2MotionDataset
data_loaders/humanml/data/dataset.py:23
Class
Text2MotionDatasetBaseline
data_loaders/humanml/data/dataset.py:381
Class
TextMotionMatchTrainer
data_loaders/humanml/networks/trainers.py:879
Class
TrainPlatform
train/train_platforms.py:4
Class
WandBPlatform
train/train_platforms.py:60
Class
npy2obj
visualize/vis_utils.py:8