MCPcopy Create free account

hub / github.com/DavidBoja/pose-independent-anthropometry / functions

Functions140 in github.com/DavidBoja/pose-independent-anthropometry

↓ 36 callersMethodcuda
(self)
scripts/add_shape_to_dataset.py:81
↓ 19 callersFunctioncreate_body_model
Create SMPL body model :param: body_model_path (str): location to SMPL .pkl models :param body_model_type (str): smpl, smplx :param:
utils.py:778
↓ 12 callersMethodupdate
:param name: meter name :param val: new value to add :return: void
meter.py:97
↓ 11 callersFunctionload_config
(path="configs/config.yaml")
utils.py:835
↓ 8 callersFunctionpairwise_dist
:param xyz1 torch tensor (B,N,1) :param xyz1 torch tensor (B,M,1) :return dist torch tensor (B,M,N)
utils.py:1624
↓ 5 callersFunctionpelvis_normalization
Normalize scan on pelvis. The steps are: 1. center body on centroid of triangle waist landmark, rt asis landmark and lt asis landmark 2.
utils.py:1425
↓ 4 callersFunctionprocess_caesar_landmarks
Process landmarks from .lnd file - reading file from AUX to END flags. :param landmark_path (str): path to landmark .lnd file :param scal
utils.py:641
↓ 4 callersMethodstack_numpy_array
(A, B)
meter.py:114
↓ 3 callersFunctioncreate_body_model
Create SMPL body model :param: smpl_path (str): location to SMPL .pkl models :param: gender (str): male, female or neutral :param: nu
scripts/evaluate_baseline.py:48
↓ 3 callersMethodget_gender
(self, subj_name)
utils.py:1218
↓ 3 callersFunctionload_landmarks
Load landmarks from file and return the landmarks as torch tensor. Landmark file is defined in the following format: .txt extension
utils.py:912
↓ 3 callersMethodprint_stats
(self)
evaluate.py:347
↓ 3 callersMethodsave_stats
(self)
evaluate.py:430
↓ 3 callersMethodtrack_loss
(self, pred_m, gt_m=None, name=None, gender=None)
evaluate.py:333
↓ 2 callersMethod__init__
(self, init_lr=0.001, decrease_every_k_epochs=5, decrease_by=2, **kwargs)
utils.py:1232
↓ 2 callersMethodbuild_network
(self)
evaluate.py:149
↓ 2 callersMethoddeform_verts
(self, pose: torch.tensor, betas: torch.tensor,
scripts/add_shape_to_dataset.py:94
↓ 2 callersFunctionfit_landmarks
(subj_landmarks, init_trans=None, init_pose=None, init_beta=None)
scripts/add_shape_to_dataset.py:128
↓ 2 callersMethodforward
(self, pose, betas)
losses.py:139
↓ 2 callersMethodforward
(self)
scripts/add_shape_to_dataset.py:33
↓ 2 callersFunctionget_average_landmark
(vertices, landmark_indices, landmark_name)
utils.py:564
↓ 2 callersFunctionload_scan
Load 3D scan. Accepted formats .ply, .ply.gz :param scan_path (str): path to the scan to load
utils.py:1638
↓ 2 callersFunctionrotation_matrix_from_vectors_torch
Find the rotation matrix that aligns vec1 to vec2 :param vec1: A 3d "source" vector :param vec2: A 3d "destination" vector :return mat: A
utils.py:1407
↓ 2 callersFunctionset_body_pose_aa
Set the pose for the joint_name as set_pose in the pose parameters current_pose :param current_pose: (torch.tensor) dim (B,72) :par
utils.py:1305
↓ 2 callersFunctionviz_scatter
Visualize point cloud with plotly
visualization.py:80
↓ 1 callersFunctionRRE
R_gt: numpy array dim (3,3) R_estim: np array dim (3,3) Returns: angle measurement in degrees
dataset.py:1562
↓ 1 callersFunctionRRE_batch_torch
R_batch_gt: torch tensor dim (N,3,3) R_estim: torch tensor dim (3,3) Returns: angle measurement in de
dataset.py:1591
↓ 1 callersMethod__init__
(self, opt)
trainers.py:19
↓ 1 callersMethod__init__
(self, curves=[])
meter.py:52
↓ 1 callersMethod__init__
(self, gender, bm_path="../data/body_models/smpl")
scripts/add_shape_to_dataset.py:38
↓ 1 callersFunctionannotate_4DHuman_landmarks
Annotate 4DHumanOutfit landmarks for all subjects and actions in fit_paths. The landmarks are saved in fit_paths :param scan_paths (str)
annotate/annotate_4DHumanOutfit_landmarks.py:34
↓ 1 callersFunctionannotate_caesar_landmarks
Annotate the caesar dataset using the landmarks defined on the fitted body model. :param caesar_path (str): path to the CAESAR dataset
annotate/annotate_CAESAR_landmarks.py:16
↓ 1 callersMethodbuild_dataset_test
Create testing dataset
evaluate.py:179
↓ 1 callersMethodbuild_dataset_train
(self)
trainers.py:247
↓ 1 callersMethodbuild_dataset_val
Create validation dataset
trainers.py:264
↓ 1 callersMethodbuild_loss_tracker
(self)
evaluate.py:320
↓ 1 callersMethodbuild_losses
(self)
trainers.py:283
↓ 1 callersMethodbuild_optimizer
Create optimizer
trainers.py:84
↓ 1 callersMethodcontinue_experiment
(self, log_path)
meter.py:61
↓ 1 callersMethodcreate_results_dir
Get paths to save and reload networks :return:
trainers.py:29
↓ 1 callersFunctioncreate_results_directory
Save results in save_path as YYYY_MM_DD_HH_MM_SS folder. If continue_run is folder of type YYYY_MM_DD_HH_MM_SS, then save results in {sav
utils.py:802
↓ 1 callersMethoddump_stats
Save stats at each epoch
trainers.py:112
↓ 1 callersMethodend_epoch
Add meters average in average list and keep in current_epoch the current statistics :return:
meter.py:76
↓ 1 callersMethodevaluate
(self)
evaluate.py:163
↓ 1 callersMethodevaluate_4DHumanOutfit
(self)
evaluate.py:284
↓ 1 callersMethodevaluate_NoisyCaesar
(self)
evaluate.py:231
↓ 1 callersMethodevaluate_npz
(self)
evaluate.py:185
↓ 1 callersFunctionfind_new_cluster_centers
(clustering_data,clustering_model,single_cluster_shape,**kwargs)
dataset.py:1539
↓ 1 callersFunctionget_moyo_poses
Load SMPL poses from the MOYO dataset :param data_path (str): path to the MOYO dataset :param sample_every_kth_pose (int): sample every
utils.py:1778
↓ 1 callersFunctionget_normalizing_landmark
Find index of normalizing landmark
utils.py:623
↓ 1 callersFunctionget_subj
(bmi_level, measurements_df)
scripts/find_landmark_measurements_ambiguity.py:40
↓ 1 callersMethodincrement_epoch
(self)
trainers.py:126
↓ 1 callersMethodinit_meters
(self)
trainers.py:47
↓ 1 callersMethodlearning_rate_scheduler
(self)
trainers.py:71
↓ 1 callersMethodlog_likelihood
Create graph operation for negative log-likelihood calculation
losses.py:110
↓ 1 callersMethodmerged_log_likelihood
(self, pose, betas)
losses.py:93
↓ 1 callersFunctionmove_points_along_mesh
For each point to move from points_to_move, find a random plane and move the point along the cross section of the mesh (defined with mesh_ve
utils.py:1813
↓ 1 callersFunctionparse_landmark_txt_coords_formatting
Parse landamrk txt file where each line is formatted as: x_coord y_coord z_coord landmark_name :param data (List[str]) list of strings,
utils.py:842
↓ 1 callersFunctionparse_landmark_txt_index_formatting
Parse landamrk txt file with formatting landmark_index landmark_name :param data (List[str]) list of strings, each string
utils.py:878
↓ 1 callersMethodprocess_opt
(self, opt)
trainers.py:141
↓ 1 callersMethodprocess_opt
(self, opt)
evaluate.py:35
↓ 1 callersFunctionpts2homo
input pts: np array dim N x 3 return pts: np array dim N x 4
utils.py:1897
↓ 1 callersFunctionpts2homo_torch
input pts: np array dim N x 3 return pts: np array dim N x 4
utils.py:1904
↓ 1 callersFunctionrepose_caesar
Repose the unposed CAESAR scan by reposing the underlying fitted SMPL body model. Each point of the scan is reposed in the same way as its co
utils.py:1681
↓ 1 callersMethodreset
(self)
meter.py:40
↓ 1 callersMethodreset
Reset all meters :return:
meter.py:89
↓ 1 callersMethodsave_network
(self)
trainers.py:55
↓ 1 callersMethodset_epoch
(self)
trainers.py:77
↓ 1 callersFunctionset_seed
(sd)
utils.py:764
↓ 1 callersMethodstart_visdom
(self)
trainers.py:24
↓ 1 callersMethodtrain_epoch
(self)
trainers.py:286
↓ 1 callersFunctionunpose_caesar
Unpose CAESAR scan by unposing the underlying fitted SMPL body model. Each point of the scan is unposed in the same way as its corresponding
utils.py:1509
↓ 1 callersMethodupdate_curves
(self, vis)
meter.py:121
↓ 1 callersMethodval_epoch
(self)
trainers.py:334
↓ 1 callersMethodvisualize_stats
(self)
evaluate.py:437
↓ 1 callersFunctionweights_init
Initialize weights of the network according to the given strategy :param network: torch model to initialize weights :param strategy_name
utils.py:722
MethodN_verts
(self)
scripts/add_shape_to_dataset.py:54
FunctionNoisyCaesar_stats
(dataset_path: str, use_landmarks: str = "SMPL_INDEX_LANDAMRKS_REVISED",
compute_stats.py:11
FunctionRRE_batch
R_batch_gt: numpy array dim (N,3,3) R_estim: numpy array dim (3,3) Returns: angle measurement in degr
dataset.py:1576
Method__call__
(self, pose, betas, **kwargs)
scripts/add_shape_to_dataset.py:84
Method__getitem__
:return (dict): dictionary with keys: "name": name of scan "vertices": (N,3) np.array "faces": (N,3) np.a
dataset.py:118
Method__getitem__
:return (dict): dictionary with keys:
dataset.py:293
Method__getitem__
(self, index)
dataset.py:414
Method__getitem__
(self, index)
dataset.py:499
Method__getitem__
:return (dict): dictionary with keys: "name": name of scan "vertices": (N,3) np.array "faces": (N,3) np.a
dataset.py:652
Method__getitem__
(self, index)
dataset.py:1142
Method__init__
(self, opt)
trainers.py:130
Method__init__
(self)
meter.py:13
Method__init__
(self, **kwargs)
utils.py:1021
Method__init__
(self, gender_mapper_path: str="/pose-independent-anthropometry/data/gender/CAESAR_GENDER_MAP
utils.py:1213
Method__init__
(self, init_lr=0.001, **kwargs)
utils.py:1258
Method__init__
(self, encoder_input_dim=368, output_dim=11, hidden_dim1=194, hidden_dim2=97)
models.py:7
Method__init__
:param dataset_path: (str) path to DYNA dataset :param subject_id: (str) subject id :param subject_action: (str) action perfo
dataset.py:45
Method__init__
:param dataset_path: (str) path to 4DHumanOutfit scans :param parameters_path: (str) path to the directory where the already
dataset.py:172
Method__init__
Dataset defined with a .npz file or folder with .npz files The .npz files are data points of subjects with landmarks and measurements
dataset.py:361
Method__init__
Extension of the NPZ dataset that allows for augmentation of the landmarks :param dataset_path: (str) path to the .npz file or folde
dataset.py:421
Method__init__
cfg: config dictionary with :param data_dir (str): path to caesar dataset :param load_countries (str or list): countries to l
dataset.py:561
Method__init__
:param caesar_dir: (str) path to caesar dataset :param fitted_bm_dir: (str) path to SMPL param fits to CAESAR dataset :param
dataset.py:771
Method__init__
(self, port, env)
visualization.py:13
Method__init__
(self, prior_folder='prior', num_gaussians=6, dtype=torch.float32, epsilon=1e-16,
losses.py:14
next →1–100 of 140, ranked by callers