MCPcopy Create free account

hub / github.com/ai4ce/RAP / functions

Functions1,383 in github.com/ai4ce/RAP

⨍ Functions 1,383◇ Types & classes 201
↓ 1 callersMethod_init_weights
(self, m)
dust3r/croco/models/croco.py:119
↓ 1 callersMethod_lazy_initialization
(self)
dust3r/datasets_preprocess/habitat/habitat_renderer/habitat_sim_envmaps_renderer.py:64
↓ 1 callersMethod_lazy_initialization
(self)
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:170
↓ 1 callersFunction_list_all_scenes
(path)
dust3r/datasets_preprocess/preprocess_staticthings3d.py:95
↓ 1 callersMethod_load_data
(self)
dust3r/dust3r/datasets/scannetpp.py:25
↓ 1 callersMethod_load_data
(self, split)
dust3r/dust3r/datasets/blendedmvs.py:25
↓ 1 callersMethod_load_data
(self, split)
dust3r/dust3r/datasets/megadepth.py:31
↓ 1 callersMethod_load_data
(self, split)
dust3r/dust3r/datasets/arkitscenes.py:30
↓ 1 callersMethod_load_data
(self)
dust3r/dust3r/datasets/waymo.py:25
↓ 1 callersFunction_load_kpts_and_poses
(root, scene_id, subscene, z_only=False, intrinsics=False)
dust3r/datasets_preprocess/preprocess_megadepth.py:106
↓ 1 callersFunction_load_model_and_criterion
(model_path, do_load_metrics, device)
dust3r/croco/stereoflow/test.py:51
↓ 1 callersMethod_load_one_view
(self, data_path, key, view_index, resolution, rng)
dust3r/dust3r/datasets/habitat.py:60
↓ 1 callersMethod_load_or_build_cache
(self)
dust3r/croco/stereoflow/datasets_flow.py:108
↓ 1 callersMethod_load_or_build_cache
(self)
dust3r/croco/stereoflow/datasets_stereo.py:125
↓ 1 callersFunction_load_pose
(path, ret_44=False)
dust3r/datasets_preprocess/preprocess_blendedMVS.py:96
↓ 1 callersMethod_load_sfm
(self, sfm_dir)
dust3r/dust3r_visloc/datasets/base_colmap.py:116
↓ 1 callersFunction_ntuple
(n)
dust3r/croco/models/blocks.py:24
↓ 1 callersMethod_prepare_data
(self)
dust3r/croco/stereoflow/datasets_flow.py:255
↓ 1 callersMethod_prepare_data
(self)
dust3r/croco/stereoflow/datasets_stereo.py:205
↓ 1 callersFunction_print_repr_dataset
(d)
dust3r/croco/stereoflow/train.py:170
↓ 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)
models/gs/gaussian_model.py:508
↓ 1 callersMethod_random_color
(self, img1, img2)
dust3r/croco/stereoflow/augmentor.py:133
↓ 1 callersMethod_random_crop
(self, img1, img2, disp)
dust3r/croco/stereoflow/augmentor.py:67
↓ 1 callersMethod_random_rotate_shift_right
(self, img2)
dust3r/croco/stereoflow/augmentor.py:86
↓ 1 callersMethod_random_scale
(self, img1, img2, disp)
dust3r/croco/stereoflow/augmentor.py:42
↓ 1 callersMethod_random_vflip
(self, img1, img2, disp)
dust3r/croco/stereoflow/augmentor.py:78
↓ 1 callersMethod_read_depthmap
(self, depthpath, input_metadata)
dust3r/dust3r/datasets/co3d.py:60
↓ 1 callersFunction_replace
Advanced string replacement. Given a text: - replace all elements in src by the corresponding element in tgt - remove all elements in rm
dust3r/croco/utils/misc.py:341
↓ 1 callersMethod_reset_parameters
(self)
models/apr/transformer_encoder.py:60
↓ 1 callersMethod_reset_parameters
(self)
models/apr/rapnet.py:49
↓ 1 callersFunction_save_batch
(pred, gt, pairnames, dataset, task, save, outdir, time, submission_dir=None)
dust3r/croco/stereoflow/test.py:82
↓ 1 callersMethod_set_decoder
(self, enc_embed_dim, dec_embed_dim, dec_num_heads, dec_depth, mlp_ratio, norm_layer, norm_im2_in_dec)
dust3r/croco/models/croco.py:95
↓ 1 callersMethod_set_focal
(self, idx, focal, force=False)
dust3r/dust3r/cloud_opt/optimizer.py:121
↓ 1 callersMethod_set_focal
(self, idx, focal, force=False)
dust3r/dust3r/cloud_opt/pair_viewer.py:92
↓ 1 callersMethod_set_focal
(self, idx, focal, force=False)
dust3r/dust3r/cloud_opt/modular_optimizer.py:88
↓ 1 callersMethod_set_mask_generator
(self, num_patches, mask_ratio)
dust3r/croco/models/croco.py:89
↓ 1 callersMethod_set_mask_token
(self, dec_embed_dim)
dust3r/croco/models/croco.py:92
↓ 1 callersMethod_set_patch_embed
(self, img_size=224, patch_size=16, enc_embed_dim=768)
dust3r/croco/models/croco.py:86
↓ 1 callersMethod_set_prediction_head
(self, dec_embed_dim, patch_size)
dust3r/croco/models/croco.py:107
↓ 1 callersMethod_set_principal_point
(self, idx, pp, force=False)
dust3r/dust3r/cloud_opt/optimizer.py:134
↓ 1 callersMethod_set_principal_point
(self, idx, pp, force=False)
dust3r/dust3r/cloud_opt/modular_optimizer.py:98
↓ 1 callersMethod_set_resolutions
(self, resolutions)
dust3r/dust3r/datasets/base/base_stereo_view_dataset.py:120
↓ 1 callersMethod_swap_view_aug
(self, views)
mast3r/datasets/base/mast3r_base_stereo_view_dataset.py:52
↓ 1 callersMethodadd_batch
(self, predictions, gt)
dust3r/croco/stereoflow/criterion.py:201
↓ 1 callersMethodadd_densification_stats
(self, viewspace_point_tensor, update_filter, width, height)
models/gs/gaussian_model.py:705
↓ 1 callersMethodadd_negatives
(self, outdesc2, desc2, batchid, x2, y2)
mast3r/losses.py:361
↓ 1 callersFunctionadd_perturbation
(img, perturbation, seed)
dataset_loaders/dataset_readers.py:346
↓ 1 callersFunctionaggreg_matches
(img1, list_matches)
mast3r/cloud_opt/sparse_ga.py:795
↓ 1 callersMethodaggregate_confs
(self, confs1, confs2)
mast3r/losses.py:474
↓ 1 callersFunctionarg_parser
()
dust3r/croco/datasets/crops/extract_crops_from_images.py:17
↓ 1 callersFunctionaverage_poses
Calculate the average pose, which is then used to center all poses using @center_poses. Its computation is as follows: 1. Compute the cen
utils/reverse_dfnet.py:9
↓ 1 callersFunctionbackproj
(K, depthmap, subsample)
mast3r/cloud_opt/sparse_ga.py:912
↓ 1 callersFunctionbatched_triangulate
(pts2d, # [B, Ncams, Npts, 2] proj_mats)
mast3r/cloud_opt/triangulation.py:12
↓ 1 callersMethodblockwise_criterion
(self, descs1, descs2, confs1, confs2, valid_matches, euc, rng=np.random, shuffle=True)
mast3r/losses.py:413
↓ 1 callersFunctionbuild_backbone
(config)
models/apr/backbone.py:55
↓ 1 callersFunctionbuild_scaling_rotation
(s, r)
utils/general_utils.py:100
↓ 1 callersMethodcalculate_image_quality_score
(self, brisque_features)
utils/brisque.py:118
↓ 1 callersMethodcalculate_mscn_coefficients
(self, image)
utils/brisque.py:125
↓ 1 callersFunctioncamera_matrix_of_crop
(input_camera_matrix, input_resolution, output_resolution, scaling=1, offset_factor=0.5, offset=None)
dust3r/dust3r/datasets/utils/cropping.py:87
↓ 1 callersFunctioncamera_to_JSON
(id, camera: Camera)
utils/camera_utils.py:91
↓ 1 callersFunctioncanonical_view
(ptmaps11, confs11, subsample, mode='avg-angle')
mast3r/cloud_opt/sparse_ga.py:817
↓ 1 callersFunctioncat_collate
r"""Custom collate function that concatenates stuff instead of stacking them, and handles NoneTypes
mast3r/utils/collate.py:28
↓ 1 callersMethodcat_tensors_to_optimizer
(self, tensors_dict)
models/gs/gaussian_model.py:551
↓ 1 callersFunctioncheck_if_same_size
(pairs)
dust3r/dust3r/inference.py:75
↓ 1 callersMethodclean_pointcloud
(self, **kw)
dust3r/dust3r/cloud_opt/base_opt.py:234
↓ 1 callersMethodclose
(self)
dust3r/datasets_preprocess/habitat/habitat_renderer/habitat_sim_envmaps_renderer.py:141
↓ 1 callersMethodcolor_transform
Photometric augmentation
dust3r/croco/stereoflow/augmentor.py:182
↓ 1 callersMethodcompute_AP
(label, score)
mast3r/losses.py:291
↓ 1 callersFunctioncompute_edge_scores
(edges, conf_i, conf_j)
dust3r/dust3r/cloud_opt/commons.py:24
↓ 1 callersMethodcompute_global_alignment
(self, init=None, niter_PnP=10, **kw)
dust3r/dust3r/cloud_opt/base_opt.py:276
↓ 1 callersMethodcompute_loss
(self, *args, **kwargs)
dust3r/dust3r/losses.py:94
↓ 1 callersFunctioncompute_min_spanning_tree
(pws)
mast3r/cloud_opt/sparse_ga.py:991
↓ 1 callersFunctioncompute_pairwise_products
(mscn_coefficients)
utils/brisque.py:86
↓ 1 callersFunctioncompute_pointmap
Compute a HxWx3 pointmap in camera frame from a HxW depth map.
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:27
↓ 1 callersMethodcompute_pointmap
(self, distancemap, position)
dust3r/datasets_preprocess/habitat/habitat_renderer/multiview_crop_generator.py:48
↓ 1 callersFunctioncompute_rotation_matrix_from_ortho6d
(poses)
utils/pose_utils.py:55
↓ 1 callersFunctioncondense_data
(imgs, tmp_paths, canonical_views, preds_21, dtype=torch.float32)
mast3r/cloud_opt/sparse_ga.py:729
↓ 1 callersMethodconstruct_list_of_attributes
(self)
models/gs/gaussian_model.py:300
↓ 1 callersFunctionconvert_dust3r_pairs_naming
(imgs, pairs_in)
mast3r/cloud_opt/sparse_ga.py:111
↓ 1 callersFunctionconvert_images
open and convert all images in a list or folder to proper input format for DUSt3R
dust3r/dust3r/utils/image.py:129
↓ 1 callersFunctionconvert_ndc_to_pinhole
(focal_length, principal_point, image_size)
dust3r/datasets_preprocess/preprocess_co3d.py:63
↓ 1 callersFunctionconvert_scene_metadata
(scene_subdir, intrinsics_dir, timestamps, quaternions, poses, poses_cam_to_world,
dust3r/datasets_preprocess/preprocess_arkitscenes.py:269
↓ 1 callersFunctioncosine_schedule
(t, lr_start, lr_end)
dust3r/dust3r/cloud_opt/commons.py:83
↓ 1 callersFunctioncreate_commandline
Create a commandline string to generate a scene.
dust3r/croco/datasets/habitat_sim/generate_multiview_images.py:119
↓ 1 callersFunctioncreate_dpt_head
return PixelwiseTaskWithDPT for given net params
dust3r/dust3r/heads/dpt_head.py:96
↓ 1 callersMethodcreate_embedding_fn
(self)
models/gs/embedder.py:12
↓ 1 callersMethodcreate_from_pcd
(self, pcd: BasicPointCloud, spatial_lr_scale: float, downsample_resolution=1)
models/gs/gaussian_model.py:155
↓ 1 callersFunctioncroco_args_from_ckpt
(ckpt)
dust3r/croco/models/croco_downstream.py:13
↓ 1 callersFunctioncrop_generator
()
dust3r/croco/stereoflow/engine.py:213
↓ 1 callersFunctioncrops_inference
(pairs, model, device, batch_size=48, verbose=True)
visloc.py:130
↓ 1 callersMethoddensify_and_clone
(self, grads, grad_threshold, scene_extent)
models/gs/gaussian_model.py:638
↓ 1 callersMethoddensify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size, prune_more)
models/gs/gaussian_model.py:660
↓ 1 callersMethoddensify_and_split
(self, grads, grad_threshold, scene_extent, N=2)
models/gs/gaussian_model.py:601
↓ 1 callersMethoddepth_to_pts3d
(self)
dust3r/dust3r/cloud_opt/base_opt.py:200
↓ 1 callersMethoddepth_to_pts3d
(self)
dust3r/dust3r/cloud_opt/optimizer.py:170
↓ 1 callersMethoddepth_to_pts3d
(self)
dust3r/dust3r/cloud_opt/modular_optimizer.py:130
↓ 1 callersFunctiondict_to_sparse_graph
(dic)
dust3r/dust3r/cloud_opt/init_im_poses.py:212
↓ 1 callersFunctiondisp_to_tensor
(disp)
dust3r/croco/stereoflow/datasets_stereo.py:48
↓ 1 callersMethoddistance
(self, a, b)
dust3r/dust3r/losses.py:50
← previousnext →401–500 of 1,383, ranked by callers