MCPcopy Create free account

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

Functions181 in github.com/Sirui-Xu/STARS

↓ 10 callersMethodparents
(self)
diverse/motion_pred/utils/skeleton.py:23
↓ 8 callersMethodnormalize_data
(self, mean=None, std=None)
diverse/motion_pred/utils/dataset.py:23
↓ 6 callersMethodsampling_generator
(self, num_samples=1000, batch_size=8)
diverse/motion_pred/utils/dataset.py:50
↓ 4 callersFunctioncreate_pose
(ax,plots,vals,pred=True,update=False,prediction=False)
deterministic/utils/h36_3d_viz.py:15
↓ 3 callersMethod__init__
(self, n_pre, nk, input_channels, input_ti
diverse/models/model.py:209
↓ 3 callersMethod__init__
(self, input_channels, input_time_frame, output_time_frame,
deterministic/model.py:254
↓ 3 callersMethod__len__
(self)
deterministic/utils/h36motion3d.py:147
↓ 3 callersFunctionloss_function
(prior_lkh, log_det_jacobian)
diverse/train_nf.py:19
↓ 3 callersFunctionmpjpe_error
(batch_pred,batch_gt)
deterministic/utils/loss_funcs.py:9
↓ 2 callersMethod__init__
(self, data_dim)
diverse/models/LinNF.py:17
↓ 2 callersMethod_compute_metadata
(self)
diverse/motion_pred/utils/skeleton.py:78
↓ 2 callersFunctioncombine_dict
(new_dict, old_dict=None)
diverse/utils/logger.py:30
↓ 2 callersFunctioncreate_logger
(filename, file_handle=True)
diverse/utils/logger.py:6
↓ 2 callersFunctiondefine_actions
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:279
↓ 2 callersFunctiondraw_skeleton
:param kpts: joint_n*(3 or 2) :param parents: :return:
diverse/utils/vis_util.py:6
↓ 2 callersFunctionfinal_mpjpe_error
(batch_pred,batch_gt)
deterministic/utils/loss_funcs.py:17
↓ 2 callersFunctionfkl
Convert joint angles and bone lenghts into the 3d points of a person. adapted from https://github.com/una-dinosauria/human-motion-predic
deterministic/utils/forward_kinematics.py:7
↓ 2 callersFunctionget_model
(cfg, dataset, model_type='h36m')
diverse/models/motion_pred.py:4
↓ 2 callersFunctionget_prediction
(data, model, sample_num, num_seeds=1, concat_hist=True)
diverse/main.py:239
↓ 2 callersFunctionget_scheduler
(optimizer, policy, nepoch_fix=None, nepoch=None, decay_step=None)
diverse/utils/torch.py:162
↓ 2 callersMethodinverse
x,x_cond: [bs,data_dim]
diverse/models/LinNF.py:45
↓ 2 callersMethoditer_generator
(self, step=25)
diverse/motion_pred/utils/dataset.py:59
↓ 2 callersMethodjoints_right
(self)
diverse/motion_pred/utils/skeleton.py:75
↓ 2 callersFunctionreadCSVasFloat
Borrowed from SRNN code. Reads a csv and returns a float matrix. https://github.com/asheshjain399/NeuralModels/blob/master/neuralmodels/utils
deterministic/utils/data_utils.py:197
↓ 2 callersMethodremove_joints
Remove the joints specified in 'joints_to_remove'.
diverse/motion_pred/utils/skeleton.py:32
↓ 2 callersMethodsample
(self)
diverse/motion_pred/utils/dataset.py:40
↓ 2 callersFunctionsave
()
diverse/motion_pred/utils/visualization.py:157
↓ 2 callersMethodsequential_mult
(self, V, X)
diverse/models/Orthogonal.py:17
↓ 2 callersFunctiontest
(model, epoch)
diverse/main.py:268
↓ 2 callersMethodupdate
(self, val, n=1)
diverse/utils/logger.py:51
↓ 2 callersFunctionupdate_video
(i)
diverse/motion_pred/utils/visualization.py:68
↓ 1 callersFunction_some_variables
borrowed from https://github.com/una-dinosauria/human-motion-prediction/blob/master/src/forward_kinematics.py#L100 We define some variab
deterministic/utils/forward_kinematics.py:68
↓ 1 callersFunctionangle_loss
(y)
diverse/main.py:64
↓ 1 callersFunctioncompute_ade
(pred, gt, *args)
diverse/utils/metrics.py:22
↓ 1 callersFunctioncompute_fde
(pred, gt, *args)
diverse/utils/metrics.py:31
↓ 1 callersFunctiondenomarlize
(*data)
diverse/main.py:309
↓ 1 callersFunctionexpmap2rotmat
Converts an exponential map angle to a rotation matrix Matlab port to python for evaluation purposes I believe this is also called Rodrig
deterministic/utils/data_utils.py:77
↓ 1 callersFunctionexpmap2xyz_torch_cmu
(expmap)
deterministic/utils/data_utils.py:588
↓ 1 callersFunctionfkl_torch
pytorch version of fkl. convert joint angles to joint locations batch pytorch version of the fkl() method above :param angles: N*99
deterministic/utils/forward_kinematics.py:219
↓ 1 callersMethodget_dct_matrix
(self, N, is_torch=True)
diverse/models/model.py:286
↓ 1 callersMethodget_dct_matrix
(self, N, is_torch=True)
deterministic/model.py:311
↓ 1 callersFunctionget_multimodal_gt
(dataset_test)
diverse/main.py:213
↓ 1 callersMethodinverse
(self, X)
diverse/models/Orthogonal.py:35
↓ 1 callersFunctionloss_function
(traj_est, traj, traj_multimodal, prior_lkh, prior_logdetjac, _lambda, mu, logvar)
diverse/main.py:87
↓ 1 callersFunctionmain
()
deterministic/utils/forward_kinematics.py:246
↓ 1 callersFunctionpose_generator
()
diverse/main.py:324
↓ 1 callersFunctionpost_process
(pred, data)
diverse/main.py:316
↓ 1 callersMethodprepare_data
(self)
diverse/motion_pred/utils/dataset.py:20
↓ 1 callersMethodprocess_data
(self)
diverse/motion_pred/utils/dataset_h36m_multimodal.py:41
↓ 1 callersMethodprocess_data
(self)
diverse/motion_pred/utils/dataset_humaneva_multimodal.py:33
↓ 1 callersMethodprocess_data
(self)
diverse/motion_pred/utils/dataset_humaneva.py:23
↓ 1 callersMethodprocess_data
(self)
diverse/motion_pred/utils/dataset_h36m.py:31
↓ 1 callersFunctionquat2expmap
Converts a quaternion to an exponential map Matlab port to python for evaluation purposes https://github.com/asheshjain399/RNNexp/blob/sr
deterministic/utils/data_utils.py:97
↓ 1 callersFunctionrecon_loss
(Y_g, Y, Y_mm, Y_hg, Y_h)
diverse/main.py:22
↓ 1 callersFunctionreload_poses
()
diverse/motion_pred/utils/visualization.py:119
↓ 1 callersFunctionrender_animation
TODO Render an animation. The supported output modes are: -- 'interactive': display an interactive figure (also w
diverse/motion_pred/utils/visualization.py:16
↓ 1 callersMethodreparameterize
(self, x)
diverse/models/model.py:300
↓ 1 callersMethodreset
(self)
diverse/utils/logger.py:45
↓ 1 callersMethodreset_parameters
(self)
diverse/models/LinNF.py:25
↓ 1 callersMethodreset_parameters
(self)
diverse/models/Orthogonal.py:14
↓ 1 callersFunctionrevert_coordinate_space
Bring a series of poses to a canonical form so they are facing the camera when they start. Args channels: n-by-99 matrix of poses: 3 + 3
deterministic/utils/data_utils.py:593
↓ 1 callersFunctionrotmat2expmap
(R)
deterministic/utils/data_utils.py:73
↓ 1 callersFunctionrotmat2quat
Converts a rotation matrix to a quaternion Matlab port to python for evaluation purposes https://github.com/asheshjain399/RNNexp/blob/srn
deterministic/utils/data_utils.py:43
↓ 1 callersMethodsample
(self, n_modality=5)
diverse/motion_pred/utils/dataset_h36m_multimodal.py:101
↓ 1 callersMethodsample
(self, n_modality=5)
diverse/motion_pred/utils/dataset_humaneva_multimodal.py:79
↓ 1 callersMethodsampling_generator
(self, num_samples=1000, batch_size=8, n_modality=5)
diverse/motion_pred/utils/dataset_h36m_multimodal.py:159
↓ 1 callersMethodsampling_generator
(self, num_samples=1000, batch_size=8, n_modality=5)
diverse/motion_pred/utils/dataset_humaneva_multimodal.py:140
↓ 1 callersFunctionsave_figs
()
diverse/motion_pred/utils/visualization.py:128
↓ 1 callersFunctionshow_animation
()
diverse/motion_pred/utils/visualization.py:112
↓ 1 callersFunctiontest
()
deterministic/main.py:123
↓ 1 callersFunctiontrain
(epoch)
diverse/train_nf.py:27
↓ 1 callersFunctiontrain
(epoch, stats)
diverse/main.py:128
↓ 1 callersFunctiontrain
()
deterministic/main.py:34
↓ 1 callersFunctionunNormalizeData
Borrowed from SRNN code. Reads a csv file and returns a float32 matrix. https://github.com/asheshjain399/RNNexp/blob/srnn/structural_rnn/CRFProble
deterministic/utils/data_utils.py:128
↓ 1 callersFunctionval
(epoch)
diverse/train_nf.py:74
↓ 1 callersFunctionvisualize
()
diverse/main.py:308
↓ 1 callersFunctionvisualize
(input_n,output_n,visualize_from,path,modello,device,n_viz,skip_rate,actions,global_translation,model_name)
deterministic/utils/h36_3d_viz.py:107
Method__enter__
(self)
diverse/utils/torch.py:22
Method__enter__
(self)
diverse/utils/torch.py:39
Method__enter__
(self)
diverse/utils/torch.py:56
Method__enter__
(self)
diverse/utils/torch.py:73
Method__exit__
(self, *args)
diverse/utils/torch.py:25
Method__exit__
(self, *args)
diverse/utils/torch.py:42
Method__exit__
(self, *args)
diverse/utils/torch.py:59
Method__exit__
(self, *args)
diverse/utils/torch.py:76
Method__getitem__
(self, item)
deterministic/utils/h36motion3d.py:150
Method__getitem__
(self, item)
deterministic/utils/h36motion3dab.py:139
Method__init__
(self, *models)
diverse/utils/torch.py:16
Method__init__
(self, device, *models)
diverse/utils/torch.py:33
Method__init__
(self, *models)
diverse/utils/torch.py:50
Method__init__
(self, *models)
diverse/utils/torch.py:67
Method__init__
(self)
diverse/utils/logger.py:42
Method__init__
(self, num_parameters: int = 1, init: float = 0.25)
diverse/models/LinNF.py:64
Method__init__
(self, data_dim, num_layer=5, with_prelu=True)
diverse/models/LinNF.py:83
Method__init__
(self, d)
diverse/models/Orthogonal.py:7
Method__init__
(self, time_dim, joints_dim )
diverse/models/model.py:21
Method__init__
(self, time_dim, joints_dim, pose_info )
diverse/models/model.py:64
Method__init__
(self, in_channels, out_channels, kernel_size,
diverse/models/model.py:130
Method__init__
(self, cfg_id, test=False, nf=False)
diverse/motion_pred/utils/config.py:7
Method__init__
(self, mode, t_his=25, t_pred=100, actions='all', use_vel=False, **kwargs)
diverse/motion_pred/utils/dataset_h36m_multimodal.py:10
next →1–100 of 181, ranked by callers