MCPcopy Create free account

hub / github.com/KyujinHan/Smplify-X-Perfect-Implementation / functions

Functions77 in github.com/KyujinHan/Smplify-X-Perfect-Implementation

⨍ Functions 77◇ Types & classes 13
↓ 7 callersFunctioncreate_prior
(prior_type, **kwargs)
smplifyx/prior.py:36
↓ 3 callersMethod_add_grad
(self, step_size, update)
smplifyx/optimizers/lbfgs_ls.py:233
↓ 3 callersMethod_gather_flat_grad
(self)
smplifyx/optimizers/lbfgs_ls.py:221
↓ 3 callersMethodupdate_mesh
(self, vertices, faces)
smplifyx/mesh_viewer.py:82
↓ 2 callersMethod__init__
(self, dtype=DEFAULT_DTYPE, reduction='sum', **kwargs)
smplifyx/prior.py:93
↓ 2 callersMethod__init__
(self, search_tree=None, pen_distance=None, tri_filtering_module=None, rho=1
smplifyx/fitting.py:292
↓ 2 callersFunction_cubic_interpolate
(x1, f1, g1, x2, f2, g2, bounds=None)
smplifyx/optimizers/lbfgs_ls.py:11
↓ 2 callersMethodclose_viewer
(self)
smplifyx/mesh_viewer.py:63
↓ 2 callersMethodcreate_fitting_closure
(self, optimizer, body_model, camera=None, gt_jo
smplifyx/fitting.py:225
↓ 2 callersFunctionparse_config
(argv=None)
smplifyx/cmd_parser.py:27
↓ 2 callersMethodread_item
(self, img_path)
smplifyx/data_parser.py:176
↓ 2 callersMethodreset_loss_weights
(self, loss_weight_dict)
smplifyx/fitting.py:362
↓ 2 callersMethodrun_fitting
Helper function for running an optimization process Parameters ---------- optimizer: torch.optim.Optimizer
smplifyx/fitting.py:151
↓ 1 callersMethod__init__
(self, rho=1)
smplifyx/utils.py:61
↓ 1 callersMethod_clone_param
(self)
smplifyx/optimizers/lbfgs_ls.py:242
↓ 1 callersMethod_directional_evaluate
(self, closure, x, t, d)
smplifyx/optimizers/lbfgs_ls.py:249
↓ 1 callersMethod_numel
(self)
smplifyx/optimizers/lbfgs_ls.py:216
↓ 1 callersMethod_set_param
(self, params_data)
smplifyx/optimizers/lbfgs_ls.py:245
↓ 1 callersFunction_strong_Wolfe
(obj_func, x, t, d, f, g, gtd, c1=1e-4, c2=0.9, tolerance_change=1e-9, max_iter=20,
smplifyx/optimizers/lbfgs_ls.py:39
↓ 1 callersFunctioncreate_camera
(camera_type='persp', **kwargs)
smplifyx/camera.py:35
↓ 1 callersFunctioncreate_dataset
(dataset='openpose', data_folder='data', **kwargs)
smplifyx/data_parser.py:45
↓ 1 callersMethodcreate_mesh
(self, vertices, faces, color=(0.3, 0.3, 0.3, 1.0), wireframe=False)
smplifyx/mesh_viewer.py:67
↓ 1 callersFunctionfit_single_frame
(img, keypoints, body_model, camera,
smplifyx/fit_single_frame.py:51
↓ 1 callersMethodget_joint_weights
(self)
smplifyx/data_parser.py:155
↓ 1 callersMethodget_mean
Returns the mean of the mixture
smplifyx/prior.py:176
↓ 1 callersMethodget_mesh_fn
(self)
smplifyx/render_results.py:53
↓ 1 callersMethodget_model2data
(self)
smplifyx/data_parser.py:143
↓ 1 callersMethodis_active
(self)
smplifyx/mesh_viewer.py:60
↓ 1 callersMethodlog_likelihood
Create graph operation for negative log-likelihood calculation
smplifyx/prior.py:198
↓ 1 callersFunctionmain
(**args)
smplifyx/main.py:44
↓ 1 callersMethodmerged_log_likelihood
(self, pose, betas)
smplifyx/prior.py:181
↓ 1 callersMethodnext
(self)
smplifyx/data_parser.py:208
↓ 1 callersFunctionread_keypoints
(keypoint_fn, use_hands=True, use_face=True, use_face_contour=False)
smplifyx/data_parser.py:52
↓ 1 callersFunctionrender_mesh
(mesh_trimesh, camera_center, camera_transl, focal_length, img_width, img_height)
smplifyx/fit_single_frame.py:561
↓ 1 callersFunctionsmpl_to_openpose
Returns the indices of the permutation that maps OpenPose to SMPL Parameters ---------- model_type: str, optional
smplifyx/utils.py:74
↓ 1 callersMethodstep
Performs a single optimization step. Arguments: closure (callable): A closure that reevaluates the model and retur
smplifyx/optimizers/lbfgs_ls.py:256
Method__enter__
(self)
smplifyx/fitting.py:134
Method__exit__
(self, exception_type, exception_value, traceback)
smplifyx/fitting.py:140
Method__getitem__
(self, idx)
smplifyx/data_parser.py:172
Method__init__
(self, width=1200, height=800, body_color=(1.0, 1.0, 0.9, 1.0), registered_k
smplifyx/mesh_viewer.py:28
Method__init__
(self, joint_maps=None)
smplifyx/utils.py:45
Method__init__
(self, data_folder, img_folder='images', keyp_folder='keypoints', use_hands=
smplifyx/data_parser.py:108
Method__init__
(self, rotation=None, translation=None, focal_length_x=None, focal_length_y=None,
smplifyx/camera.py:46
Method__init__
(self, mesh_fns, verbose=False)
smplifyx/render_results.py:33
Method__init__
(self, dtype=torch.float32, **kwargs)
smplifyx/prior.py:54
Method__init__
(self, prior_folder='prior', num_gaussians=6, dtype=DEFAULT_DTYPE, epsilon=1e-16,
smplifyx/prior.py:102
Method__init__
(self, summary_steps=1, visualize=False, maxiters=100, ftol=2e-09, gtol=1e-05,
smplifyx/fitting.py:118
Method__init__
(self, init_joints_idxs, trans_estimation=None, reduction='sum', data_weight
smplifyx/fitting.py:463
Method__init__
(self, params, lr=1, max_iter=20, max_eval=None, tolerance_grad=1e-5, tolerance_change=1e-9,
smplifyx/optimizers/lbfgs_ls.py:199
Method__iter__
(self)
smplifyx/data_parser.py:202
Method__len__
(self)
smplifyx/data_parser.py:169
Method__next__
(self)
smplifyx/data_parser.py:205
Functioncreate_loss
(loss_type='smplify', **kwargs)
smplifyx/fitting.py:281
Functioncreate_optimizer
Creates the optimizer
smplifyx/optimizers/optim_factory.py:27
Methodextra_repr
(self)
smplifyx/utils.py:65
Methodfitting_func
(backward=True)
smplifyx/fitting.py:238
Methodforward
(self, joints, **kwargs)
smplifyx/utils.py:53
Methodforward
(self, residual)
smplifyx/utils.py:68
Methodforward
(self, points)
smplifyx/camera.py:93
Methodforward
Returns the angle prior loss for the given pose Args: pose: (Bx[23 + 1] * 3) torch tensor with the axis-angle repres
smplifyx/prior.py:73
Methodforward
(self, module_input, *args)
smplifyx/prior.py:96
Methodforward
(self, pose, betas)
smplifyx/prior.py:227
Methodforward
(self, body_model_output, camera, gt_joints, joints_conf, body_model_faces, joint_weights,
smplifyx/fitting.py:374
Methodforward
(self, body_model_output, camera, gt_joints, **kwargs)
smplifyx/fitting.py:495
Methodget_left_shoulder
(self)
smplifyx/data_parser.py:149
Methodget_right_shoulder
(self)
smplifyx/data_parser.py:152
Functionguess_init
Initializes the camera translation vector Parameters ---------- model: nn.Module The PyTorch module of the body
smplifyx/fitting.py:37
Functionmax_grad_change
(grad_arr)
smplifyx/utils.py:40
Methodnext_mesh
(self, viewer)
smplifyx/render_results.py:39
Functionno_prior
(*args, **kwargs)
smplifyx/prior.py:45
Methodobj_func
(x, t, d)
smplifyx/optimizers/lbfgs_ls.py:391
Methodprev_mesh
(self, viewer)
smplifyx/render_results.py:46
Methodquit_viewer
(self, viewer)
smplifyx/render_results.py:56
Functionrel_change
(prev_val, curr_val)
smplifyx/utils.py:36
Methodreset_loss_weights
(self, loss_weight_dict)
smplifyx/fitting.py:486
Methodset_colors
(self, vertex_color)
smplifyx/fitting.py:144
Functionto_tensor
(tensor, dtype=torch.float32)
smplifyx/utils.py:29