MCPcopy Create free account

hub / github.com/GuyTevet/motion-diffusion-model / types & classes

Types & classes98 in github.com/GuyTevet/motion-diffusion-model

↓ 10 callersClassst_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 callersClassClassifierFreeSampleModel
utils/sampler_util.py:10
↓ 6 callersClassSkeleton
data_loaders/humanml/common/skeleton.py:4
↓ 6 callersClassst_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 callersClassLogger
data_loaders/humanml/networks/trainers.py:16
↓ 3 callersClassMovementConvEncoder
data_loaders/humanml/networks/modules.py:79
↓ 3 callersClassRotation2xyz
model/rotation2xyz.py:11
↓ 3 callersClassWordVectorizer
data_loaders/humanml/utils/word_vectorizer.py:46
↓ 3 callersClassjoints2smpl
visualize/simplify_loc2rot.py:13
↓ 2 callersClassAutoRegressiveSampler
utils/sampler_util.py:41
↓ 2 callersClassConvTemporalGraphical
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 callersClassEvaluatorMDMWrapper
data_loaders/humanml/networks/evaluator_wrapper.py:121
↓ 2 callersClassHumanOutputFormat
diffusion/logger.py:36
↓ 2 callersClassMMGeneratedDataset
data_loaders/humanml/motion_loaders/model_motion_loaders.py:14
↓ 2 callersClassMaxMixturePrior
visualize/joints2smpl/src/prior.py:99
↓ 2 callersClassModelOutput
sample/predict.py:26
↓ 2 callersClassMotionEncoderBiGRUCo
data_loaders/humanml/networks/modules.py:353
↓ 2 callersClassPositionalEncoding
data_loaders/humanml/networks/modules.py:62
↓ 2 callersClassSMPLify3D
Implementation of SMPLify, use 3D joints.
visualize/joints2smpl/src/smplify.py:44
↓ 2 callersClassSTGCN
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 callersClassSTGCN
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 callersClassText2MotionDatasetV2
data_loaders/humanml/data/dataset.py:207
↓ 2 callersClassTextEncoderBiGRUCo
data_loaders/humanml/networks/modules.py:311
↓ 1 callersClassA2MEvaluation
eval/a2m/action2motion/evaluate.py:9
↓ 1 callersClassAttLayer
data_loaders/humanml/networks/modules.py:232
↓ 1 callersClassBERT
model/BERT/BERT_encoder.py:12
↓ 1 callersClassCSVOutputFormat
diffusion/logger.py:113
↓ 1 callersClassCompMDMGeneratedDataset
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:148
↓ 1 callersClassCompTrainerV6
data_loaders/humanml/networks/trainers.py:211
↓ 1 callersClassCompV6GeneratedDataset
data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:52
↓ 1 callersClassContrastiveLoss
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 callersClassEmbedAction
model/mdm.py:389
↓ 1 callersClassEmbedTargetLocMulti
model/mdm.py:449
↓ 1 callersClassEmbedTargetLocSingle
model/mdm.py:399
↓ 1 callersClassEmbedTargetLocSplit
model/mdm.py:422
↓ 1 callersClassGaussianDiffusion
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 callersClassGraph
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 callersClassGraph
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 callersClassGroupNorm32
diffusion/nn.py:18
↓ 1 callersClassInputProcess
model/mdm.py:333
↓ 1 callersClassJSONOutputFormat
diffusion/logger.py:98
↓ 1 callersClassL2Prior
visualize/joints2smpl/src/prior.py:91
↓ 1 callersClassLogger
diffusion/logger.py:332
↓ 1 callersClassLossSecondMomentResampler
diffusion/resample.py:124
↓ 1 callersClassMDM
model/mdm.py:11
↓ 1 callersClassMixedPrecisionTrainer
diffusion/fp16_util.py:148
↓ 1 callersClassMotionDiscriminator
eval/a2m/action2motion/models.py:6
↓ 1 callersClassMotionDiscriminatorForFID
eval/a2m/action2motion/models.py:44
↓ 1 callersClassMotionLenEstimatorBiGRU
data_loaders/humanml/networks/modules.py:389
↓ 1 callersClassMovementConvDecoder
data_loaders/humanml/networks/modules.py:101
↓ 1 callersClassOutputProcess
model/mdm.py:360
↓ 1 callersClassPositionalEncoding
model/mdm.py:296
↓ 1 callersClassSMPL
Extension of the official SMPL implementation to support more joints
model/smpl.py:64
↓ 1 callersClassSMPLifyAnglePrior
visualize/joints2smpl/src/prior.py:52
↓ 1 callersClassSpacedDiffusion
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 callersClassTensorBoardOutputFormat
Dumps key/value pairs into TensorBoard's numeric format.
diffusion/logger.py:150
↓ 1 callersClassTextDecoder
data_loaders/humanml/networks/modules.py:187
↓ 1 callersClassTextEncoderBiGRU
data_loaders/humanml/networks/modules.py:267
↓ 1 callersClassTextOnlyDataset
data_loaders/humanml/data/dataset.py:673
↓ 1 callersClassTextVAEDecoder
data_loaders/humanml/networks/modules.py:123
↓ 1 callersClassTimestepEmbedder
model/mdm.py:316
↓ 1 callersClassTrainLoop
train/training_loop.py:37
↓ 1 callersClassUESTC
data_loaders/a2m/uestc.py:51
↓ 1 callersClassUniformSampler
diffusion/resample.py:61
↓ 1 callersClassWeightedSum
utils/misc.py:5
↓ 1 callersClass_WrappedModel
diffusion/respace.py:118
ClassCheckpointFunction
diffusion/nn.py:148
ClassClassifierFreeSampleModel
model/cfg_sampler.py:8
ClassClearmlPlatform
train/train_platforms.py:22
ClassDataset
data_loaders/a2m/dataset.py:10
ClassDecompTrainerV3
data_loaders/humanml/networks/trainers.py:25
ClassEvaluation
eval/a2m/stgcn/evaluate.py:10
ClassEvaluatorModelWrapper
data_loaders/humanml/networks/evaluator_wrapper.py:27
ClassHumanAct12Poses
data_loaders/a2m/humanact12poses.py:7
ClassHumanML3D
data_loaders/humanml/data/dataset.py:753
ClassKIT
data_loaders/humanml/data/dataset.py:821
ClassKVWriter
diffusion/logger.py:26
ClassLengthEstTrainer
data_loaders/humanml/networks/trainers.py:748
ClassLossAwareSampler
diffusion/resample.py:70
ClassLossType
diffusion/gaussian_diffusion.py:93
ClassModelMeanType
Which type of output the model predicts.
diffusion/gaussian_diffusion.py:69
ClassModelVarType
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
ClassMotionDatasetV2
data_loaders/humanml/data/dataset.py:535
ClassNewDataloader
eval/a2m/stgcn_eval.py:34
ClassNewDataloader
eval/a2m/gru_eval.py:19
ClassNoPlatform
train/train_platforms.py:56
ClassPredictor
sample/predict.py:53
ClassRawTextDataset
data_loaders/humanml/data/dataset.py:608
ClassScheduleSampler
A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unbia
diffusion/resample.py:23
ClassSeqWriter
diffusion/logger.py:31
ClassSiLU
diffusion/nn.py:13
ClassTensorboardPlatform
train/train_platforms.py:44
ClassText2MotionDataset
data_loaders/humanml/data/dataset.py:23
ClassText2MotionDatasetBaseline
data_loaders/humanml/data/dataset.py:381
ClassTextMotionMatchTrainer
data_loaders/humanml/networks/trainers.py:879
ClassTrainPlatform
train/train_platforms.py:4
ClassWandBPlatform
train/train_platforms.py:60
Classnpy2obj
visualize/vis_utils.py:8