Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/705062791/PGBIG
/ functions
Functions
88 in github.com/705062791/PGBIG
⨍
Functions
88
◇
Types & classes
12
↓ 6 callers
Method
__len__
(self)
utils/amass.py:136
↓ 6 callers
Function
run_model
(net_pred, optimizer=None, is_train=0, data_loader=None, epo=1, opt=None, dim_used=None)
main_cmu_3d.py:205
↓ 5 callers
Method
parse
(self, makedir=True)
utils/opt.py:71
↓ 5 callers
Function
run_model
(net_pred, optimizer=None, is_train=0, data_loader=None, epo=1, opt=None, dim_used=None)
main_3dpw_3d.py:163
↓ 5 callers
Function
run_model
(net_pred, optimizer=None, is_train=0, data_loader=None, epo=1, opt=None)
main_h36m_3d.py:177
↓ 4 callers
Method
__init__
:param input_feature: num of input feature :param hidden_feature: num of hidden feature :param p_dropout: drop out prob.
model/BaseModel.py:110
↓ 4 callers
Function
readCSVasFloat
Borrowed from SRNN code. Reads a csv and returns a float matrix. https://github.com/asheshjain399/NeuralModels/blob/master/neuralmodels/utils
utils/data_utils.py:197
↓ 3 callers
Method
__len__
(self)
utils/CMU_motion_3d.py:42
↓ 3 callers
Function
expmap2xyz_torch_cmu
(opt, expmap)
utils/data_utils.py:715
↓ 3 callers
Function
smooth
data:[bs, 60, 96]
main_3dpw_3d.py:153
↓ 3 callers
Function
smooth
data:[bs, 60, 96]
main_cmu_3d.py:195
↓ 3 callers
Function
smooth
data:[bs, 60, 96]
main_h36m_3d.py:168
↓ 2 callers
Function
fkl
Convert joint angles and bone lenghts into the 3d points of a person. adapted from https://github.com/una-dinosauria/human-motion-predic
utils/forward_kinematics.py:7
↓ 2 callers
Function
with_zeros
Append a [0, 0, 0, 1] tensor to a [3, 4] tensor. Parameter: --------- x: Tensor to be appended. Return: ------ Tensor a
utils/ang2joint.py:88
↓ 1 callers
Method
_initial
(self)
utils/opt.py:16
↓ 1 callers
Method
_print
(self)
utils/opt.py:66
↓ 1 callers
Function
_some_variables
borrowed from https://github.com/una-dinosauria/human-motion-prediction/blob/master/src/forward_kinematics.py#L100 We define some variab
utils/forward_kinematics.py:68
↓ 1 callers
Function
eval
(opt)
main_3dpw_3d.py:119
↓ 1 callers
Function
eval
(opt)
main_cmu_3d.py:136
↓ 1 callers
Function
eval
(opt)
main_h36m_3d.py:117
↓ 1 callers
Function
fkl_torch
pytorch version of fkl. convert joint angles to joint locations batch pytorch version of the fkl() method above :param angles: N*99
utils/forward_kinematics.py:219
↓ 1 callers
Function
main
(opt)
main_3dpw_3d.py:15
↓ 1 callers
Function
main
(opt)
main_cmu_3d.py:15
↓ 1 callers
Function
main
(opt)
main_h36m_3d.py:19
↓ 1 callers
Function
main
()
utils/forward_kinematics.py:244
↓ 1 callers
Function
quat2expmap
Converts a quaternion to an exponential map Matlab port to python for evaluation purposes https://github.com/asheshjain399/RNNexp/blob/sr
utils/data_utils.py:97
↓ 1 callers
Method
reset_parameters
(self)
model/BaseModel.py:33
↓ 1 callers
Function
rodrigues
Rodrigues' rotation formula that turns axis-angle tensor into rotation matrix in a batch-ed manner. Parameter: ---------- r: Axi
utils/ang2joint.py:55
↓ 1 callers
Function
rotmat2quat
Converts a rotation matrix to a quaternion Matlab port to python for evaluation purposes https://github.com/asheshjain399/RNNexp/blob/srn
utils/data_utils.py:43
↓ 1 callers
Function
unNormalizeData
Borrowed from SRNN code. Reads a csv file and returns a float32 matrix. https://github.com/asheshjain399/RNNexp/blob/srnn/structural_rnn/CRFProble
utils/data_utils.py:128
Method
__getitem__
(self, item)
utils/CMU_motion_3d.py:45
Method
__getitem__
(self, item)
utils/h36motion3d.py:128
Method
__getitem__
(self, item)
utils/h36motion.py:121
Method
__getitem__
(self, item)
utils/amass.py:139
Method
__getitem__
(self, item)
utils/dpw3_3d.py:68
Method
__init__
(self, opt, split, actions='all')
utils/CMU_motion_3d.py:8
Method
__init__
(self)
utils/opt.py:12
Method
__init__
:param path_to_data: :param actions: :param input_n: :param output_n: :param dct_used: :param split:
utils/h36motion3d.py:9
Method
__init__
:param path_to_data: :param actions: :param input_n: :param output_n: :param dct_used: :param split:
utils/h36motion.py:11
Method
__init__
:param path_to_data: :param actions: :param input_n: :param output_n: :param dct_used: :param split:
utils/amass.py:14
Method
__init__
(self, opt, actions=None, split=0)
utils/dpw3_3d.py:14
Method
__init__
(self, in_c, out_c, node_n = 22, seq_len = 35, bias=True)
model/BaseModel.py:16
Method
__init__
Define a residual block of GCN
model/BaseModel.py:69
Method
__init__
:param input_feature: num of input feature :param hidden_feature: num of hidden feature :param p_dropout: drop out prob.
model/BaseModel.py:157
Method
__init__
:param input_feature: num of input feature :param hidden_feature: num of hidden feature :param p_dropout: drop out prob.
model/BaseModel.py:204
Method
__init__
(self, opt)
model/stage_4.py:17
Method
__len__
(self)
utils/h36motion3d.py:125
Method
__len__
(self)
utils/h36motion.py:118
Method
__len__
(self)
utils/dpw3_3d.py:65
Method
__repr__
(self)
model/BaseModel.py:62
Method
__repr__
(self)
model/BaseModel.py:104
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
utils/forward_kinematics.py:138
Function
ang2joint
:param p3d0:[batch_size, joint_num, 3] :param pose:[batch_size, joint_num, 3] :param parent: :return:
utils/ang2joint.py:4
Function
define_actions
Define the list of actions we are using. Args action: String with the passed action. Could be "all" Returns actions: List of
utils/data_utils.py:279
Function
define_actions_cmu
Define the list of actions we are using. Args action: String with the passed action. Could be "all" Returns actions: List of
utils/data_utils.py:310
Function
expmap2quat_torch
Converts expmap to quaternion batch pytorch version ported from the corresponding numpy method above :param R: N*3 :return: N*4
utils/data_utils.py:667
Function
expmap2rotmat
Converts an exponential map angle to a rotation matrix Matlab port to python for evaluation purposes I believe this is also called Rodrig
utils/data_utils.py:77
Function
expmap2rotmat_torch
Converts expmap matrix to rotation batch pytorch version ported from the corresponding method above :param r: N*3 :return: N*3*3
utils/data_utils.py:683
Function
expmap2xyz_torch
convert expmaps to joint locations :param expmap: N*99 :return: N*32*3
utils/data_utils.py:705
Function
find_indices_n
Adapted from https://github.com/una-dinosauria/human-motion-prediction/blob/master/src/seq2seq_model.py#L478 which originaly from In ord
utils/data_utils.py:732
Function
find_indices_srnn
Adapted from https://github.com/una-dinosauria/human-motion-prediction/blob/master/src/seq2seq_model.py#L478 which originaly from In ord
utils/data_utils.py:766
Method
forward
(self, input)
model/BaseModel.py:41
Method
forward
(self, x)
model/BaseModel.py:86
Method
forward
(self, x)
model/BaseModel.py:136
Method
forward
(self, x)
model/BaseModel.py:183
Method
forward
(self, x)
model/BaseModel.py:231
Method
forward
(self, src, input_n=10, output_n=10, itera=1)
model/stage_4.py:86
Function
get_dct_matrix
(N)
utils/data_utils.py:720
Function
get_dct_matrix
(N)
utils/util.py:58
Function
load_data_cmu
(path_to_dataset, actions, input_n, output_n, data_std=0, data_mean=0, is_test=False)
utils/data_utils.py:333
Function
load_data_cmu_3d_8
(opt, path_to_dataset, actions, input_n, output_n, data_std=0, data_mean=0, is_test=False)
utils/data_utils.py:397
Function
load_data_cmu_3d_all
(opt, path_to_dataset, actions, input_n, output_n, is_test=False)
utils/data_utils.py:541
Function
load_data_cmu_3d_n
(opt, path_to_dataset, actions, input_n, output_n, is_test=False)
utils/data_utils.py:468
Function
lr_decay_mine
(optimizer, lr_now, gamma)
utils/util.py:5
Function
normalization_stats
Also borrowed for SRNN code. Computes mean, stdev and dimensions to ignore. https://github.com/asheshjain399/RNNexp/blob/srnn/structural_rnn/
utils/data_utils.py:252
Function
normalize_data
Normalize input data by removing unused dimensions, subtracting the mean and dividing by the standard deviation Args data: nx99 ma
utils/data_utils.py:218
Function
opt_cam
:param x: N K 3 or N K 2 :param x_target: N K 3 or N K 2 :return:
utils/util.py:25
Function
orth_project
:param cam: b*[s,tx,ty] :param pts: b*k*3 :return:
utils/util.py:12
Function
pack
Append zero tensors of shape [4, 3] to a batch of [4, 1] shape tensor. Parameter: ---------- x: A tensor of shape [batch_size, 4, 1]
utils/ang2joint.py:108
Function
revert_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
utils/data_utils.py:167
Function
rotmat2euler
Converts a rotation matrix to Euler angles Matlab port to python for evaluation purposes https://github.com/asheshjain399/RNNexp/blob/srn
utils/data_utils.py:11
Function
rotmat2euler_torch
Converts a rotation matrix to euler angles batch pytorch version ported from the corresponding numpy method above :param R:N*3*3 :re
utils/data_utils.py:595
Function
rotmat2expmap
(R)
utils/data_utils.py:73
Function
rotmat2quat_torch
Converts a rotation matrix to quaternion batch pytorch version ported from the corresponding numpy method above :param R: N * 3 * 3 :
utils/data_utils.py:640
Function
save_ckpt
(state, is_best=True, file_name=['ckpt_best.pth.tar', 'ckpt_last.pth.tar'], opt=None)
utils/log.py:39
Function
save_csv_eval_log
(opt, head, value, is_create=False, file_name='test')
utils/log.py:24
Function
save_csv_log
(opt, head, value, is_create=False, file_name='test')
utils/log.py:12
Function
save_options
(opt)
utils/log.py:47