MCPcopy Create free account

hub / github.com/GuyTevet/MotionCLIP / functions

Functions171 in github.com/GuyTevet/MotionCLIP

↓ 1 callersFunctionstack_images_sequence
(visu)
src/visualize/visualize.py:413
↓ 1 callersFunctionstack_var_and_reduce
Equation (5) from the paper.
src/models/tools/hessian_penalty.py:91
↓ 1 callersFunctionstandardize_quaternion
Convert a unit quaternion to a standard form: one in which the real part is non negative. Args: quaternions: Quaternions with re
src/utils/rotation_conversions.py:325
↓ 1 callersFunctionviz_clip_edit
Generate & viz samples
src/visualize/visualize.py:349
↓ 1 callersFunctionviz_clip_interp
Generate & viz samples
src/visualize/visualize.py:288
↓ 1 callersFunctionviz_clip_text
Generate & viz samples
src/visualize/visualize.py:232
Method__call__
(self, x, mask, pose_rep, translation, glob, jointstype, vertstrans, betas=None, beta=0,
src/models/rotation2xyz.py:13
Method__enter__
(self)
src/utils/video.py:17
Method__exit__
(self, exc_type, exc_value, exc_traceback)
src/utils/video.py:23
Method__getitem__
(self, index)
src/datasets/dataset.py:95
Method__iadd__
(self, data)
src/utils/video.py:26
Method__init__
(self, outname, fps)
src/utils/video.py:13
Method__init__
(self, num_frames=1, sampling="conseq", sampling_step=1, split="train", pose_rep="rot6d", tra
src/datasets/dataset.py:15
Method__init__
(self, datapath="data/amass/amass_30fps_legacy_db.pt", split="train", use_z=1, **kwargs)
src/datasets/amass.py:139
Method__init__
(self, model_path=SMPL_MODEL_PATH, **kwargs)
src/models/smpl.py:63
Method__init__
(self, device)
src/models/rotation2xyz.py:9
Method__init__
(self, encoder, decoder, device, lambdas, latent_dim, outputxyz, pose_rep, glob, glob_rot, tr
src/models/modeltype/motionclip.py:17
Method__init__
(self, d_model, dropout=0.1, max_len=5000)
src/models/architectures/transformer.py:8
Method__init__
(self, modeltype, njoints, nfeats, num_frames, num_classes, translation, pose_rep, glob, glob_rot,
src/models/architectures/transformer.py:43
Method__init__
(self, modeltype, njoints, nfeats, num_frames, num_classes, translation, pose_rep, glob, glob_rot,
src/models/architectures/transformer.py:113
Method__len__
(self)
src/datasets/dataset.py:351
Method__repr__
(self)
src/datasets/dataset.py:362
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
src/utils/rotation_conversions.py:122
Method_load_betas
(self, ind, frame_ix)
src/datasets/amass.py:267
Method_load_body_features
(self, ind, frame_ix)
src/datasets/amass.py:275
Method_load_gender
(self, ind, frame_ix)
src/datasets/amass.py:271
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,
src/utils/rotation_conversions.py:417
Functioncleanexit
()
src/utils/misc.py:22
Functioncompute_hp_loss
(model, batch)
src/models/tools/losses.py:67
Functioncompute_mmd_loss
(model, batch)
src/models/tools/losses.py:72
Functioncompute_rc_loss
(model, batch, use_txt_output=False)
src/models/tools/losses.py:7
Functioncompute_rcxyz_loss
(model, batch, use_txt_output=False)
src/models/tools/losses.py:20
Functioncompute_vel_loss
(model, batch, use_txt_output=False)
src/models/tools/losses.py:33
Functioncompute_velxyz_loss
(model, batch, use_txt_output=False)
src/models/tools/losses.py:50
Functioneuler_angles_to_matrix
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sh
src/utils/rotation_conversions.py:150
Methodforward
(self, *args, **kwargs)
src/models/smpl.py:82
Methodforward
(self, batch)
src/models/modeltype/motionclip.py:271
Methodforward
(self, x)
src/models/architectures/transformer.py:21
Methodforward
(self, x, mask, lengths)
src/models/architectures/transformer.py:33
Methodforward
(self, batch)
src/models/architectures/transformer.py:85
Methodforward
(self, batch, use_text_emb=False)
src/models/architectures/transformer.py:164
Functiongenerate_by_video_sequences
(visualization, label_to_action_name, params, nats, nspa, tmp_path)
src/visualize/visualize.py:189
Methodgenerate_from_embedding
(self, classes, durations, nspa=1, is_amass=False, classes_gaussians=None)
src/models/modeltype/motionclip.py:233
Methodgenerate_one
(self, cls, duration, fact=1, xyz=False)
src/models/modeltype/motionclip.py:173
Functionget_dataset
(name="amass")
src/datasets/get_dataset.py:3
Methodget_label_sample_batch
(self, labels)
src/datasets/dataset.py:302
Functionget_loss_names
()
src/models/tools/losses.py:88
Functionget_model
(parameters, clip_model)
src/models/get_model.py:8
Methodget_stats
(self)
src/datasets/dataset.py:331
Functionget_trans_from_vibe
(vibe, use_z=True)
src/datasets/amass.py:111
Functioninit
()
src/visualize/anim.py:73
Methodlabel_to_action_name
(self, label)
src/datasets/dataset.py:91
Functionload_video
(filename)
src/utils/video.py:5
Functionmatrix_to_axis_angle
Convert rotations given as rotation matrices to axis/angle. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Ret
src/utils/rotation_conversions.py:433
Functionmatrix_to_euler_angles
Convert rotations given as rotation matrices to Euler angles in radians. Args: matrix: Rotation matrices as tensor of shape (..., 3,
src/utils/rotation_conversions.py:217
Functionmatrix_to_rotation_6d
Converts rotation matrices to 6D rotation representation by Zhou et al. [1] by dropping the last row. Note that 6D representation is not uniq
src/utils/rotation_conversions.py:536
Methodparse_action
(self, path, return_int=True)
src/datasets/dataset.py:78
Functionparser
()
src/parser/evaluation.py:8
Functionparser
()
src/parser/generate.py:25
Functionparser
()
src/parser/finetunning.py:5
Functionplot_3d_motion_dico
(x)
src/visualize/anim.py:152
Functionquaternion_apply
Apply the rotation given by a quaternion to a 3D point. Usual torch rules for broadcasting apply. Args: quaternion: Tensor of qu
src/utils/rotation_conversions.py:394
Functionquaternion_multiply
Multiply two quaternions representing rotations, returning the quaternion representing their composition, i.e. the versor with nonnegative re
src/utils/rotation_conversions.py:361
Functionrandom_rotation
Generate a single random 3x3 rotation matrix. Args: dtype: Type to return device: Device of returned tensor. Default: if Non
src/utils/rotation_conversions.py:306
Methodreset_shuffle
(self)
src/datasets/dataset.py:380
Functionrotation_6d_to_matrix
Converts 6D rotation representation by Zhou et al. [1] to rotation matrix using Gram--Schmidt orthogonalisation per Section B of [1]. Arg
src/utils/rotation_conversions.py:512
Methodshuffle
(self)
src/datasets/dataset.py:374
Functiontest
(model, optimizer, iterator, device)
src/train/trainer.py:68
Functionto_numpy
(tensor)
src/utils/misc.py:4
Functiontraining_parser
()
src/parser/recognition.py:10
Functionupdate
(index)
src/visualize/anim.py:125
← previous101–171 of 171, ranked by callers