Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/DavidBoja/pose-independent-anthropometry
/ functions
Functions
140 in github.com/DavidBoja/pose-independent-anthropometry
⨍
Functions
140
◇
Types & classes
21
↓ 36 callers
Method
cuda
(self)
scripts/add_shape_to_dataset.py:81
↓ 19 callers
Function
create_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 callers
Method
update
:param name: meter name :param val: new value to add :return: void
meter.py:97
↓ 11 callers
Function
load_config
(path="configs/config.yaml")
utils.py:835
↓ 8 callers
Function
pairwise_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 callers
Function
pelvis_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 callers
Function
process_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 callers
Method
stack_numpy_array
(A, B)
meter.py:114
↓ 3 callers
Function
create_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 callers
Method
get_gender
(self, subj_name)
utils.py:1218
↓ 3 callers
Function
load_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 callers
Method
print_stats
(self)
evaluate.py:347
↓ 3 callers
Method
save_stats
(self)
evaluate.py:430
↓ 3 callers
Method
track_loss
(self, pred_m, gt_m=None, name=None, gender=None)
evaluate.py:333
↓ 2 callers
Method
__init__
(self, init_lr=0.001, decrease_every_k_epochs=5, decrease_by=2, **kwargs)
utils.py:1232
↓ 2 callers
Method
build_network
(self)
evaluate.py:149
↓ 2 callers
Method
deform_verts
(self, pose: torch.tensor, betas: torch.tensor,
scripts/add_shape_to_dataset.py:94
↓ 2 callers
Function
fit_landmarks
(subj_landmarks, init_trans=None, init_pose=None, init_beta=None)
scripts/add_shape_to_dataset.py:128
↓ 2 callers
Method
forward
(self, pose, betas)
losses.py:139
↓ 2 callers
Method
forward
(self)
scripts/add_shape_to_dataset.py:33
↓ 2 callers
Function
get_average_landmark
(vertices, landmark_indices, landmark_name)
utils.py:564
↓ 2 callers
Function
load_scan
Load 3D scan. Accepted formats .ply, .ply.gz :param scan_path (str): path to the scan to load
utils.py:1638
↓ 2 callers
Function
rotation_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 callers
Function
set_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 callers
Function
viz_scatter
Visualize point cloud with plotly
visualization.py:80
↓ 1 callers
Function
RRE
R_gt: numpy array dim (3,3) R_estim: np array dim (3,3) Returns: angle measurement in degrees
dataset.py:1562
↓ 1 callers
Function
RRE_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 callers
Method
__init__
(self, opt)
trainers.py:19
↓ 1 callers
Method
__init__
(self, curves=[])
meter.py:52
↓ 1 callers
Method
__init__
(self, gender, bm_path="../data/body_models/smpl")
scripts/add_shape_to_dataset.py:38
↓ 1 callers
Function
annotate_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 callers
Function
annotate_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 callers
Method
build_dataset_test
Create testing dataset
evaluate.py:179
↓ 1 callers
Method
build_dataset_train
(self)
trainers.py:247
↓ 1 callers
Method
build_dataset_val
Create validation dataset
trainers.py:264
↓ 1 callers
Method
build_loss_tracker
(self)
evaluate.py:320
↓ 1 callers
Method
build_losses
(self)
trainers.py:283
↓ 1 callers
Method
build_optimizer
Create optimizer
trainers.py:84
↓ 1 callers
Method
continue_experiment
(self, log_path)
meter.py:61
↓ 1 callers
Method
create_results_dir
Get paths to save and reload networks :return:
trainers.py:29
↓ 1 callers
Function
create_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 callers
Method
dump_stats
Save stats at each epoch
trainers.py:112
↓ 1 callers
Method
end_epoch
Add meters average in average list and keep in current_epoch the current statistics :return:
meter.py:76
↓ 1 callers
Method
evaluate
(self)
evaluate.py:163
↓ 1 callers
Method
evaluate_4DHumanOutfit
(self)
evaluate.py:284
↓ 1 callers
Method
evaluate_NoisyCaesar
(self)
evaluate.py:231
↓ 1 callers
Method
evaluate_npz
(self)
evaluate.py:185
↓ 1 callers
Function
find_new_cluster_centers
(clustering_data,clustering_model,single_cluster_shape,**kwargs)
dataset.py:1539
↓ 1 callers
Function
get_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 callers
Function
get_normalizing_landmark
Find index of normalizing landmark
utils.py:623
↓ 1 callers
Function
get_subj
(bmi_level, measurements_df)
scripts/find_landmark_measurements_ambiguity.py:40
↓ 1 callers
Method
increment_epoch
(self)
trainers.py:126
↓ 1 callers
Method
init_meters
(self)
trainers.py:47
↓ 1 callers
Method
learning_rate_scheduler
(self)
trainers.py:71
↓ 1 callers
Method
log_likelihood
Create graph operation for negative log-likelihood calculation
losses.py:110
↓ 1 callers
Method
merged_log_likelihood
(self, pose, betas)
losses.py:93
↓ 1 callers
Function
move_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 callers
Function
parse_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 callers
Function
parse_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 callers
Method
process_opt
(self, opt)
trainers.py:141
↓ 1 callers
Method
process_opt
(self, opt)
evaluate.py:35
↓ 1 callers
Function
pts2homo
input pts: np array dim N x 3 return pts: np array dim N x 4
utils.py:1897
↓ 1 callers
Function
pts2homo_torch
input pts: np array dim N x 3 return pts: np array dim N x 4
utils.py:1904
↓ 1 callers
Function
repose_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 callers
Method
reset
(self)
meter.py:40
↓ 1 callers
Method
reset
Reset all meters :return:
meter.py:89
↓ 1 callers
Method
save_network
(self)
trainers.py:55
↓ 1 callers
Method
set_epoch
(self)
trainers.py:77
↓ 1 callers
Function
set_seed
(sd)
utils.py:764
↓ 1 callers
Method
start_visdom
(self)
trainers.py:24
↓ 1 callers
Method
train_epoch
(self)
trainers.py:286
↓ 1 callers
Function
unpose_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 callers
Method
update_curves
(self, vis)
meter.py:121
↓ 1 callers
Method
val_epoch
(self)
trainers.py:334
↓ 1 callers
Method
visualize_stats
(self)
evaluate.py:437
↓ 1 callers
Function
weights_init
Initialize weights of the network according to the given strategy :param network: torch model to initialize weights :param strategy_name
utils.py:722
Method
N_verts
(self)
scripts/add_shape_to_dataset.py:54
Function
NoisyCaesar_stats
(dataset_path: str, use_landmarks: str = "SMPL_INDEX_LANDAMRKS_REVISED",
compute_stats.py:11
Function
RRE_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