MCPcopy Create free account

hub / github.com/ai4ce/RAP / functions

Functions1,383 in github.com/ai4ce/RAP

↓ 382 callersFunctionprint
(*args, **kwargs)
dust3r/croco/utils/misc.py:182
↓ 53 callersFunctiongeotrf
Apply a geometric transformation to a list of 3-D points. H: 3x3 or 4x4 projection matrix (typically a Homography) p: numpy/torch/tuple of c
dust3r/dust3r/utils/geometry.py:40
↓ 53 callersMethodsize
(self)
dust3r/dust3r/datasets/utils/cropping.py:41
↓ 50 callersMethodmax
(self)
dust3r/croco/utils/misc.py:73
↓ 44 callersFunctionto_numpy
(x)
dust3r/dust3r/utils/device.py:42
↓ 38 callersMethodsave
(self, iteration)
utils/scene.py:78
↓ 30 callersMethodrender
Render the scene. Background tensor (bg_color) must be on GPU!
models/gs/gaussian_model.py:713
↓ 27 callersFunctionsafe_path
(path)
utils/general_utils.py:19
↓ 26 callersFunctionadd_arg
(name_or_flags, default=None, default_stereo=None, default_flow=None, **kwargs)
dust3r/croco/stereoflow/train.py:43
↓ 26 callersFunctionview_name
(view, batch_index=None)
dust3r/dust3r/datasets/base/base_stereo_view_dataset.py:194
↓ 21 callersMethodresize
(self, *args, **kwargs)
dust3r/dust3r/datasets/utils/cropping.py:46
↓ 20 callersMethodupdate
(self, **kwargs)
dust3r/croco/utils/misc.py:94
↓ 19 callersFunctionimread_cv2
Open an image or a depthmap with opencv-python.
dust3r/dust3r/utils/image.py:31
↓ 19 callersMethodmedian
(self)
dust3r/croco/utils/misc.py:59
↓ 19 callersFunctionwrite_next_bytes
pack and write to a binary file. :param fid: :param data: data to send, if multiple elements are sent at the same time, they should be enc
utils/read_write_model.py:90
↓ 17 callersFunctioninv
Invert a torch or numpy matrix
dust3r/dust3r/utils/geometry.py:104
↓ 17 callersMethodload_state_dict
(self, state_dict)
utils/early_stopper.py:77
↓ 16 callersMethodadd_pointcloud
(self, pts3d, color=(0,0,0), mask=None, denoise=False)
dust3r/dust3r/viz.py:137
↓ 16 callersMethodclose
(self)
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:218
↓ 16 callersMethodconvert
(self, img, interpolation=cv2.INTER_LINEAR, borderMode=cv2.BORDER_WRAP, single_map=False)
dust3r/datasets_preprocess/habitat/habitat_renderer/projections_conversions.py:34
↓ 14 callersFunctionedge_str
(i, j)
dust3r/dust3r/cloud_opt/commons.py:12
↓ 14 callersFunctionrgb
(ftensor, true_shape=None)
dust3r/dust3r/utils/image.py:44
↓ 13 callersMethoddevice
(self)
dust3r/dust3r/cloud_opt/base_opt.py:120
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
utils/read_write_model.py:78
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
dataset_loaders/colmap_loader.py:75
↓ 12 callersMethod_set_root
(self)
dust3r/croco/stereoflow/datasets_stereo.py:121
↓ 12 callersMethodstate_dict
(self)
utils/early_stopper.py:68
↓ 12 callersMethodupdate
(self, *args)
utils/cameras.py:55
↓ 11 callersMethodadd_camera
(self, pose_c2w, focal=None, color=(0, 0, 0), image=None, imsize=None, cam_size=0.03)
dust3r/dust3r/viz.py:190
↓ 11 callersMethodtrain
(self)
rap.py:166
↓ 10 callersFunctionauto_cam_size
(im_poses)
dust3r/dust3r/viz.py:115
↓ 10 callersFunctionfast_reciprocal_NNs
(pts1, pts2, subsample_or_initxy1=8, ret_xy=True, pixel_tol=0, ret_basin=False, device
mast3r/fast_nn.py:109
↓ 10 callersFunctioninvalid_to_nans
(arr, valid_mask, ndim=999)
dust3r/dust3r/utils/misc.py:103
↓ 10 callersMethodset_eval
(self, eval=True)
models/gs/gaussian_model.py:683
↓ 9 callersFunctionParameterStack
(params, keys=None, is_param=None, fill=0)
dust3r/dust3r/cloud_opt/optimizer.py:214
↓ 9 callersMethodbackward
(ctx, grad_res)
dust3r/croco/models/curope/curope2d.py:25
↓ 9 callersFunctionfocal2fov
(focal, pixels)
utils/graphics_utils.py:94
↓ 9 callersMethodshow
(self, show_pw_cams=False, show_pw_pts3d=False, cam_size=None, **kw)
dust3r/dust3r/cloud_opt/base_opt.py:297
↓ 8 callersMethod_crop_resize_if_necessary
This function: - first downsizes the image with LANCZOS inteprolation, which is better than bilinear interpolation in
dust3r/dust3r/datasets/base/base_stereo_view_dataset.py:137
↓ 8 callersFunctiondepthmap_to_absolute_camera_coordinates
Args: - depthmap (HxW array): - camera_intrinsics: a 3x3 matrix - camera_pose: a 4x3 or 4x4 cam2world matrix Returns:
dust3r/dust3r/utils/geometry.py:200
↓ 8 callersMethodmatch
(self, query, reference, conf_thr=-1, vis_fig=None)
matcher.py:120
↓ 8 callersMethodq_to_Q
(self, q)
utils/pose_utils.py:247
↓ 8 callersFunctionxy_grid
Output a (H,W,2) array of int32 with output[j,i,0] = i + origin[0] output[j,i,1] = j + origin[1]
dust3r/dust3r/utils/geometry.py:15
↓ 7 callersMethod__init__
(self, mask_bg=True, *args, ROOT, **kwargs)
mast3r/datasets/__init__.py:30
↓ 7 callersMethod__init__
(self, max_gtnorm=None)
dust3r/croco/stereoflow/criterion.py:22
↓ 7 callersMethod_set_root
(self)
dust3r/croco/stereoflow/datasets_flow.py:104
↓ 7 callersFunctionfix_seed
(seed=7)
utils/general_utils.py:112
↓ 7 callersFunctionget_pose_error
(gt_pose, pred_pose)
utils/eval_utils.py:56
↓ 7 callersMethodinit
Initialize parts of decoder that are dependent on dimension of encoder tokens. Should be called when setting up MultiMAE. :p
dust3r/croco/models/dpt_block.py:341
↓ 7 callersMethodload_state_dict
(self, state_dict)
dust3r/croco/utils/misc.py:271
↓ 7 callersMethodshow
(self, point_size=2)
dust3r/dust3r/viz.py:208
↓ 6 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, bias=True, drop=0.)
dust3r/croco/models/blocks.py:60
↓ 6 callersFunctionapply_log_to_norm
(xyz)
mast3r/losses.py:20
↓ 6 callersFunctioncat
(a, b)
dust3r/dust3r/viz.py:322
↓ 6 callersMethodcrop
(self, *args, **kwargs)
dust3r/dust3r/datasets/utils/cropping.py:49
↓ 6 callersFunctionget_combined_args
(parser: ArgumentParser)
arguments/__init__.py:151
↓ 6 callersFunctionget_resize_function
(maxdim, patch_size, H, W, is_mask=False)
dust3r/dust3r_visloc/datasets/utils.py:51
↓ 6 callersFunctionhash_md5
(s)
mast3r/utils/misc.py:16
↓ 6 callersFunctionmkdir_for
(f)
mast3r/utils/misc.py:11
↓ 6 callersFunctionpsnr
(img1, img2)
utils/image_utils.py:27
↓ 6 callersFunctionsRT_to_4x4
(scale, R, T, device)
dust3r/dust3r/cloud_opt/init_im_poses.py:226
↓ 6 callersFunctionse3_2_qt_parallel
(wu)
utils/pose_utils.py:345
↓ 6 callersMethodset_epoch
(self, epoch)
dust3r/dust3r/datasets/base/easy_dataset.py:132
↓ 6 callersFunctionto_cpu
(x)
dust3r/dust3r/utils/device.py:43
↓ 6 callersFunctiontodevice
Transfer some variables to another device (i.e. GPU, CPU:torch, CPU:numpy). batch: list, tuple, dict of tensors or other things device: pyto
dust3r/dust3r/utils/device.py:11
↓ 5 callersMethod__init__
(self, criterion, norm_mode='avg_dis', gt_scale=False, opt_fit_gt=False, sky_loss_value=2, ma
mast3r/losses.py:28
↓ 5 callersMethod_get_depthmaps
(self, TSDF_filtering_thresh=None)
mast3r/cloud_opt/tsdf_optimizer.py:27
↓ 5 callersMethod_set_pose
(self, poses, idx, R, T=None, scale=None, force=False)
dust3r/dust3r/cloud_opt/base_opt.py:157
↓ 5 callersFunctioncolmap_to_opencv_intrinsics
Modify camera intrinsics to follow a different convention. Coordinates of the center of the top-left pixels are by default: - (0.5, 0.5)
dust3r/dust3r/utils/geometry.py:223
↓ 5 callersFunctionconfig_parser
()
arguments/options.py:9
↓ 5 callersFunctiondepthmap_to_pts3d
Args: - depthmap (BxHxW array): - pseudo_focal: [B,H,W] ; [B,2,H,W] or [B,1,H,W] Returns: pointmap of absolute coordi
dust3r/dust3r/utils/geometry.py:114
↓ 5 callersMethodexp_r2q_parallel
(self, r, eps=1e-9)
utils/pose_utils.py:296
↓ 5 callersMethodextract
(self, args)
arguments/__init__.py:42
↓ 5 callersFunctionfov2focal
(fov, pixels)
utils/graphics_utils.py:90
↓ 5 callersFunctionget_HW_resolution
(H, W, maxdim, patchsize=16)
dust3r/dust3r_visloc/datasets/utils.py:27
↓ 5 callersMethodget_focals
(self)
mast3r/cloud_opt/sparse_ga.py:58
↓ 5 callersMethodget_im_poses
(self)
mast3r/cloud_opt/sparse_ga.py:64
↓ 5 callersMethodget_pts3d
(self, raw=False)
dust3r/dust3r/cloud_opt/base_opt.py:203
↓ 5 callersMethodlog_every
(self, iterable, print_freq, header=None, max_iter=None)
dust3r/croco/utils/misc.py:126
↓ 5 callersFunctionloss_of_one_batch
(batch, model, criterion, device, symmetrize_batch=False, use_amp=False, ret=None)
dust3r/dust3r/inference.py:32
↓ 5 callersMethodrender_viewpoint
(self, viewpoint_position, viewpoint_orientation)
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:273
↓ 5 callersFunctionrescale_points3d
(pts2d, pts3d, to_resize, HR, WR)
dust3r/dust3r_visloc/datasets/utils.py:93
↓ 5 callersFunctionrun_pnp
use OPENCV model for distortion (4 values)
dust3r/dust3r_visloc/localization.py:30
↓ 4 callersFunctionNoGradParamDict
(x)
dust3r/dust3r/cloud_opt/commons.py:28
↓ 4 callersMethod__init__
(self, criterion, norm_mode='avg_dis', gt_scale=False)
dust3r/dust3r/losses.py:153
↓ 4 callersFunction_cell_size
(cell2)
mast3r/utils/coarse_to_fine.py:43
↓ 4 callersMethod_decoder
return_all_blocks: if True, return the features at the end of every block instead of just the features from the l
dust3r/croco/models/croco.py:166
↓ 4 callersMethod_downstream_head
(self, head_num, decout, img_shape)
dust3r/dust3r/model.py:192
↓ 4 callersMethod_encode_image
image has B x 3 x img_size x img_size do_mask: whether to perform masking or not return_all_blocks: if True, return the feat
dust3r/croco/models/croco.py:129
↓ 4 callersFunction_get_gtnorm
(gt)
dust3r/croco/stereoflow/criterion.py:12
↓ 4 callersMethod_get_msk_indices
(self, msk)
dust3r/dust3r/cloud_opt/optimizer.py:103
↓ 4 callersFunction_read_img
(filename)
dust3r/croco/stereoflow/datasets_stereo.py:541
↓ 4 callersFunction_resize_pil_image
(img, long_edge_size)
dust3r/dust3r/utils/image.py:63
↓ 4 callersMethodadd_cameras
(self, poses, focals=None, images=None, imsizes=None, colors=None, **kw)
dust3r/dust3r/viz.py:202
↓ 4 callersFunctionadd_scene_cam
(scene, pose_c2w, edge_color, image=None, focal=None, imsize=None, screen_width=0.03, marke
dust3r/dust3r/viz.py:246
↓ 4 callersFunctionargmin
(X, dim)
mast3r/fast_nn.py:33
↓ 4 callersFunctioncalculate_features
(coefficients)
utils/brisque.py:79
↓ 4 callersFunctioncompute_pointcloud
Return a 3D point cloud corresponding to valid pixels of the depth map
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:39
↓ 4 callersMethodconjugate
(self, q)
utils/pose_utils.py:276
↓ 4 callersFunctioncrop
(img, mask, pts3d, crop, intrinsics=None)
visloc.py:195
next →1–100 of 1,383, ranked by callers