Functions — github.com/Walter0807/MotionBERT

CodeHub MCPcopy Create free account

hub / github.com/Walter0807/MotionBERT / functions

Functions186 in github.com/Walter0807/MotionBERT

⨍ Functions 186◇ Types & classes 30

↓ 1 callersFunctionread_input

(json_path, vid_size, scale_range, focus)

lib/data/dataset_wild.py:67

↓ 1 callersMethodreset

lib/utils/learning.py:13

↓ 1 callersFunctionrigid_transform_3D

lib/utils/utils_mesh.py:333

↓ 1 callersFunctionrot6d_to_rotmat

lib/utils/utils_mesh.py:316

↓ 1 callersFunctionrotation_matrix_to_angle_axis

This function is borrowed from https://github.com/kornia/kornia Convert 3x4 rotation matrix to Rodrigues vector Args: rotation_

lib/utils/utils_mesh.py:54

↓ 1 callersFunctionrotation_matrix_to_quaternion

This function is borrowed from https://github.com/kornia/kornia Convert 3x4 rotation matrix to 4d quaternion vector This algorithm is b

lib/utils/utils_mesh.py:139

↓ 1 callersFunctionsave_image

(image_numpy, image_path)

lib/utils/vismo.py:340

↓ 1 callersFunctionset_random_seed

train_mesh.py:46

↓ 1 callersFunctionset_random_seed

↓ 1 callersMethodsmpl_losses

(self, pred_rotmat, pred_betas, gt_pose, gt_betas)

lib/model/loss_mesh.py:57

↓ 1 callersFunctionsolve_scale

infer_wild_mesh.py:45

↓ 1 callersFunctiontrain_with_config

train_mesh.py:218

↓ 1 callersFunctiontrain_with_config

↓ 1 callersFunctiontrain_with_config

train_action_1shot.py:71

↓ 1 callersFunctiontrain_with_config

train_action.py:78

↓ 1 callersFunctiontraverse

tools/compress_amass.py:13

Method__getitem__

lib/data/dataset_motion_3d.py:31

Method__getitem__

Generates one sample of data

lib/data/dataset_motion_3d.py:42

Method__getitem__

lib/data/dataset_mesh.py:55

Method__getitem__

Generates one sample of data

lib/data/dataset_mesh.py:63

Method__getitem__

Generates one sample of data

lib/data/dataset_wild.py:98

Method__getitem__

lib/data/dataset_action.py:166

Method__getitem__

Generates one sample of data

lib/data/dataset_action.py:173

Method__getitem__

Generates one sample of data

lib/data/dataset_action.py:197

Method__getitem__

Generates one sample of data

lib/data/dataset_motion_2d.py:116

Method__getitem__

Generates one sample of data

lib/data/dataset_motion_2d.py:139

(self, log_path)

lib/utils/tools.py:11

Initialise Loader.

lib/utils/tools.py:22

(self, *args, **kwargs)

lib/utils/utils_smpl.py:61

lib/utils/learning.py:10

(self, drop_prob=None)

lib/model/drop.py:38

(self, dropout_ratio=0., dim_rep=512, num_classes=60, num_joints=17, hidden_dim=2048)

lib/model/model_action.py:7

(self, dropout_ratio=0., dim_rep=512, num_joints=17, hidden_dim=2048)

lib/model/model_action.py:32

( self, loss_type='MSE', device='cuda', )

lib/model/loss_mesh.py:8

(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., st_mode='vanilla')

lib/model/DSTformer.py:89

