Functions — github.com/HongwenZhang/PyMAF

CodeHub MCPcopy Create free account

hub / github.com/HongwenZhang/PyMAF / functions

Functions303 in github.com/HongwenZhang/PyMAF

⨍ Functions 303◇ Types & classes 36

datasets/inference.py:57

datasets/inference.py:92

datasets/mixed_dataset.py:46

Method__setitem__

Update dictionary entries

core/fits_dict.py:77

datasets/coco_keypoint_dataset.py:125

datasets/JointsDataset.py:34

Method_get_deconv_cfg

(deconv_kernel, index)

models/hmr.py:109

Method_get_deconv_cfg

(deconv_kernel, index)

models/pymaf_net.py:225

Method_make_deconv_layer

(self, num_layers, num_filters, num_kernels)

models/hmr.py:103

Method_make_layer

(self, block, planes, blocks, stride=1)

models/pymaf_net.py:197

Method_make_layer

(self, block, planes, blocks, stride=1)

models/iuv_predictor.py:54

utils/saver.py:99

Functionaxis_angle_add

Composition of two axis-angle rotations (PyTorch version) Args: theta: N x 3 roll_axis: N x 3 alph: N x 1 Returns:

utils/pose_utils.py:79

Functionaxis_angle_add_np

Composition of two axis-angle rotations (NumPy version) Args: theta: N x 3 roll_axis: N x 3 alph: N x 1 Returns:

utils/pose_utils.py:114

Functionconvert_kps

(joints2d, src, dst)

datasets/data_utils/kp_utils.py:28

Functioncrop_image

(image, kp_2d, center_x, center_y, width, height, patch_width, patch_height, do_augment)

datasets/data_utils/img_utils.py:103

Functiondownload_ckpt

(outdir='data/vibe_data', use_3dpw=False)

utils/demo_utils.py:131

Functiondraw_skeleton

(image, kp_2d, dataset='common', unnormalize=True, thickness=2)

utils/vis.py:543

core/base_trainer.py:89

(self, cfg, preds, output_dir, *args, **kwargs)

datasets/JointsDataset.py:37

core/base_trainer.py:92

models/hmr.py:34

models/hmr.py:142

(self, x, init_pose=None, init_shape=None, init_cam=None, n_iter=3)

models/hmr.py:228

models/pose_resnet.py:43

models/pose_resnet.py:80

models/pose_resnet.py:200

