MCPcopy Create free account

hub / github.com/bharat-b7/RVH_Mesh_Registration / functions

Functions181 in github.com/bharat-b7/RVH_Mesh_Registration

↓ 1 callersFunctionmain
(args)
smpl_registration/fit_SMPLH_IPNet.py:287
↓ 1 callersFunctionmake_list
(tensor)
lib/smpl/smplpytorch/smplpytorch/pytorch/tensutils.py:56
↓ 1 callersFunctionmean_hand_pose
mean hand pose computed from grab dataset
lib/smpl/priors/th_hand_prior.py:29
↓ 1 callersFunctionnp2tensor
(x, device=None)
lib/torch_functions.py:8
↓ 1 callersMethodoptimize_pose_only
(self, th_scan_meshes, smpl, iterations, steps_per_iter, th_pose_3d, prior_weight=N
smpl_registration/fit_SMPLH.py:117
↓ 1 callersMethodoptimize_pose_only
(self, th_scan_meshes, smpl, iterations, steps_per_iter, scan_parts, smpl_parts)
smpl_registration/fit_SMPLH_IPNet.py:123
↓ 1 callersMethodoptimize_pose_shape
(self, pclouds, smpl, iterations, steps_per_iter, joints_3d=None)
smpl_registration/fit_SMPLH_pcloud.py:70
↓ 1 callersMethodoptimize_pose_shape
(self, th_scan_meshes, smpl, iterations, steps_per_iter, th_pose_3d=None)
smpl_registration/fit_SMPLH.py:56
↓ 1 callersMethodoptimize_pose_shape
Optimize SMPL. :param display: if not None, pass index of the scan in th_scan_meshes to visualize.
smpl_registration/fit_SMPLH_IPNet.py:167
↓ 1 callersFunctionpc2vox
Convert PC to voxels for IPNet
smpl_registration/fit_SMPLH_IPNet.py:34
↓ 1 callersFunctionpreset_params
(args)
utils/keypoints_3d_estimation/predict_2d_pose.py:18
↓ 1 callersFunctionpreset_params
(args)
utils/keypoints_3d_estimation/02_predict_2d_pose.py:15
↓ 1 callersFunctionprocess
(scan, body, name, path)
utils/preprocess_scan.py:38
↓ 1 callersFunctionproject_points
project 3d points to 2d color image color_mat: (3, 4) kinect_pose: (3, 4) d2c: (3, 4)
lib/body_objectives.py:47
↓ 1 callersFunctionquat2mat
Convert quaternion coefficients to rotation matrix. Args: quat: size = [batch_size, 4] 4 <===>(w, x, y, z) Returns: Rotation m
lib/smpl/smplpytorch/smplpytorch/pytorch/rodrigues_layer.py:13
↓ 1 callersFunctionready_arguments
(fname_or_dict)
lib/serialization.py:49
↓ 1 callersMethodsave_outputs
(self, save_path, pc_files, smpl, pclouds, save_name='smpl')
smpl_registration/fit_SMPLH_pcloud.py:171
↓ 1 callersFunctionsubtract_flat_id
(rot_mats, hands=False)
lib/smpl/smplpytorch/smplpytorch/pytorch/tensutils.py:41
↓ 1 callersFunctionth_pack
(tensor)
lib/smpl/smplpytorch/smplpytorch/pytorch/tensutils.py:32
↓ 1 callersFunctionth_posemap_axisang
Converts axis-angle to rotmat pose_vectors (Tensor (batch_size x 72)): pose parameters in axis-angle representation
lib/smpl/smplpytorch/smplpytorch/pytorch/tensutils.py:6
↓ 1 callersFunctionto_rotation_matrix
(euler_angles)
lib/libmesh/implicit_waterproofing.py:21
↓ 1 callersFunctionverts_core
(pose, v, J, weights, kintree_table, want_Jtr=False, xp=chumpy)
lib/smpl/smplpytorch/smplpytorch/native/webuser/lbs.py:59
↓ 1 callersFunctionverts_core
(pose, v, J, weights, kintree_table, bs_style, want_Jtr=False, xp=chumpy)
lib/smpl/smplpytorch/smplpytorch/native/webuser/verts.py:91
↓ 1 callersMethodviz_fitting
(self, smpl, pclouds, ind=0, smpl_vc=np.array([0, 1, 0]))
smpl_registration/fit_SMPLH_pcloud.py:221
↓ 1 callersFunctionvoxelized_pointcloud_sampling
(mesh_path, name, out_path, res, num_points, bounds=(-1., 1.), ext='')
utils/voxelized_pointcloud_sampling.py:33
Method__call__
:param pose: Batch x pose_dims :return: weighted L2 distance of the N pose parameters, where N = 72 - prefix for SMPL model, for smpl
lib/smpl/priors/th_smpl_prior.py:28
Method__call__
full_pose also include body poses, this function can be used to compute loss
lib/smpl/priors/th_hand_prior.py:55
Method__getitem__
(self, pid)
lib/smpl/priors/th_smpl_prior.py:71
Method__init__
(self, model_root, device='cuda:0', save_name='smpl', debug=False, hands=True)
smpl_registration/base_fitter.py:22
Method__init__
(self, model_root, device='cuda:0', save_name='smpld', debug=False, hands=False)
smpl_registration/fit_SMPLHD.py:18
Method__init__
initialize model, data etc. Args: args: command line arguments
smpl_registration/fit_SMPLH_IPNet.py:43
Method__init__
(self, hidden_dim=256, num_parts=24, tex=False)
smpl_registration/models/ipnet_models.py:148
Method__init__
(self, hidden_dim=256, num_parts=24, tex=False)
smpl_registration/models/ipnet_models.py:272
Method__init__
(self, model, threshold, exp_name, checkpoint=None, device=torch.device("cuda"), resolution=16,
smpl_registration/models/generator.py:18
Method__init__
(self, mesh, resolution=512)
lib/libmesh/inside_mesh.py:13
Method__init__
(self, triangles, resolution=128)
lib/libmesh/inside_mesh.py:117
Method__init__
(self, model_root, batch_sz, betas=None, pose=None, trans=None, offsets=None
lib/smpl/wrapper_pytorch.py:19
Method__init__
(self, model_root, batch_sz, top_betas=None, other_betas=None,
lib/smpl/wrapper_pytorch.py:96
Method__init__
(self, model_root)
lib/smpl/joint_regressor.py:12
Method__init__
Args: center_idx: index of center joint in our computations, model_root: path to pkl files for the model
lib/smpl/smplpytorch/smplpytorch/pytorch/smpl_layer.py:20
Method__init__
(self, mean, prec, prefix, end=66, device="cuda:0")
lib/smpl/priors/th_smpl_prior.py:22
Method__init__
end=66 for smplh, 69 for smpl
lib/smpl/priors/th_smpl_prior.py:40
Method__init__
prefix is the index from where hand pose starts, 66 for SMPL-H
lib/smpl/priors/th_hand_prior.py:40
Functionbarycentric_coordinates
Calculate barycentric coordinates of the given point :param p: a given point :param q: triangle vertex :param u: triangle vertex
lib/geometry.py:48
Functionbatch_gather
:param arr: B x N x D :param ind: B x M :return: B x M x D
lib/torch_functions.py:28
Functionbatch_get_pose_obj
Comapre landmarks/keypoints ontained from the existing SMPL against those observed on the scan. Naive implementation as batching currently
lib/body_objectives.py:161
Functionbatch_reprojection_loss_kinect
Reprojection loss between 2D joints and SMPL joints projected using calibrated Kinect cameras. Parameters: img_bodyjoints: (B,
lib/body_objectives.py:64
Functionbatch_rodrigues
(axisang)
lib/smpl/smplpytorch/smplpytorch/pytorch/rodrigues_layer.py:41
Functionbuild_bmlmovi_prior
Args: data_path: path to amass registration, e.g. .../amass/BMLmovi out_path: dataset: name of the dataset used, for ou
utils/build_prior.py:22
Functionclosest_index
Given two point clouds, finds closest point id :param src_points: B x N x 3 :param tgt_points: B x M x 3 :return B x N
lib/torch_functions.py:16
Methodcompute_dr_wrt
(self, wrt)
lib/smpl/smplpytorch/smplpytorch/native/webuser/posemapper.py:27
Methodcompute_r
(self)
lib/smpl/smplpytorch/smplpytorch/native/webuser/posemapper.py:24
Methodcopy_smpl_params
(self, split_smpl, smpl)
smpl_registration/base_fitter.py:248
Functioncreate_grid_points_from_all_bounds
(res, x_min, x_max, y_min, y_max, z_min, z_max)
lib/libmesh/implicit_waterproofing.py:112
Functioncreate_grid_points_from_bounds_indexed
(minimun, maximum, res)
lib/libmesh/implicit_waterproofing.py:92
Methodfit
(self, scans, pose_files, gender='male', save_path=None)
smpl_registration/base_fitter.py:36
Methodforward
(self, p, x)
smpl_registration/models/ipnet_models.py:81
Methodforward
(self, p, x)
smpl_registration/models/ipnet_models.py:209
Methodforward
(self, p, x)
smpl_registration/models/ipnet_models.py:335
Methodforward
(self)
lib/smpl/wrapper_pytorch.py:64
Methodforward
Args: th_pose_axisang (Tensor (batch_size x 72)): pose parameters in axis-angle representation th_betas (Tensor (batch_size x
lib/smpl/smplpytorch/smplpytorch/pytorch/smpl_layer.py:80
Methodgenerate_mesh_all
Predict occupancy and create a mesh
smpl_registration/models/generator.py:69
Methodgenerate_meshs_all_parts
(self, data)
smpl_registration/models/generator.py:165
Functionget_hres
Get an upsampled version of the mesh. OUTPUT: - nv: new vertices - nf: faces of the upsampled - mapping: mapping from
lib/geometry.py:34
Methodget_landmarks
Computes body25 joints for SMPL along with hand and facial landmarks
lib/smpl/wrapper_pytorch.py:183
Methodget_loss_weights
Set loss weights
smpl_registration/base_fitter.py:141
Functionget_prior_weight
(no_right_hand_batch, no_left_hand_batch)
lib/body_objectives.py:143
Functionlaplacian
(v, f)
lib/geometry.py:9
Functionload_config
(config_path)
utils/keypoints_3d_estimation/io.py:14
Functionload_keypoints_2d
(keypoints_2d_file,device="cpu")
utils/keypoints_3d_estimation/io.py:72
Functionload_model
(fname_or_dict)
lib/serialization.py:84
Functionload_model
(fname_or_dict)
lib/smpl/smplpytorch/smplpytorch/native/webuser/serialization.py:88
Functionlrotmin
(p)
lib/smpl/smplpytorch/smplpytorch/native/webuser/posemapper.py:32
Methodoptimize_pose_only
Initially we want to only optimize the global rotation of SMPL. Next we optimize full pose. We optimize pose based on the 3D keypoint
smpl_registration/base_fitter.py:54
Methodoptimize_pose_shape
optimize smpl pose and shape parameters together Args: th_scan_meshes: smpl: iterations:
smpl_registration/base_fitter.py:39
Methodquery
(self, points)
lib/libmesh/inside_mesh.py:121
Methodsave_pclouds
(self, points, save_paths, colors=None)
smpl_registration/fit_SMPLH_pcloud.py:182
Functiontensor2np
(x)
lib/torch_functions.py:12
Functionth_get_axis_angle
(vector)
lib/smpl/smplpytorch/smplpytorch/pytorch/rodrigues_layer.py:55
Functionto_occ
(mesh, res)
lib/libmesh/implicit_waterproofing.py:127
Functionverts_decorated
(trans, pose, v_template, J, weights, kintree_table, bs_style, f, bs_t
lib/smpl/smplpytorch/smplpytorch/native/webuser/verts.py:31
← previous101–181 of 181, ranked by callers