(self, dim, num_heads, mlp_ratio=4., mlp_out_ratio=1., qkv_bias=True, qk_scale=None, drop=0., attn_drop=0.,

lib/model/DSTformer.py:216

(self, dim_in=3, dim_out=3, dim_feat=256, dim_rep=512, depth=5, num_heads=8, mlp_ratio=4,

lib/model/DSTformer.py:270

(self, temperature=0.07, contrast_mode='all', base_temperature=0.07)

lib/model/loss_supcon.py:14

(self, args, backbone, dim_rep=512, num_joints=17, hidden_dim=2048, dropout_ratio=0.5)

lib/model/model_mesh.py:83

(self, args, subset_list, data_split)

lib/data/dataset_motion_3d.py:35

(self, n_frames, sample_stride, data_stride_train, data_stride_test, read_confidence=True, dt_root = 'data/mot

lib/data/datareader_h36m.py:12

(self, args, data_split, dataset)

lib/data/dataset_mesh.py:20

(self, args, data_split, dataset)

lib/data/dataset_mesh.py:59

lib/data/augmentation.py:14

lib/data/augmentation.py:87

(self, json_path, clip_len=243, vid_size=None, scale_range=None, focus=None)

lib/data/dataset_wild.py:89

(self, data_path, data_split, n_frames=243, random_move=True, scale_range=[1,1], check_split=True)

lib/data/dataset_action.py:131

(self, data_path, data_split, n_frames=243, random_move=True, scale_range=[1,1], check_split=False)

lib/data/dataset_action.py:185

(self, flip=True, scale_range=[0.25, 1])

lib/data/dataset_motion_2d.py:78

(self, n_frames, sample_stride, data_stride_train, data_stride_test, read_confidence=True, dt_root = 'data/mes

lib/data/datareader_mesh.py:8

Denotes the total number of samples

lib/data/dataset_motion_3d.py:27

Denotes the total number of samples

lib/data/dataset_mesh.py:51

Denotes the total number of samples

lib/data/dataset_wild.py:94

Denotes the total number of samples

lib/data/dataset_action.py:162

Denotes the total number of samples

lib/data/dataset_motion_2d.py:112

Denotes the total number of samples

lib/data/dataset_motion_2d.py:135

Method_init_weights

lib/model/DSTformer.py:313

Functioncompute_error_frames

(output, target)

lib/utils/utils_mesh.py:376

Functionconstruct_include

Include file referenced at node.

lib/utils/tools.py:32

Methodcount_attn

lib/model/DSTformer.py:202

infer_wild_mesh.py:42

Functionestimate_translation

This function is borrowed from https://github.com/nkolot/SPIN/utils/geometry.py Find camera translation that brings 3D joints S closest to 2

lib/utils/utils_mesh.py:266

(self, *args, **kwargs)

lib/utils/utils_smpl.py:71

lib/model/drop.py:42

Input: (N, M, T, J, C)

lib/model/model_action.py:15

Input: (N, M, T, J, C)

lib/model/model_action.py:36

Input: (N, M x T x 17 x 3)

lib/model/model_action.py:62

( self, smpl_output, data_gt, )

lib/model/loss_mesh.py:23

lib/model/DSTformer.py:79

(self, x, seqlen=1)

lib/model/DSTformer.py:109

(self, x, seqlen=1)

lib/model/DSTformer.py:239

Compute loss for model. If both `labels` and `mask` are None, it degenerates to SimCLR unsupervised loss: https://arxiv.org/pdf/2002.0

lib/model/loss_supcon.py:21

(self, feat, init_pose=None, init_shape=None)

lib/model/model_mesh.py:37

Input: (N x T x 17 x 3)

lib/model/model_mesh.py:89

Functionget_action_names

(file_path = "data/action/ntu_actions.txt")

lib/data/dataset_action.py:10

Methodget_classifier

lib/model/DSTformer.py:322

Methodget_sliced_data

lib/data/datareader_mesh.py:51

Functionloss_joint

(predicted, target)

lib/model/loss.py:144

Functionpixel2world_vis

lib/utils/vismo.py:39

Functionrectify_pose

Rectify "upside down" people in global coord Args: pose (72,): Pose. Returns: Rotated pose.

lib/utils/utils_mesh.py:441

Methodreset_classifier

(self, dim_out, global_pool='')

lib/model/DSTformer.py:325

Functionrot6d_to_rotmat_spin

Convert 6D rotation representation to 3x3 rotation matrix. Based on Zhou et al., "On the Continuity of Rotation Representations in Neural Networks

lib/utils/utils_mesh.py:294

Functionvis_data_batch

data_input: [N,T,17,2/3] data_label: [N,T,17,3]

lib/utils/vismo.py:55

Functionweighted_bonelen_loss

(predict_3d_length, gt_3d_length)

lib/model/loss.py:90

Functionweighted_boneratio_loss

(predict_3d_length, gt_3d_length)

lib/model/loss.py:94

Functionweighted_mpjpe

Weighted mean per-joint position error (i.e. mean Euclidean distance)

lib/model/loss.py:64

← previous101–186 of 186, ranked by callers