Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/XunshanMan/MVGFormer
/ functions
Functions
561 in github.com/XunshanMan/MVGFormer
⨍
Functions
561
◇
Types & classes
78
Function
perspective_projection
This function computes the perspective projection of a set of points. Input: points (bs, N, 3): 3D points rotation (bs, 3, 3)
lib/smpl/utils.py:82
Function
plot_logs
Function to plot specific fields from training log(s). Plots both training and test results. :: Inputs - logs = list containing Path obj
lib/models/util/plot_utils.py:23
Function
plot_precision_recall
(files, naming_scheme='iter')
lib/models/util/plot_utils.py:91
Function
prepare_batch
(batch, device, config, is_train=True)
lib/mvn/datasets/utils.py:45
Function
procrustes_transform
(target_pose, from_pose)
lib/utils/transforms.py:299
Function
project_pose
Args x: 3xN points in world coordinates R: 3x3 Camera rotation matrix T: 3x1 Camera translation parameters f: 2 C
lib/utils/cameras_cuda_col.py:26
Function
project_ref_points
(reference_points, cameras, nviews, batch_size, nbins, device, center, scale,
lib/utils/cameras.py:50
Method
projection
(self)
lib/mvn/utils/multiview.py:47
Method
record_stream
(self, *args, **kwargs)
lib/models/util/misc.py:436
Function
reduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
lib/utils/misc.py:158
Function
reduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
lib/models/util/misc.py:244
Method
render
(self, proj_matrix, canvas)
lib/mvn/utils/volumetric.py:14
Method
render
(self, proj_matrix, canvas)
lib/mvn/utils/volumetric.py:76
Function
render_points_as_2d_gaussians
(points, sigmas, image_shape, normalize=True)
lib/mvn/utils/op.py:178
Method
residual_function
(x)
lib/mvn/models/triangulation.py:116
Function
rot6d_to_rotmat
Convert 6D rotation representation to 3x3 rotation matrix. Based on Zhou et al., "On the Continuity of Rotation Representations in Neural Netw
lib/smpl/utils.py:64
Function
rotate_coord_volume
(coord_volume, theta, axis)
lib/mvn/utils/volumetric.py:102
Function
rotate_points
:param points: N*2 :param center: 2 :param rot_rad: scalar :return: N*2
lib/utils/transforms.py:217
Function
rotation_matrix_to_angle_axis
Convert 3x4 rotation matrix to Rodrigues vector Args: rotation_matrix (torch.Tensor): rotation matrix. Returns: torch.Tensor:
lib/smpl/utils.py:129
Function
save_3d_images
(config, meta, preds, file_name)
lib/utils/vis.py:478
Function
save_3d_images_direct
(config, meta, preds, file_name)
lib/utils/vis.py:284
Function
save_3d_images_novel_view
(config, meta, preds, file_name)
lib/utils/vis.py:533
Function
save_debug_2d_images_seq
(batch_image,gt,prefix,index=None)
lib/utils/vis.py:586
Function
save_debug_3d_cubes
(config, meta, root, prefix)
lib/utils/vis.py:757
Function
save_debug_3d_images_seq
(pd_list,gt_list,prefix,index=None)
lib/utils/vis.py:625
Function
save_debug_epipolar
(inputs, meta, targets_2d, out, prefix)
lib/utils/vis.py:812
Function
save_debug_images_multi
(config, input, meta, target, output, prefix)
lib/utils/vis.py:449
Function
save_on_master
(*args, **kwargs)
lib/utils/misc.py:395
Function
save_on_master
(*args, **kwargs)
lib/models/util/misc.py:499
Method
set_epoch
(self, epoch)
lib/dataset/samplers.py:85
Method
set_epoch
(self, epoch)
lib/dataset/samplers.py:152
Function
soft_oks_nms
Soft OKS NMS implementations. Args: kpts_db: keypoints and scores. thr: retain oks overlap < thr. max_dets: max number of
lib/core/nms.py:155
Method
synchronize_between_processes
Warning: does not synchronize the deque!
lib/utils/misc.py:68
Method
synchronize_between_processes
(self)
lib/utils/misc.py:214
Method
synchronize_between_processes
Warning: does not synchronize the deque!
lib/models/util/misc.py:104
Method
synchronize_between_processes
(self)
lib/models/util/misc.py:300
Function
time_synchronized
()
lib/models/mvp_decoder.py:44
Function
transform_preds
(coords, center, scale, output_size)
lib/utils/transforms.py:64
Function
triangulate_batch_of_points
(proj_matricies_batch, points_batch, confidences_batch=None, device='default')
lib/mvn/utils/multiview.py:230
Function
triangulate_batch_of_points_batch_version
Triangulates for a batch of points
lib/mvn/utils/multiview.py:257
Function
triangulate_point_from_multiple_views_linear
Triangulates one point from multiple (N) views using direct linear transformation (DLT). For more information look at "Multiple view geometry in c
lib/mvn/utils/multiview.py:113
Function
umeyama
Estimate N-D similarity transformation with or without scaling. Parameters ---------- src : (M, N) array Source coordinates. d
lib/smpl/umeyama.py:46
Function
undistort_and_save
(idx)
lib/mvn/datasets/human36m_preprocessing/undistort-h36m.py:79
Function
unproject_heatmaps
(heatmaps, proj_matricies, coord_volumes, volume_aggregation_method='sum', vol_confidences=None)
lib/mvn/utils/op.py:99
Method
update
(self, **kwargs)
lib/utils/misc.py:191
Method
update
(self, val, n=1)
lib/mvn/utils/misc.py:24
Function
update_dir
(model_dir, log_dir, data_dir)
lib/core/config.py:405
Function
uv_to_image_frame
:param uv: (N, 2) :param f: scalar :param c: (2, 1) :param k: :param p: :return:
lib/utils/cameras.py:258
Method
value
(self)
lib/utils/misc.py:101
Method
value
(self)
lib/models/util/misc.py:137
Method
vis_camera_detail
(pd,gt,prefix)
lib/dataset/panoptic.py:589
Function
visualize_batch
(images_batch, heatmaps_batch, keypoints_2d_batch, proj_matricies_batch, keypoints_3d_batc
lib/mvn/utils/vis.py:84
Function
visualize_heatmaps
(images_batch, heatmaps_batch, kind="cmu", batch_index=0, size=5
lib/mvn/utils/vis.py:182
Function
visualize_joints_ids
(loader, save_dir, prefix)
process/extract_dataset.py:162
Function
visualize_volumes
(images_batch, volumes_batch, proj_matricies_batch, kind="cmu", cu
lib/mvn/utils/vis.py:223
Function
weight_reduce_loss
Apply element-wise weight and reduce loss. Args: loss (Tensor): Element-wise loss. weight (Tensor): Element-wise weights.
lib/core/loss.py:373
Function
worker_init_fn
(worker_id)
lib/mvn/datasets/utils.py:42
Function
world_to_camera_frame
Args x: 3xN 3d points in world coordinates R: 3x3 Camera rotation matrix T: 3x1 Camera translation parameters Returns
lib/utils/cameras_cuda_col.py:59
Function
world_to_camera_frame
Args x: Nx3 3d points in world coordinates R: 3x3 Camera rotation matrix T: 3x1 Camera translation parameters Returns
lib/utils/cameras.py:225
Function
world_to_camera_frame
Args x: Nx3 3d points in world coordinates R: 3x3 Camera rotation matrix T: 3x1 Camera translation parameters Returns
lib/utils/cameras_cpu.py:68
Function
xmlread
(filename)
lib/utils/zipreader.py:58
← previous
501–561 of 561, ranked by callers