MCPcopy Create free account

hub / github.com/Sirui-Xu/STARS / functions

Functions181 in github.com/Sirui-Xu/STARS

Method__init__
(self, mode, t_his, t_pred, actions='all')
diverse/motion_pred/utils/dataset.py:6
Method__init__
(self, mode, t_his=15, t_pred=60, actions='all', **kwargs)
diverse/motion_pred/utils/dataset_humaneva_multimodal.py:10
Method__init__
(self, mode, t_his=15, t_pred=60, actions='all', **kwargs)
diverse/motion_pred/utils/dataset_humaneva.py:9
Method__init__
(self, mode, t_his=25, t_pred=100, actions='all', use_vel=False)
diverse/motion_pred/utils/dataset_h36m.py:9
Method__init__
(self, parents, joints_left, joints_right)
diverse/motion_pred/utils/skeleton.py:12
Method__init__
(self, time_dim, joints_dim )
deterministic/model.py:37
Method__init__
(self, time_dim, joints_dim, dim_used = [ 2, 3, 4, 5, 7
deterministic/model.py:92
Method__init__
(self, in_channels, out_channels, kernel_size,
deterministic/model.py:177
Method__init__
:param path_to_data: :param actions: :param input_n: :param output_n: :param dct_used: :param split:
deterministic/utils/h36motion3d.py:17
Method__init__
:param path_to_data: :param actions: :param input_n: :param output_n: :param dct_used: :param split:
deterministic/utils/h36motion3dab.py:17
Method__len__
(self)
deterministic/utils/h36motion3dab.py:136
Function_pairwise_distances
Compute the 2D matrix of distances between all the embeddings. Args: embeddings: tensor of shape (batch_size, embed_dim) squared:
diverse/utils/util.py:20
Function_pairwise_distances_l1
Compute the 2D matrix of distances between all the embeddings. Args: embeddings: tensor of shape (batch_size, embed_dim) squared:
diverse/utils/util.py:57
Function_some_variables_cmu
We define some variables that are useful to run the kinematic tree Args None Returns parent: 32-long vector with parent-chil
deterministic/utils/forward_kinematics.py:138
Functionabsolute2relative
x: [bs,..., jn, 3] or [bs,..., jn-1, 3] if invert x0: [1,..., jn, 3] parents: [-1,0,1 ...]
diverse/utils/util.py:92
Functionabsolute2relative_torch
x: [bs,..., jn, 3] or [bs,..., jn-1, 3] if invert x0: [1,..., jn, 3] parents: [-1,0,1 ...]
diverse/utils/util.py:113
Functionbatch_to
(dst, *args)
diverse/utils/torch.py:82
Methodchildren
(self)
diverse/motion_pred/utils/skeleton.py:29
Functioncompute_amse
(pred, gt, *args)
diverse/utils/metrics.py:39
Functioncompute_diversity
(pred, *args)
diverse/utils/metrics.py:14
Functioncompute_flat_grad
(output, inputs, filter_input_ids=set(), retain_graph=False, create_graph=False)
diverse/utils/torch.py:122
Functioncompute_fmse
(pred, gt, *args)
diverse/utils/metrics.py:45
Functioncompute_mmade
(pred, gt, gt_multi)
diverse/utils/metrics.py:50
Functioncompute_mmfde
(pred, gt, gt_multi)
diverse/utils/metrics.py:61
Functiondefine_actions_cmu
Define the list of actions we are using. Args action: String with the passed action. Could be "all" Returns actions: List of
deterministic/utils/data_utils.py:310
Functionexpmap2quat_torch
Converts expmap to quaternion batch pytorch version ported from the corresponding numpy method above :param R: N*3 :return: N*4
deterministic/utils/data_utils.py:539
Functionexpmap2rotmat
Converts an exponential map angle to a rotation matrix Matlab port to python for evaluation purposes I believe this is also called Rodrig
diverse/utils/util.py:72
Functionexpmap2rotmat_torch
Converts expmap matrix to rotation batch pytorch version ported from the corresponding method above :param r: N*3 :return: N*3*3
deterministic/utils/data_utils.py:555
Functionexpmap2xyz_global
convert expmaps to joint locations :param expmap: N*99 :return: N*32*3
deterministic/utils/data_utils.py:627
Functionexpmap2xyz_torch
convert expmaps to joint locations :param expmap: N*99 :return: N*32*3
deterministic/utils/data_utils.py:577
Functionfilter_state_dict
(state_dict, filter_keys)
diverse/utils/torch.py:154
Functionfind_indices_256
Adapted from https://github.com/una-dinosauria/human-motion-prediction/blob/master/src/seq2seq_model.py#L478 which originaly from In ord
deterministic/utils/data_utils.py:659
Functionfind_indices_srnn
Adapted from https://github.com/una-dinosauria/human-motion-prediction/blob/master/src/seq2seq_model.py#L478 which originaly from In ord
deterministic/utils/data_utils.py:691
Methodforward
x,x_cond: [bs,data_dim]
diverse/models/LinNF.py:32
Methodforward
(self, input, logdet)
diverse/models/LinNF.py:69
Methodforward
(self, x)
diverse/models/LinNF.py:93
Methodforward
@param X: @return:
diverse/models/Orthogonal.py:23
Methodforward
(self, x)
diverse/models/model.py:44
Methodforward
(self, x)
diverse/models/model.py:108
Methodforward
(self, x)
diverse/models/model.py:186
Methodforward
(self, x, y0)
diverse/models/model.py:310
Methodforward
(self, x)
deterministic/model.py:60
Methodforward
(self, x)
deterministic/model.py:153
Methodforward
(self, x)
deterministic/model.py:231
Methodforward
(self, x)
deterministic/model.py:325
Functionget_dct_matrix
(N, is_torch=True)
diverse/utils/util.py:5
Functionget_dct_matrix
(N)
deterministic/utils/data_utils.py:647
Functionget_flat_grad_from
(inputs, grad_grad=False)
diverse/utils/torch.py:107
Functionget_flat_params_from
(models)
diverse/utils/torch.py:86
Functionh36m_valid_angle_check
p3d: [bs,16,3] or [bs,48]
diverse/utils/valid_angle_check.py:6
Functionh36m_valid_angle_check_torch
p3d: [bs,16,3] or [bs,48]
diverse/utils/valid_angle_check.py:145
Methodhas_children
(self)
diverse/motion_pred/utils/skeleton.py:26
Functionhumaneva_valid_angle_check
p3d: [bs,14,3] or [bs,42]
diverse/utils/valid_angle_check.py:280
Functionhumaneva_valid_angle_check_torch
p3d: [bs,14,3] or [bs,42]
diverse/utils/valid_angle_check.py:375
Methodinverse
(self, input, logdet)
diverse/models/LinNF.py:75
Methodinverse
(self, z)
diverse/models/LinNF.py:103
Methoditer_generator
(self, step=25, n_modality=10)
diverse/motion_pred/utils/dataset_h36m_multimodal.py:171
Methoditer_generator
(self, step=25)
diverse/motion_pred/utils/dataset_humaneva_multimodal.py:198
Methodjoints_left
(self)
diverse/motion_pred/utils/skeleton.py:72
Functionlambda_rule
(epoch)
diverse/utils/torch.py:164
Methodlgdet
(self, X)
diverse/models/Orthogonal.py:39
Functionload_data_cmu
(path_to_dataset, actions, input_n, output_n, data_std=0, data_mean=0, is_test=False)
deterministic/utils/data_utils.py:333
Functionload_data_cmu_3d
(path_to_dataset, actions, input_n, output_n, data_std=0, data_mean=0, is_test=False)
deterministic/utils/data_utils.py:397
Functionmpjpe_error
(pred, gt, *args)
diverse/utils/metrics.py:4
Functionnormalization_stats
Also borrowed for SRNN code. Computes mean, stdev and dimensions to ignore. https://github.com/asheshjain399/RNNexp/blob/srnn/structural_rnn/
deterministic/utils/data_utils.py:252
Functionnormalize_data
Normalize input data by removing unused dimensions, subtracting the mean and dividing by the standard deviation Args data: nx99 ma
deterministic/utils/data_utils.py:218
Methodnum_joints
(self)
diverse/motion_pred/utils/skeleton.py:20
Functionon_key
(event)
diverse/motion_pred/utils/visualization.py:136
Methodprepare_data
(self)
diverse/motion_pred/utils/dataset_h36m_multimodal.py:25
Methodprepare_data
(self)
diverse/motion_pred/utils/dataset_humaneva_multimodal.py:22
Methodprepare_data
(self)
diverse/motion_pred/utils/dataset_humaneva.py:12
Methodprepare_data
(self)
diverse/motion_pred/utils/dataset_h36m.py:15
Functionrevert_output_format
Converts the output of the neural network to a format that is more easy to manipulate for, e.g. conversion to other format or visualization
deterministic/utils/data_utils.py:167
Functionrotmat2euler
Converts a rotation matrix to Euler angles Matlab port to python for evaluation purposes https://github.com/asheshjain399/RNNexp/blob/srn
deterministic/utils/data_utils.py:11
Functionrotmat2euler_torch
Converts a rotation matrix to euler angles batch pytorch version ported from the corresponding numpy method above :param R:N*3*3 :re
deterministic/utils/data_utils.py:467
Functionrotmat2quat_torch
Converts a rotation matrix to quaternion batch pytorch version ported from the corresponding numpy method above :param R: N * 3 * 3 :
deterministic/utils/data_utils.py:512
Functionset_flat_params_to
(model, flat_params)
diverse/utils/torch.py:98
Functionset_optimizer_lr
(optimizer, lr)
diverse/utils/torch.py:149
Functionupdate
(num,data_gt,data_pred,plots_gt,plots_pred,fig,ax,input_n)
deterministic/utils/h36_3d_viz.py:81
Functionvis_poses
@param gt_pos: seq_l x jn x 3 @param pre_pos: n seq_l x jn x 3 @param labels: n @param comments: @return:
diverse/utils/vis_poses.py:63
Functionvis_traj
@param gt_pos: seq_l x jn x 3 @param pre_pos: n seq_l x jn x 3 @param labels: n @param comments: @return:
diverse/utils/vis_poses.py:12
← previous101–181 of 181, ranked by callers