Returns mesh-aligned features for the 3D mesh points. Args: p (tensor): [B, N_m, 3] mesh vertices s_feat (tensor): [

models/maf_extractor.py:124

(self, *args, **kwargs)

models/smpl.py:101

(self, *args, **kwargs)

models/smpl.py:380

(self, *args, **kwargs)

models/smpl.py:532

(self, *args, **kwargs)

models/smpl.py:564

(self, x, init_pose=None, init_shape=None, init_cam=None, n_iter=1, J_regressor=None)

models/pymaf_net.py:48

(self, x, J_regressor=None)

models/pymaf_net.py:258

models/iuv_predictor.py:71

Functiongenerate_heatmap

param joints: [num_joints, 3] param joints_vis: [num_joints, 3] return: target, target_weight(1: visible, 0: invisible)

utils/imutils.py:178

Functionget_3dpw_joint_names

datasets/data_utils/kp_utils.py:586

Functionget_3dpw_skeleton

datasets/data_utils/kp_utils.py:604

Functionget_aich_joint_names

datasets/data_utils/kp_utils.py:535

Functionget_aich_skeleton

datasets/data_utils/kp_utils.py:553

Functionget_bbox_from_kp2d

datasets/data_utils/img_utils.py:287

Functionget_coco_joint_names

datasets/data_utils/kp_utils.py:415

Functionget_coco_skeleton

datasets/data_utils/kp_utils.py:436

Functionget_common_joint_names

datasets/data_utils/kp_utils.py:376

Functionget_common_skeleton

datasets/data_utils/kp_utils.py:394

Methodget_global_rotation

Forward pass for the SMPLX model Parameters ---------- global_orient: torch.tensor, optional, shape Bx3x

models/smpl.py:146

Functionget_h36m_joint_names

datasets/data_utils/kp_utils.py:265

Functionget_image

datasets/data_utils/img_utils.py:26

Functionget_image_crops

(image_file, bboxes)

datasets/data_utils/img_utils.py:155

Functionget_insta_joint_names

datasets/data_utils/kp_utils.py:99

Functionget_insta_skeleton

datasets/data_utils/kp_utils.py:128

Functionget_mpii3d_joint_names

datasets/data_utils/kp_utils.py:67

Functionget_mpii3d_test_joint_names

datasets/data_utils/kp_utils.py:46

Functionget_mpii_joint_names

datasets/data_utils/kp_utils.py:478

Functionget_mpii_skeleton

datasets/data_utils/kp_utils.py:498

Functionget_part_joints

models/smpl.py:695

Functionget_pennaction_joint_names

datasets/data_utils/kp_utils.py:359

Functionget_perm_idxs

datasets/data_utils/kp_utils.py:40

Functionget_posetrack_joint_names

datasets/data_utils/kp_utils.py:317

Functionget_posetrack_original_kp_names

datasets/data_utils/kp_utils.py:338

Methodget_projection_matrix

(self, width=None, height=None)

utils/renderer.py:67

Functionget_single_image_crop

(image, bbox, scale=1.3)

datasets/data_utils/img_utils.py:180

Functionget_smooth_bbox_params

Computes smooth bounding box parameters from keypoints: 1. Computes bbox by rescaling the person to be around 150 px. 2. Linearly int

utils/smooth_bbox.py:9

Functionget_smpl_joint_names

datasets/data_utils/kp_utils.py:671

Functionget_smpl_skeleton

datasets/data_utils/kp_utils.py:699

Functionget_smplcoco_joint_names

datasets/data_utils/kp_utils.py:624

Functionget_smplcoco_skeleton

datasets/data_utils/kp_utils.py:647

Functionget_spin_joint_names

datasets/data_utils/kp_utils.py:212

Functionget_spin_skeleton

datasets/data_utils/kp_utils.py:286

Functionget_staf_joint_names

datasets/data_utils/kp_utils.py:187

Functionget_staf_skeleton

datasets/data_utils/kp_utils.py:159

Methodget_symmetric_densepose

(self, I, U, V, x, y, Mask)

utils/densepose_methods.py:30

Methodget_tpose

(self, *args, **kwargs)

models/smpl.py:597

Methodget_vaild_state

(self, dataset_name, ind)

core/fits_dict.py:69

core/trainer.py:40

Functionkeypoint_hflip

(kp, img_width)

datasets/data_utils/kp_utils.py:19

Methodmove_dict_to_device

(self, dict, device, tensor2float=False)

core/base_trainer.py:59

Methodnatural_keys

alist.sort(key=natural_keys) sorts in human order http://nedbatchelder.com/blog/200712/human_sorting.html (See To

utils/saver.py:106

Functionnormalize_2d_kp

(kp_2d, crop_size=224, inv=False)

datasets/data_utils/img_utils.py:307

Functionpreprocess_video

Read video, do normalize and crop it according to the bounding box. If there are bounding box annotations, use them to crop the image. If

utils/demo_utils.py:29

Functionread_image

datasets/data_utils/img_utils.py:244

Methodreset_res

(self, resolution)

utils/renderer.py:281

FunctionrotateX

Rotate all points in a 2D array around the x axis.

utils/renderer.py:413

FunctionrotateZ

Rotate all points in a 2D array around the z axis.

utils/renderer.py:422

Functionrotmat_to_rot6d

Convert 3x3 rotation matrix to 6D rotation representation. Based on Zhou et al., "On the Continuity of Rotation Representations in Neural Networks

utils/geometry.py:228

Methodselect_data

datasets/JointsDataset.py:43

Functionsplit_into_chunks

(vid_names, seqlen, stride)

datasets/data_utils/img_utils.py:328

core/base_trainer.py:86

Functiontorch2numpy

datasets/data_utils/img_utils.py:260

Functiontorch_inv_normal

datasets/data_utils/img_utils.py:254

Functiontorch_vid2numpy

datasets/data_utils/img_utils.py:272

Methodtrain_step

(self, input_batch)

core/base_trainer.py:74

Methodtrain_summaries

(self, input_batch)

core/base_trainer.py:77

Functiontransform_pts

Transform coordinates (N x 2) to different reference.

utils/imutils.py:47

Functiontransfrom_keypoints

(kp_2d, center_x, center_y, width, height, patch_width, patch_height, do_augment)

datasets/data_utils/img_utils.py:130

Functiontrim_videos

(filename, start_time, end_time, output_filename)

utils/demo_utils.py:82

core/base_trainer.py:83

Functionvis_batch_image_with_joints

batch_image: [batch_size, channel, height, width] batch_joints: [batch_size, num_joints, 3], batch_joints_vis: [batch_size, num_joints, 1

utils/vis.py:378

Functionvis_bbox_opencv

Visualizes a bounding box.

utils/vis.py:116

Functionvis_img_2Djoint

(batch_img, joints, pairs=None, joint_group=None)

utils/vis.py:485

← previous next →201–300 of 303, ranked by callers