MCPcopy Create free account

hub / github.com/GuyTevet/CLoSD / types & classes

Types & classes226 in github.com/GuyTevet/CLoSD

↓ 25 callersClassGeom
closd/utils/smpllib/smpllib/smpl_local_robot.py:563
↓ 12 callersClassSMPL_Parser
closd/utils/smpllib/smpllib/smpl_parser.py:258
↓ 10 callersClassRunningMeanStd
closd/utils/running_mean_std.py:9
↓ 10 callersClassSkeleton
closd/diffusion_planner/data_loaders/humanml/common/skeleton.py:4
↓ 7 callersClassRNN
closd/utils/smpllib/khrylib/models/rnn.py:5
↓ 6 callersClassBone
closd/utils/smpllib/khrylib/mocap/skeleton_bvh.py:9
↓ 6 callersClassClassifierFreeSampleModel
closd/diffusion_planner/utils/sampler_util.py:10
↓ 6 callersClassMatplotlib3DPlotter
closd/utils/poselib/poselib/visualization/plt_plotter.py:207
↓ 4 callersClassDraw3DLines
closd/utils/poselib/poselib/visualization/simple_plotter_tasks.py:164
↓ 4 callersClassDraw3DSkeletonState
closd/utils/poselib/poselib/visualization/skeleton_plotter_tasks.py:18
↓ 4 callersClassLogger
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:16
↓ 4 callersClassMLP
closd/utils/smpllib/khrylib/models/mlp.py:5
↓ 4 callersClassMjViewer
Extends :class:`.MjViewerBasic` to add video recording, interactive time and interaction controls. The key bindings are as follows: - TAB
closd/utils/smpllib/khrylib/rl/envs/common/mjviewer.py:106
↓ 4 callersClassSMPLH_Parser
closd/utils/smpllib/smpllib/smpl_parser.py:463
↓ 4 callersClassTemporalConvNet
closd/utils/smpllib/khrylib/models/tcn.py:54
↓ 3 callersClassActuator
closd/utils/smpllib/smpllib/smpl_local_robot.py:828
↓ 3 callersClassBody
closd/utils/smpllib/smpllib/smpl_local_robot.py:892
↓ 3 callersClassConvBlock
closd/learning/network_builder.py:512
↓ 3 callersClassJoint
closd/utils/smpllib/smpllib/smpl_local_robot.py:316
↓ 3 callersClassMotionLibSMPL
closd/utils/motion_lib_smpl.py:39
↓ 3 callersClassMovementConvEncoder
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:79
↓ 3 callersClassSMPLX_Parser
closd/utils/smpllib/smpllib/smpl_parser.py:567
↓ 3 callersClassSkeletonTree
A skeleton tree gives a complete description of a rigid skeleton. It describes a tree structure over a list of nodes with their names indicat
closd/utils/poselib/poselib/skeleton/skeleton3d.py:21
↓ 3 callersClassWordVectorizer
closd/diffusion_planner/data_loaders/humanml/utils/word_vectorizer.py:46
↓ 3 callersClassto_test
closd/utils/smpllib/khrylib/utils/torch.py:52
↓ 2 callersClassAutoRegressiveSampler
closd/diffusion_planner/utils/sampler_util.py:38
↓ 2 callersClassBone
closd/utils/smpllib/khrylib/mocap/skeleton_local.py:75
↓ 2 callersClassBone
closd/utils/smpllib/khrylib/mocap/skeleton_mesh_local.py:38
↓ 2 callersClassChomp1d
closd/utils/smpllib/khrylib/models/tcn.py:6
↓ 2 callersClassDraw3DDots
closd/utils/poselib/poselib/visualization/simple_plotter_tasks.py:176
↓ 2 callersClassDraw3DSkeletonMotion
closd/utils/poselib/poselib/visualization/skeleton_plotter_tasks.py:73
↓ 2 callersClassEvaluatorMDMWrapper
closd/diffusion_planner/data_loaders/humanml/networks/evaluator_wrapper.py:123
↓ 2 callersClassHumanOutputFormat
closd/diffusion_planner/diffusion/logger.py:36
↓ 2 callersClassMMGeneratedDataset
closd/diffusion_planner/data_loaders/humanml/motion_loaders/model_motion_loaders.py:14
↓ 2 callersClassMobileNet
closd/utils/smpllib/khrylib/models/mobile_net.py:5
↓ 2 callersClassModelOutput
closd/diffusion_planner/sample/predict.py:26
↓ 2 callersClassMotionEncoderBiGRUCo
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:353
↓ 2 callersClassPositionalEncoding
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:62
↓ 2 callersClassResNet
closd/utils/smpllib/khrylib/models/resnet.py:6
↓ 2 callersClassResidualBlock
closd/learning/network_builder.py:528
↓ 2 callersClassSMPL_Robot
closd/utils/smpllib/smpllib/smpl_local_robot.py:1178
↓ 2 callersClassSkeletonMotion
closd/utils/poselib/poselib/skeleton/skeleton3d.py:910
↓ 2 callersClassText2MotionDatasetV2
closd/diffusion_planner/data_loaders/humanml/data/dataset.py:207
↓ 2 callersClassTextEncoderBiGRUCo
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:311
↓ 1 callersClassAttLayer
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:232
↓ 1 callersClassBERT
closd/diffusion_planner/model/BERT/BERT_encoder.py:12
↓ 1 callersClassBasePlotterTasks
closd/utils/poselib/poselib/visualization/core.py:40
↓ 1 callersClassCSVOutputFormat
closd/diffusion_planner/diffusion/logger.py:113
↓ 1 callersClassCategorical
closd/utils/smpllib/khrylib/rl/core/distributions.py:28
↓ 1 callersClassCompMDMExternalDataset
closd/diffusion_planner/data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:389
↓ 1 callersClassCompMDMGeneratedDataset
closd/diffusion_planner/data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:154
↓ 1 callersClassCompTrainerV6
closd/diffusion_planner/data_loaders/humanml/networks/trainers.py:211
↓ 1 callersClassCompV6GeneratedDataset
closd/diffusion_planner/data_loaders/humanml/motion_loaders/comp_v6_model_dataset.py:58
↓ 1 callersClassContrastiveLoss
Contrastive loss function. Based on: http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:11
↓ 1 callersClassConv2dAuto
closd/learning/network_builder.py:505
↓ 1 callersClassDeviceCache
closd/utils/motion_lib_base.py:70
↓ 1 callersClassDiagGaussian
closd/utils/smpllib/khrylib/rl/core/distributions.py:6
↓ 1 callersClassDiagGaussianActor
torch.distributions implementation of an diagonal Gaussian policy.
closd/learning/network_builder.py:758
↓ 1 callersClassDoubleQCritic
Critic network, employes double Q-learning.
closd/learning/network_builder.py:788
↓ 1 callersClassDraw3DTrail
closd/utils/poselib/poselib/visualization/simple_plotter_tasks.py:188
↓ 1 callersClassERDNet
closd/utils/smpllib/khrylib/models/erd_net.py:7
↓ 1 callersClassEmbedAction
closd/diffusion_planner/model/mdm.py:392
↓ 1 callersClassEmbedTargetLocMulti
closd/diffusion_planner/model/mdm.py:452
↓ 1 callersClassEmbedTargetLocSingle
closd/diffusion_planner/model/mdm.py:402
↓ 1 callersClassEmbedTargetLocSplit
closd/diffusion_planner/model/mdm.py:425
↓ 1 callersClassFakeMDMArgs
closd/env/tasks/closd.py:37
↓ 1 callersClassFlags
closd/utils/flags.py:3
↓ 1 callersClassFlags
closd/utils/smpllib/utils/flags.py:3
↓ 1 callersClassGaussianDiffusion
Utilities for training and sampling diffusion models. Ported directly from here, and then adapted over time to further experimentation.
closd/diffusion_planner/diffusion/gaussian_diffusion.py:105
↓ 1 callersClassGroupNorm32
closd/diffusion_planner/diffusion/nn.py:18
↓ 1 callersClassHumanoidVisEnv
closd/utils/smpllib/khrylib/rl/envs/visual/humanoid_vis.py:7
↓ 1 callersClassImpalaSequential
closd/learning/network_builder.py:561
↓ 1 callersClassInputProcess
closd/diffusion_planner/model/mdm.py:336
↓ 1 callersClassJSONOutputFormat
closd/diffusion_planner/diffusion/logger.py:98
↓ 1 callersClassLogger
closd/diffusion_planner/diffusion/logger.py:332
↓ 1 callersClassLossSecondMomentResampler
closd/diffusion_planner/diffusion/resample.py:124
↓ 1 callersClassMDM
closd/diffusion_planner/model/mdm.py:13
↓ 1 callersClassMemory
closd/utils/smpllib/khrylib/utils/memory.py:4
↓ 1 callersClassMixedPrecisionTrainer
closd/diffusion_planner/diffusion/fp16_util.py:148
↓ 1 callersClassMotionLenEstimatorBiGRU
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:389
↓ 1 callersClassMovementConvDecoder
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:101
↓ 1 callersClassOutputProcess
closd/diffusion_planner/model/mdm.py:363
↓ 1 callersClassPlotParams
closd/utils/poselib/poselib/visualization/common.py:74
↓ 1 callersClassPositionalEncoding
closd/diffusion_planner/model/mdm.py:299
↓ 1 callersClassRLGPUAlgoObserver
closd/run.py:149
↓ 1 callersClassRLGPUEnv
closd/run.py:184
↓ 1 callersClassRepresentationHandler
closd/utils/rep_util.py:68
↓ 1 callersClassRunningStat
closd/utils/smpllib/khrylib/utils/zfilter.py:7
↓ 1 callersClassSimpleCNN
closd/utils/smpllib/khrylib/models/simple_cnn.py:5
↓ 1 callersClassSkeleton
closd/utils/smpllib/khrylib/mocap/skeleton_local.py:100
↓ 1 callersClassSkeletonState
A skeleton state contains all the information needed to describe a static state of a skeleton. It requires a skeleton tree, local/global rota
closd/utils/poselib/poselib/skeleton/skeleton3d.py:253
↓ 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
closd/diffusion_planner/diffusion/respace.py:65
↓ 1 callersClassTemporalBlock
closd/utils/smpllib/khrylib/models/tcn.py:15
↓ 1 callersClassTensorBoardOutputFormat
Dumps key/value pairs into TensorBoard's numeric format.
closd/diffusion_planner/diffusion/logger.py:150
↓ 1 callersClassTextDecoder
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:187
↓ 1 callersClassTextEncoderBiGRU
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:267
↓ 1 callersClassTextVAEDecoder
closd/diffusion_planner/data_loaders/humanml/networks/modules.py:123
↓ 1 callersClassTimestepEmbedder
closd/diffusion_planner/model/mdm.py:319
↓ 1 callersClassTrainLoop
closd/diffusion_planner/train/training_loop.py:37
↓ 1 callersClassUniformSampler
closd/diffusion_planner/diffusion/resample.py:61
next →1–100 of 226, ranked by callers