MCPcopy Create free account

hub / github.com/NVlabs/InstantSplat / functions

Functions864 in github.com/NVlabs/InstantSplat

⨍ Functions 864◇ Types & classes 85
↓ 2 callersFunctionimg_to_arr
( img )
dust3r/utils/image.py:26
↓ 2 callersMethodinit
(self, dim_tokens_enc=768)
dust3r/heads/dpt_head.py:26
↓ 2 callersFunctioninit_filestructure
(save_path, n_views=None)
utils/sfm_utils.py:107
↓ 2 callersFunctionintegrate_weights
Compute the cumulative sum of w, assuming all weight vectors sum to 1. The output's size on the last dimension is one greater than that of the inpu
utils/stepfun.py:108
↓ 2 callersFunctioninterleave
(tensor1, tensor2)
dust3r/utils/misc.py:48
↓ 2 callersFunctioninvalid_to_zeros
(arr, valid_mask, ndim=999)
dust3r/utils/misc.py:112
↓ 2 callersFunctioninverse_sigmoid
(x)
utils/general_utils.py:18
↓ 2 callersFunctionis_good_type
returns (is_good, err_msg)
dust3r/datasets/base/base_stereo_view_dataset.py:184
↓ 2 callersFunctionl1_loss
(network_output, gt)
utils/loss_utils.py:39
↓ 2 callersFunctionlistify
(elems)
dust3r/utils/device.py:75
↓ 2 callersFunctionloadCameras
(poses, viewpoint_stack)
scene/dataset_readers.py:74
↓ 2 callersFunctionload_corres
(path_corres, device, min_conf_thr)
mast3r/cloud_opt/sparse_ga.py:717
↓ 2 callersFunctionload_images
open and convert all images in a list or folder to proper input format for DUSt3R
utils/sfm_utils.py:123
↓ 2 callersFunctionload_images
open and convert all images in a list or folder to proper input format for DUSt3R
dust3r/utils/image.py:73
↓ 2 callersFunctionmast3r_head_factory
build a prediction head for the decoder
mast3r/catmlp_dpt_head.py:99
↓ 2 callersFunctionmultiple_of_16
(x)
mast3r/utils/coarse_to_fine.py:29
↓ 2 callersFunctionnorm
(p)
mast3r/datasets/utils/cropping.py:65
↓ 2 callersFunctionnormalize_depth
Normalize the depth map to a range between 0 and 1.
utils/sfm_utils.py:371
↓ 2 callersFunctionoptimize_loop
(loss_func, lr_base, niter, pix_loss, lr_end=0)
mast3r/cloud_opt/sparse_ga.py:393
↓ 2 callersFunctionposes_avg
New pose using average position, z-axis, and up vector of input poses.
utils/pose_utils.py:230
↓ 2 callersMethodpreset_focal
(self, known_focals, msk=None)
dust3r/cloud_opt/optimizer.py:83
↓ 2 callersMethodprune_points
(self, mask)
scene/gaussian_model.py:362
↓ 2 callersFunctionpsnr
(img1, img2)
utils/image_utils.py:17
↓ 2 callersFunctionpts3d_to_trimesh
(img, pts3d, valid=None)
dust3r/viz.py:38
↓ 2 callersFunctionpts_dist_max
:param pts: (N, 3) torch or np :return: scalar
utils/align_traj.py:8
↓ 2 callersFunctionpts_dist_max
:param pts: (N, 3) torch or np :return: scalar
utils/utils_poses/align_traj.py:16
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png")
scene/dataset_readers.py:372
↓ 2 callersFunctionreg_dense_depth
extract 3D points from prediction head output
dust3r/heads/postprocess.py:22
↓ 2 callersFunctionrender_set
(model_path, name, iteration, views, gaussians, pipeline, background)
render.py:78
↓ 2 callersFunctionreproject_view
(pts3d, view2)
mast3r/datasets/utils/cropping.py:77
↓ 2 callersFunctionsample
Piecewise-Constant PDF sampling from a step function. Args: rand: random number generator (or None for `linspace` sampling). t: [..., num_b
utils/stepfun.py:175
↓ 2 callersFunctionsave_pose
(path, quat_pose, train_cams, llffhold=2)
train.py:46
↓ 2 callersFunctionsearchsorted
Find indices where v should be inserted into a to maintain order. Args: a: tensor, the sorted reference points that we are scanning to see wher
utils/stepfun.py:6
↓ 2 callersMethodselect_scene
(self, scene, *instances, opposite=False)
dust3r/datasets/megadepth.py:43
↓ 2 callersFunctionselout
(ar)
dust3r/utils/misc.py:82
↓ 2 callersFunctionshift
(pose, dz)
utils/pose_utils.py:494
↓ 2 callersMethodshow
(self, show_cams=True)
mast3r/cloud_opt/sparse_ga.py:104
↓ 2 callersFunctionssim
(img1, img2, window_size=11, size_average=True)
utils/loss_utils.py:55
↓ 2 callersFunctionstorePly
(path, xyz, rgb)
scene/dataset_readers.py:222
↓ 2 callersMethodtraining_setup
(self, training_args)
scene/gaussian_model.py:174
↓ 2 callersFunctiontransposed
(dic)
dust3r/utils/misc.py:99
↓ 2 callersFunctionuint8
(colors)
dust3r/viz.py:336
↓ 2 callersFunctionunpad_poses
Remove the homogeneous bottom row from [..., 4, 4] pose matrices.
utils/pose_utils.py:253
↓ 2 callersFunctionviewmatrix
(z, up, pos)
utils/utils_poses/vis_pose_utils.py:65
↓ 2 callersFunctionvisualizer
(camera_poses, colors, save_path="/mnt/data/1.png")
utils/camera_utils.py:208
↓ 1 callersFunctionExp
so(3) vector to SO(3) matrix :param r: (3, ) axis-angle, torch tensor :return: (3, 3)
utils/utils_poses/lie_group_helper.py:68
↓ 1 callersFunctionPILtoTorch
(pil_image, resolution)
utils/general_utils.py:21
↓ 1 callersFunctionRGB2SH
(rgb)
utils/sh_utils.py:114
↓ 1 callersMethod_TSDF_postprocess_or_not
(self, pts3d, depthmaps, confs, niter=1)
mast3r/cloud_opt/tsdf_optimizer.py:253
↓ 1 callersMethod_TSDF_query
TSDF query call: returns the weighted TSDF value for each query point [N, 3]
mast3r/cloud_opt/tsdf_optimizer.py:81
↓ 1 callersMethod__init__
(self, optimizer, subsample=8, TSDF_thresh=0., TSDF_batchsize=int(1e7))
mast3r/cloud_opt/tsdf_optimizer.py:16
↓ 1 callersMethod__init__
(self, *args, **kwargs)
dust3r/cloud_opt/base_opt.py:33
↓ 1 callersMethod_adjust_per_point_lr
Adjusts per-point learning rates based on gradient magnitudes.
scene/per_point_adam.py:26
↓ 1 callersMethod_check_edges
(self)
dust3r/cloud_opt/base_opt.py:130
↓ 1 callersMethod_compute_img_conf
(self, pred1_conf, pred2_conf)
dust3r/cloud_opt/base_opt.py:136
↓ 1 callersFunction_convert_scene_output_to_glb
(outfile, imgs, pts3d, mask, focals, cams2world, cam_size=0.05, cam_color=Non
mast3r/demo.py:65
↓ 1 callersFunction_convert_scene_output_to_glb
(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05, cam_color=None
dust3r/demo.py:66
↓ 1 callersMethod_crop_resize_if_necessary
This function: - first downsizes the image with LANCZOS inteprolation, which is better than bilinear interpolation in
mast3r/datasets/base/mast3r_base_stereo_view_dataset.py:56
↓ 1 callersMethod_encode_symmetrized
(self, view1, view2)
dust3r/model.py:152
↓ 1 callersFunction_fast_depthmap_to_pts3d
(depth, pixel_grid, focal, pp)
dust3r/cloud_opt/optimizer.py:204
↓ 1 callersMethod_get_confs_with_normals
(self)
mast3r/cloud_opt/tsdf_optimizer.py:129
↓ 1 callersMethod_get_depthpath
(self, obj, instance, view_idx)
dust3r/datasets/co3d.py:54
↓ 1 callersMethod_get_impath
(self, obj, instance, view_idx)
dust3r/datasets/co3d.py:51
↓ 1 callersMethod_get_maskpath
(self, obj, instance, view_idx)
dust3r/datasets/co3d.py:57
↓ 1 callersMethod_get_metadatapath
(self, obj, instance, view_idx)
dust3r/datasets/co3d.py:48
↓ 1 callersMethod_get_pixel_depths
Recover depth value for each input pixel coordinate, along with OOB validity mask
mast3r/cloud_opt/tsdf_optimizer.py:111
↓ 1 callersMethod_get_views
(self, idx, resolution, rng)
dust3r/datasets/co3d.py:65
↓ 1 callersMethod_get_views
(self, idx, resolution, rng)
dust3r/datasets/base/base_stereo_view_dataset.py:60
↓ 1 callersMethod_init_from_views
(self, view1, view2, pred1, pred2, dist='l1', conf='log',
dust3r/cloud_opt/base_opt.py:44
↓ 1 callersMethod_load_data
(self)
dust3r/datasets/scannetpp.py:25
↓ 1 callersMethod_load_data
(self, split)
dust3r/datasets/blendedmvs.py:25
↓ 1 callersMethod_load_data
(self, split)
dust3r/datasets/megadepth.py:31
↓ 1 callersMethod_load_data
(self, split)
dust3r/datasets/arkitscenes.py:30
↓ 1 callersMethod_load_data
(self)
dust3r/datasets/waymo.py:25
↓ 1 callersMethod_load_one_view
(self, data_path, key, view_index, resolution, rng)
dust3r/datasets/habitat.py:60
↓ 1 callersMethod_proj_pts3d
Projection operation: from 3D points to 2D coordinates + depths
mast3r/cloud_opt/tsdf_optimizer.py:159
↓ 1 callersMethod_prune_optimizer
(self, mask)
scene/gaussian_model.py:344
↓ 1 callersMethod_read_depthmap
(self, depthpath, input_metadata)
dust3r/datasets/co3d.py:60
↓ 1 callersMethod_set_focal
(self, idx, focal, force=False)
dust3r/cloud_opt/optimizer.py:121
↓ 1 callersMethod_set_focal
(self, idx, focal, force=False)
dust3r/cloud_opt/pair_viewer.py:92
↓ 1 callersMethod_set_focal
(self, idx, focal, force=False)
dust3r/cloud_opt/modular_optimizer.py:88
↓ 1 callersMethod_set_principal_point
(self, idx, pp, force=False)
dust3r/cloud_opt/optimizer.py:134
↓ 1 callersMethod_set_principal_point
(self, idx, pp, force=False)
dust3r/cloud_opt/modular_optimizer.py:98
↓ 1 callersMethod_set_resolutions
(self, resolutions)
dust3r/datasets/base/base_stereo_view_dataset.py:120
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0. Source: https://pytorch3d.readthedocs.io/en/latest/_modules
utils/pose_utils.py:106
↓ 1 callersMethod_swap_view_aug
(self, views)
mast3r/datasets/base/mast3r_base_stereo_view_dataset.py:52
↓ 1 callersMethodadd_negatives
(self, outdesc2, desc2, batchid, x2, y2)
mast3r/losses.py:361
↓ 1 callersFunctionaggreg_matches
(img1, list_matches)
mast3r/cloud_opt/sparse_ga.py:795
↓ 1 callersMethodaggregate_confs
(self, confs1, confs2)
mast3r/losses.py:474
↓ 1 callersFunctionalignPositionYaw
(p_es, p_gt, q_es, q_gt, n_aligned=1)
utils/utils_poses/ATE/align_utils.py:48
↓ 1 callersFunctionalignPositionYawSingle
calcualte the 4DOF transformation: yaw R and translation t so that: gt = R * est + t
utils/utils_poses/ATE/align_utils.py:27
↓ 1 callersFunctionalignSE3
Calculate SE3 transformation R and t so that: gt = R * est + t
utils/utils_poses/ATE/align_utils.py:86
↓ 1 callersFunctionalignSE3Single
Calculate SE3 transformation R and t so that: gt = R * est + t Using only the first poses of est and gt
utils/utils_poses/ATE/align_utils.py:65
↓ 1 callersFunctionalignSIM3
calculate s, R, t so that: gt = R * s * est + t
utils/utils_poses/ATE/align_utils.py:107
↓ 1 callersFunctionalign_ate_c2b_use_a2b
Align c to b using the sim3 from a to b. :param traj_a: (N0, 3/4, 4) torch tensor :param traj_b: (N0, 3/4, 4) torch tensor :param traj_c
utils/align_traj.py:26
↓ 1 callersFunctionalign_pose
(pose1, pose2)
utils/sfm_utils.py:464
↓ 1 callersFunctionarcball_nearest_axis
Return axis, which arc is nearest to point.
utils/utils_poses/ATE/transformations.py:1769
↓ 1 callersFunctionaverage_distance
(points)
utils/pose_utils.py:490
↓ 1 callersFunctionaxis_angle
Compute a rotation matrix from an axis and an angle. Returns 3x3 Matrix. Is the same as transformations.rotation_matrix(theta, axis). cfo,
utils/utils_poses/ATE/transformations.py:216
↓ 1 callersFunctionbackproj
(K, depthmap, subsample)
mast3r/cloud_opt/sparse_ga.py:912
← previousnext →201–300 of 864, ranked by callers