MCPcopy Create free account

hub / github.com/ai4ce/RAP / functions

Functions1,383 in github.com/ai4ce/RAP

⨍ Functions 1,383◇ Types & classes 201
↓ 2 callersFunctionoff_diagonal
(mat)
utils/vic_reg_loss.py:6
↓ 2 callersFunctionoptimize_loop
(loss_func, lr_base, niter, pix_loss, lr_end=0)
mast3r/cloud_opt/sparse_ga.py:393
↓ 2 callersFunctionperturb_rotation
(pose, theta, phi, psi=0)
utils/nvs_utils.py:51
↓ 2 callersFunctionprepare_crop
(img, rect, rot=0)
dust3r/croco/datasets/crops/extract_crops_from_images.py:120
↓ 2 callersFunctionprocess_pose_file
(path)
refine.py:50
↓ 2 callersMethodproject
(self, rays)
dust3r/datasets_preprocess/habitat/habitat_renderer/projections.py:102
↓ 2 callersMethodprune_points
(self, mask)
models/gs/gaussian_model.py:528
↓ 2 callersFunctionpts3d_to_trimesh
(img, pts3d, valid=None)
dust3r/dust3r/viz.py:38
↓ 2 callersMethodq_to_R
(self, q)
utils/pose_utils.py:167
↓ 2 callersFunctionquaternion_to_rotation_matrix
(qw, qx, qy, qz)
refine.py:33
↓ 2 callersFunctionqvec2rotmat
(qvec)
utils/read_write_model.py:523
↓ 2 callersFunctionqvec2rotmat
(qvec)
dataset_loaders/colmap_loader.py:43
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png", data_perturb=None, sp
dataset_loaders/dataset_readers.py:254
↓ 2 callersFunctionread_alignments
(path_to_alignment)
dust3r/dust3r_visloc/datasets/inloc.py:23
↓ 2 callersFunctionreg_dense_depth
extract 3D points from prediction head output
dust3r/dust3r/heads/postprocess.py:22
↓ 2 callersFunctionrender_interpolation
(args, name, iteration, views, gaussians, background, select_idxs=None)
render.py:32
↓ 2 callersFunctionrender_intrinsics
(model_path, name, iteration, views, gaussians, pipeline, background)
render.py:184
↓ 2 callersFunctionrender_lego
(model_path, name, iteration, views, view0, gaussians, pipeline, background)
render.py:138
↓ 2 callersFunctionrender_multiview_video
(args, name, train_views, gaussians, background)
render.py:66
↓ 2 callersMethodrender_perturbed_imgs
(self, name, poses, indexes=None, disable_tqdm=False)
utils/nvs_utils.py:391
↓ 2 callersMethodreport
(self)
refine.py:188
↓ 2 callersFunctionreproject_view
(pts3d, view2)
mast3r/datasets/utils/cropping.py:77
↓ 2 callersFunctionreverse_poses
Reverse the transformations applied to the poses and reconstruct the original poses. Inputs: all_poses: (N_images, 4, 4) the transfo
utils/reverse_dfnet.py:40
↓ 2 callersFunctionrot_phi
Generate a rotation matrix for rotation around the X-axis.
utils/reverse_dfnet.py:53
↓ 2 callersMethodsample_random_viewpoint
Sample a random viewpoint using the navmesh
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:227
↓ 2 callersFunctionselect_pairs_of_crops
(img_q, img_b, pos2d_in_query, pos2d_in_ref, maxdim=512, overlap=.5, forced_resolution=None)
mast3r/utils/coarse_to_fine.py:184
↓ 2 callersMethodselect_scene
(self, scene, *instances, opposite=False)
dust3r/dust3r/datasets/megadepth.py:43
↓ 2 callersFunctionselout
(ar)
dust3r/dust3r/utils/misc.py:82
↓ 2 callersMethodset_resolution
(self, model)
dust3r/dust3r_visloc/datasets/base_dataset.py:11
↓ 2 callersMethodsetup
(self, croconet)
dust3r/croco/models/head_downstream.py:36
↓ 2 callersFunctionsetup_for_distributed
This function disables printing when not in master process
dust3r/croco/utils/misc.py:176
↓ 2 callersMethodshow
(self, show_cams=True)
mast3r/cloud_opt/sparse_ga.py:104
↓ 2 callersFunctionshow_raw_pointcloud
(pts3d, colors, point_size=2)
dust3r/dust3r/viz.py:29
↓ 2 callersFunctionskew_sym_mat
(x)
utils/pose_utils.py:471
↓ 2 callersFunctionsplit_prediction_conf
(predictions, with_conf=False)
dust3r/croco/stereoflow/engine.py:18
↓ 2 callersFunctionsquares_mean
(x)
utils/brisque.py:27
↓ 2 callersMethodstate_dict
(self, trainable=True)
dust3r/dust3r/cloud_opt/base_opt.py:123
↓ 2 callersFunctionstorePly
(path, xyz, rgb)
dataset_loaders/dataset_readers.py:135
↓ 2 callersMethodtaylor_C
(self, x, nth=10)
utils/pose_utils.py:156
↓ 2 callersFunctiontiled_pred
(model, criterion, img1, img2, gt, overlap=0.5, bad_crop_thr=0.05, downscale=Fal
dust3r/croco/stereoflow/engine.py:179
↓ 2 callersFunctiontraining_report
(tb_writer, iteration, Ll1, loss, elapsed, test_iterations, scene: Scene, renderFunc, renderArgs)
gs.py:414
↓ 2 callersFunctiontransposed
(dic)
dust3r/dust3r/utils/misc.py:99
↓ 2 callersFunctionuint8
(colors)
dust3r/dust3r/viz.py:336
↓ 2 callersMethodunpatchify
x: (N, L, patch_size**2 *channels) imgs: (N, 3, H, W)
dust3r/croco/models/croco.py:218
↓ 2 callersMethodupdate
(self, value, n=1)
dust3r/croco/utils/misc.py:40
↓ 2 callersMethodvisualize_matches
visualize a few matches
matcher.py:198
↓ 2 callersFunctionwith_pos_embed
(tensor, pos: Optional[Tensor])
models/apr/transformer_encoder.py:103
↓ 2 callersFunctionwriteFlowKitti
(filename, uv)
dust3r/croco/stereoflow/datasets_flow.py:467
↓ 2 callersFunctionwritePFM
(file, image, scale=1)
dust3r/croco/stereoflow/datasets_stereo.py:618
↓ 2 callersFunctionwrite_scene_list
(scenes, n, fpath)
dust3r/datasets_preprocess/habitat/find_scenes.py:55
↓ 1 callersFunctionK_from_colmap
(elems)
dust3r/dust3r_visloc/datasets/base_colmap.py:79
↓ 1 callersFunctionRGB2SH
(rgb)
utils/sh_utils.py:112
↓ 1 callersFunctionRT2QT
processes the 1x12 raw pose from dataset by aligning and then normalizing :param poses_in: N x 12 :param mean_t: 3 :param std_t: 3
dataset_loaders/seven_scenes.py:16
↓ 1 callersMethodR_cam_to_world
(self)
dust3r/datasets_preprocess/habitat/habitat_renderer/habitat_sim_envmaps_renderer.py:169
↓ 1 callersFunctionSE3_exp
(tau)
utils/pose_utils.py:520
↓ 1 callersFunctionSH2RGB
(sh)
utils/sh_utils.py:115
↓ 1 callersFunctionSO3_exp
(theta)
utils/pose_utils.py:484
↓ 1 callersMethodSO3_to_so3
(self, R, eps=1e-7)
utils/pose_utils.py:86
↓ 1 callersFunctionV
(theta)
utils/pose_utils.py:502
↓ 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, args)
rap.py:242
↓ 1 callersMethod__init__
(self, args)
refine.py:91
↓ 1 callersMethod__init__
(self, configs, cam_params, hw, dl, start_epoch, *args, **kwargs)
utils/nvs_utils.py:314
↓ 1 callersMethod__init__
(self, uid, colmap_id, image_name, R, T, K, FoVx, FoVy, image, scale=1, render_device="cuda", data_device="cud
utils/cameras.py:20
↓ 1 callersMethod__init__
(self, in_dim=1024, hidden_dim=512, out_dim=128)
utils/vic_reg_loss.py:13
↓ 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/dust3r/cloud_opt/base_opt.py:33
↓ 1 callersMethod__init__
Build network for binocular downstream task It takes an extra argument head, that is called with the features and a dictionary img
dust3r/croco/models/croco_downstream.py:69
↓ 1 callersMethod__init__
(self, num_pos_feats=256, device="cuda")
models/apr/pencoder.py:49
↓ 1 callersFunction_append_camera_parameters
Add camera parameters to the observation dictionnary produced by Habitat-Sim In-place modifications.
dust3r/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:69
↓ 1 callersMethod_build_cache
(self)
dust3r/croco/stereoflow/datasets_flow.py:265
↓ 1 callersMethod_build_cache
(self)
dust3r/croco/stereoflow/datasets_stereo.py:215
↓ 1 callersMethod_check_edges
(self)
dust3r/dust3r/cloud_opt/base_opt.py:130
↓ 1 callersMethod_compute_img_conf
(self, pred1_conf, pred2_conf)
dust3r/dust3r/cloud_opt/base_opt.py:136
↓ 1 callersMethod_compute_metrics
(self)
dust3r/croco/stereoflow/criterion.py:172
↓ 1 callersMethod_compute_metrics
(self)
dust3r/croco/stereoflow/criterion.py:238
↓ 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/dust3r/demo.py:66
↓ 1 callersFunction_crop_image
(intrinsics_in, color_image_in, depthmap_in, resolution_out=(800, 800))
dust3r/datasets_preprocess/preprocess_blendedMVS.py:89
↓ 1 callersFunction_crop_image
(intrinsics_in, color_image_in, depthmap_in, resolution_out=(512, 512))
dust3r/datasets_preprocess/preprocess_staticthings3d.py:88
↓ 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_decoder
(self, f1, pos1, f2, pos2)
dust3r/dust3r/model.py:171
↓ 1 callersFunction_downscale_image
(camera_intrinsics, image, depthmap, resolution_out=(512, 384))
dust3r/datasets_preprocess/preprocess_megadepth.py:95
↓ 1 callersMethod_encode_symmetrized
(self, view1, view2)
dust3r/dust3r/model.py:152
↓ 1 callersFunction_fast_depthmap_to_pts3d
(depth, pixel_grid, focal, pp)
dust3r/dust3r/cloud_opt/optimizer.py:204
↓ 1 callersFunction_get_activation_fn
Return an activation function given a string
models/apr/transformer_encoder.py:167
↓ 1 callersFunction_get_clones
(module, N)
models/apr/transformer_encoder.py:163
↓ 1 callersMethod_get_confs_with_normals
(self)
mast3r/cloud_opt/tsdf_optimizer.py:129
↓ 1 callersMethod_get_depthpath
(self, obj, instance, view_idx)
dust3r/dust3r/datasets/co3d.py:54
↓ 1 callersMethod_get_impath
(self, obj, instance, view_idx)
dust3r/dust3r/datasets/co3d.py:51
↓ 1 callersMethod_get_maskpath
(self, obj, instance, view_idx)
dust3r/dust3r/datasets/co3d.py:57
↓ 1 callersMethod_get_metadatapath
(self, obj, instance, view_idx)
dust3r/dust3r/datasets/co3d.py:48
↓ 1 callersFunction_get_num_layer_for_vit
(var_name, enc_depth, dec_depth)
dust3r/croco/utils/misc.py:368
↓ 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_view_map
(self, imgname, idx)
dust3r/dust3r_visloc/datasets/base_colmap.py:220
↓ 1 callersMethod_get_view_query
(self, imgname)
dust3r/dust3r_visloc/datasets/base_colmap.py:174
↓ 1 callersMethod_get_views
(self, idx, resolution, rng)
dust3r/dust3r/datasets/co3d.py:65
↓ 1 callersMethod_get_views
(self, idx, resolution, rng)
dust3r/dust3r/datasets/base/base_stereo_view_dataset.py:60
↓ 1 callersMethod_init_from_views
(self, view1, view2, pred1, pred2, dist='l1', conf='log',
dust3r/dust3r/cloud_opt/base_opt.py:44
← previousnext →301–400 of 1,383, ranked by callers