MCPcopy Create free account

hub / github.com/ai4ce/RAP / functions

Functions1,383 in github.com/ai4ce/RAP

↓ 2 callersFunction_read_pfm
(file)
dust3r/croco/stereoflow/datasets_stereo.py:581
↓ 2 callersMethod_refine_depths_with_TSDF
Leverage TSDF to post-process estimated depths for each pixel, find zero level of TSDF along ray (or closest to 0)
mast3r/cloud_opt/tsdf_optimizer.py:34
↓ 2 callersMethod_resize_flow
(self, flow, scale_x, scale_y, factor=1.0)
dust3r/croco/stereoflow/augmentor.py:198
↓ 2 callersFunction_resize_img
(img, new_size)
dust3r/croco/stereoflow/engine.py:163
↓ 2 callersFunction_resize_stereo_or_flow
(data, new_size)
dust3r/croco/stereoflow/engine.py:165
↓ 2 callersFunction_rotation_origin_to_pt
Align the origin (0,0,1) with the target point (x,y,1) in projective space. Method: rotate z to put target on (x'+,0,1), then rotate on Y to get
mast3r/datasets/utils/cropping.py:107
↓ 2 callersMethod_save_full_checkpoint
Saves the full training state, allowing exact resume: - model, optimizers, schedulers - discriminator (if used) - Ear
rap.py:201
↓ 2 callersFunction_score_cell
(cell1, H2, W2, p1, p2, min_corres=10, forced_resolution=None)
mast3r/utils/coarse_to_fine.py:111
↓ 2 callersMethod_set_depthmap
(self, idx, depth, force=False)
dust3r/dust3r/cloud_opt/pair_viewer.py:83
↓ 2 callersFunction_start_pos
(total_size, win_size, overlap)
mast3r/utils/coarse_to_fine.py:18
↓ 2 callersFunction_weight_pixels
(cell, pix, assigned, gauss_var=2)
mast3r/utils/coarse_to_fine.py:91
↓ 2 callersMethodadapt_tokens
(self, encoder_tokens)
dust3r/croco/models/dpt_block.py:412
↓ 2 callersFunctionadjust_learning_rate_by_lr
(optimizer, lr)
dust3r/dust3r/optim_factory.py:9
↓ 2 callersFunctionaggregate_stats
(info_str, pose_errors, angular_errors)
dust3r/dust3r_visloc/evaluation.py:15
↓ 2 callersFunctionalign_multiple_poses
(src_poses, target_poses)
dust3r/dust3r/cloud_opt/init_im_poses.py:308
↓ 2 callersMethodapply_rope1d
(self, tokens, pos1d, cos, sin)
dust3r/croco/models/pos_embed.py:136
↓ 2 callersFunctionarea
(sz)
dust3r/croco/datasets/crops/extract_crops_from_images.py:115
↓ 2 callersFunctionasymmetric_generalized_gaussian_fit
(x)
utils/brisque.py:70
↓ 2 callersFunctionbruteforce_reciprocal_nns
(A, B, device='cuda', block_size=None, dist='l2')
mast3r/fast_nn.py:17
↓ 2 callersFunctionbuild_dataset
(dataset, batch_size, num_workers, test=False)
dust3r/dust3r/training.py:253
↓ 2 callersFunctionbuild_position_encoding
(config)
models/apr/pencoder.py:72
↓ 2 callersFunctionbuild_rotation
(r)
utils/general_utils.py:76
↓ 2 callersMethodcalculate_brisque_features
(self, image)
utils/brisque.py:131
↓ 2 callersFunctioncameraList_from_camInfos
(cam_infos, args)
utils/camera_utils.py:82
↓ 2 callersFunctioncat_3d
(vecs)
dust3r/dust3r/viz.py:23
↓ 2 callersFunctioncat_meshes
(meshes)
dust3r/dust3r/viz.py:78
↓ 2 callersFunctioncenter_and_z
(poses)
dust3r/dust3r/cloud_opt/init_im_poses.py:312
↓ 2 callersFunctioncoarse_matching
(query_view, map_view, model, device, pixel_tol, fast_nn_params)
visloc.py:73
↓ 2 callersFunctioncollate_with_cat
(whatever, lists=False)
dust3r/dust3r/utils/device.py:47
↓ 2 callersFunctioncompute_camera_intrinsics
(height, width, hfov)
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:17
↓ 2 callersFunctioncompute_pointcloud_overlaps_scikit
Compute 'overlapping' metrics based on a distance threshold between two point clouds.
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:50
↓ 2 callersFunctionconvert_im_matches_pairs
(img0, img1, image_to_colmap, im_keypoints, matches_im0, matches_im1, viz)
mast3r/colmap/database.py:25
↓ 2 callersFunctioncrop_tag
(cell)
mast3r/utils/coarse_to_fine.py:10
↓ 2 callersFunctioncrop_to_homography
Given an image and its intrinsics, we want to replicate a rectangular crop with an homography, so that the principal point of the n
mast3r/datasets/utils/cropping.py:158
↓ 2 callersFunctiondecoder
(feat1, feat2, pos1, pos2, shape1, shape2)
mast3r/cloud_opt/sparse_ga.py:580
↓ 2 callersMethoddensification_postfix
(self, new_tensor)
models/gs/gaussian_model.py:579
↓ 2 callersFunctiondepthmap_to_camera_coordinates
Args: - depthmap (HxW array): - camera_intrinsics: a 3x3 matrix Returns: pointmap of absolute coordinates (HxWx3 arra
dust3r/dust3r/utils/geometry.py:165
↓ 2 callersFunctiondetect_model_format
(path, ext)
utils/read_write_model.py:477
↓ 2 callersFunctiondist_func
(A, B)
mast3r/fast_nn.py:30
↓ 2 callersMethodembed
(self, inputs)
models/gs/embedder.py:37
↓ 2 callersFunctionestimate_R_hat
(r_hat, gamma)
utils/brisque.py:21
↓ 2 callersFunctionestimate_phi
(alpha)
utils/brisque.py:10
↓ 2 callersFunctionestimate_r_hat
(x)
utils/brisque.py:16
↓ 2 callersFunctioneval_model
Convert Rotation matrix to quaternion, then calculate the location errors. original from PoseNet Paper
utils/eval_utils.py:100
↓ 2 callersFunctionexport_results
(output_dir, xp_label, query_names, poses_pred)
dust3r/dust3r_visloc/evaluation.py:38
↓ 2 callersFunctionextract_correspondences_from_pts3d
(view1, view2, target_n_corres, rng=np.random, ret_xy=True, nneg=0)
mast3r/datasets/utils/cropping.py:22
↓ 2 callersFunctionextract_correspondences_nonsym
(A, B, confA, confB, subsample=8, device=None, ptmap_key='pred_desc', pixel_tol=0)
mast3r/fast_nn.py:191
↓ 2 callersFunctionfast_pnp
(pts3d, focal, msk, device, pp=None, niter_PnP=10)
dust3r/dust3r/cloud_opt/init_im_poses.py:247
↓ 2 callersFunctionfetchPly
(path)
dataset_loaders/dataset_readers.py:126
↓ 2 callersFunctionfilter_pairs_seq
(pairs, seq_dis_thr, cyclic=False)
dust3r/dust3r/image_pairs.py:94
↓ 2 callersFunctionfind_reciprocal_matches
returns 3 values: 1 - reciprocal_in_P2: a boolean array of size P2.shape[0], a "True" value indicates a match 2 - nn2_in_P1: a int array
dust3r/dust3r/utils/geometry.py:345
↓ 2 callersFunctionfine_matching
(query_views, map_views, model, device, max_batch_size, pixel_tol, fast_nn_params)
visloc.py:150
↓ 2 callersMethodforward
(self, img1, img2)
dust3r/croco/datasets/transforms.py:18
↓ 2 callersFunctionfreeze_bn_layer
freeze bn layer by not require grad but still behave differently when model.train() vs. model.eval()
utils/model_utils.py:36
↓ 2 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, **kw)
dust3r/dust3r/model.py:77
↓ 2 callersFunctiongenerate_orientation_noise
(pan_range, tilt_range, roll_range)
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:111
↓ 2 callersFunctiongetNerfppNorm
(cam_info)
dataset_loaders/dataset_readers.py:50
↓ 2 callersFunctionget_1d_sincos_pos_embed_from_grid
embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)
dust3r/croco/models/pos_embed.py:51
↓ 2 callersFunctionget_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [n_cls_token+grid_size*grid_size, embe
dust3r/croco/models/pos_embed.py:22
↓ 2 callersMethodget_adaptors
(self)
dust3r/dust3r/cloud_opt/base_opt.py:143
↓ 2 callersFunctionget_conf_trf
(mode)
dust3r/dust3r/cloud_opt/commons.py:48
↓ 2 callersFunctionget_context
(delete_cache)
mast3r/demo.py:226
↓ 2 callersMethodget_dense_pts3d
(self, clean_depth=True, subsample=8)
mast3r/cloud_opt/sparse_ga.py:70
↓ 2 callersMethodget_dense_pts3d
(self, clean_depth=True)
mast3r/cloud_opt/tsdf_optimizer.py:269
↓ 2 callersFunctionget_embedder
(multires)
models/gs/embedder.py:41
↓ 2 callersMethodget_focals
(self)
dust3r/dust3r/cloud_opt/optimizer.py:127
↓ 2 callersMethodget_focals
(self)
dust3r/dust3r/cloud_opt/modular_optimizer.py:94
↓ 2 callersMethodget_gaussian_trans
(self, alpha=0)
utils/cameras.py:137
↓ 2 callersMethodget_im_poses
(self)
dust3r/dust3r/cloud_opt/base_opt.py:225
↓ 2 callersMethodget_intrinsics
(self)
dust3r/dust3r/cloud_opt/pair_viewer.py:104
↓ 2 callersFunctionget_known_poses
(self)
dust3r/dust3r/cloud_opt/init_im_poses.py:290
↓ 2 callersFunctionget_med_dist_between_poses
(poses)
dust3r/dust3r/utils/geometry.py:364
↓ 2 callersMethodget_name
(self)
dust3r/dust3r/losses.py:97
↓ 2 callersFunctionget_pose_error
(pr_camtoworld, gt_cam_to_world)
dust3r/dust3r_visloc/evaluation.py:31
↓ 2 callersMethodget_principal_points
(self)
dust3r/dust3r/cloud_opt/optimizer.py:141
↓ 2 callersMethodget_principal_points
(self)
dust3r/dust3r/cloud_opt/pair_viewer.py:101
↓ 2 callersMethodget_principal_points
(self)
dust3r/dust3r/cloud_opt/modular_optimizer.py:105
↓ 2 callersMethodget_pw_norm_scale_factor
(self)
dust3r/dust3r/cloud_opt/base_opt.py:178
↓ 2 callersFunctionget_rank
()
dust3r/croco/utils/misc.py:207
↓ 2 callersFunctionget_similarities
(desc1, desc2, euc=False)
mast3r/losses.py:202
↓ 2 callersFunctionglobal_alignment_iter
(net, cur_iter, niter, lr_base, lr_min, optimizer, schedule)
dust3r/dust3r/cloud_opt/base_opt.py:352
↓ 2 callersFunctionhead_factory
build a prediction head for the decoder
dust3r/dust3r/heads/__init__.py:11
↓ 2 callersFunctionimg_to_arr
( img )
dust3r/dust3r/utils/image.py:26
↓ 2 callersFunctioninterleave
(tensor1, tensor2)
dust3r/dust3r/utils/misc.py:48
↓ 2 callersFunctioninvalid_to_zeros
(arr, valid_mask, ndim=999)
dust3r/dust3r/utils/misc.py:112
↓ 2 callersFunctionis_good_type
returns (is_good, err_msg)
dust3r/dust3r/datasets/base/base_stereo_view_dataset.py:184
↓ 2 callersMethodis_other_pointcloud_overlapping
Check if a viewpoint is valid and overlaps significantly with a reference one.
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:256
↓ 2 callersMethodjitter_one
(self, img, fn_idx, brightness_factor, contrast_factor, saturation_factor, hue_factor)
dust3r/croco/datasets/transforms.py:41
↓ 2 callersFunctionlistify
(elems)
dust3r/dust3r/utils/device.py:75
↓ 2 callersFunctionload_corres
(path_corres, device, min_conf_thr)
mast3r/cloud_opt/sparse_ga.py:717
↓ 2 callersFunctionload_image
(impath)
dust3r/croco/datasets/pairs_dataset.py:10
↓ 2 callersFunctionload_images
open and convert all images in a list or folder to proper input format for DUSt3R
dust3r/dust3r/utils/image.py:73
↓ 2 callersFunctionload_sfm
(sfm_dir, cam_type='dslr')
dust3r/datasets_preprocess/preprocess_scannetpp.py:72
↓ 2 callersMethodload_state_dict
(self, ckpt, **kw)
dust3r/dust3r/model.py:90
↓ 2 callersFunctionlocal_mean_opencv
(image, kernel)
utils/brisque.py:66
↓ 2 callersFunctionlook_at_for_habitat
(eye, center, up, return_cam2world=True)
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:106
↓ 2 callersFunctionmake_pairs
(imgs, scene_graph='complete', prefilter=None, symmetrize=True)
dust3r/dust3r/image_pairs.py:11
↓ 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 callersFunctionnormalize
Normalize a vector.
utils/reverse_dfnet.py:4
← previousnext →201–300 of 1,383, ranked by callers