MCPcopy Create free account

hub / github.com/CUT3R/CUT3R / functions

Functions1,487 in github.com/CUT3R/CUT3R

↓ 2 callersMethodget_conf
(self, mode=None)
cloud_opt/dust3r_opt/base_opt.py:311
↓ 2 callersMethodget_conf_log
(self, x)
src/dust3r/losses.py:1067
↓ 2 callersFunctionget_conf_trf
(mode)
cloud_opt/dust3r_opt/commons.py:48
↓ 2 callersMethodget_focals
(self)
cloud_opt/dust3r_opt/optimizer.py:166
↓ 2 callersMethodget_intrinsics
(self)
cloud_opt/dust3r_opt/optimizer.py:185
↓ 2 callersFunctionget_joint_pointcloud_center_scale
(pts, valid_masks=None, z_only=False, center=True)
eval/mv_recon/criterion.py:254
↓ 2 callersFunctionget_joint_pointcloud_depth
(zs, valid_masks=None, quantile=0.5)
eval/mv_recon/criterion.py:235
↓ 2 callersFunctionget_known_poses
(self)
cloud_opt/dust3r_opt/init_im_poses.py:349
↓ 2 callersMethodget_masks
(self)
cloud_opt/dust3r_opt/base_opt.py:287
↓ 2 callersFunctionget_med_dist_between_poses
(poses)
src/dust3r/utils/geometry.py:510
↓ 2 callersMethodget_name
(self)
eval/mv_recon/criterion.py:50
↓ 2 callersMethodget_name
(self)
src/dust3r/losses.py:118
↓ 2 callersMethodget_name
(self)
src/dust3r/losses.py:1064
↓ 2 callersFunctionget_norm_factor
(pts, norm_mode="avg_dis", valids=None, fix_first=True)
eval/mv_recon/criterion.py:168
↓ 2 callersFunctionget_patch_embed
(patch_embed_cls, img_size, patch_size, enc_embed_dim, in_chans=3)
src/dust3r/patch_embed.py:12
↓ 2 callersMethodget_principal_points
(self)
cloud_opt/dust3r_opt/base_opt.py:308
↓ 2 callersMethodget_principal_points
(self)
cloud_opt/dust3r_opt/optimizer.py:182
↓ 2 callersMethodget_pw_norm_scale_factor
(self)
cloud_opt/dust3r_opt/base_opt.py:266
↓ 2 callersMethodget_ray_map
(self, c2w, h, w, intrinsics=None)
viser_utils.py:453
↓ 2 callersFunctionget_timestamp
Extract the timestamp from an image filename. Assumes the timestamp is the last underscore-separated token in the name (before the file exten
datasets_preprocess/generate_set_arkitscenes.py:21
↓ 2 callersFunctionget_timestamp
Convert an image name to a timestamp (integer). If the image name starts with 'DSC', the timestamp is the integer part after 'DSC'. Othe
datasets_preprocess/generate_set_scannetpp.py:27
↓ 2 callersFunctionget_tum_poses
poses: list of 4x4 arrays
eval/relpose/utils.py:74
↓ 2 callersFunctionget_vis_imgs_new
(loss_details, num_imgs_vis, num_views, is_metric)
src/train.py:767
↓ 2 callersFunctionglobal_alignment_iter
( net, cur_iter, niter, lr_base, lr_min, optimizer, schedule, temporal_smoothi
cloud_opt/base_opt.py:259
↓ 2 callersFunctionglobal_alignment_iter
(net, cur_iter, niter, lr_base, lr_min, optimizer, schedule)
cloud_opt/dust3r_opt/base_opt.py:563
↓ 2 callersFunctionimg_to_arr
(img)
src/dust3r/utils/image.py:28
↓ 2 callersMethodinit
Initialize parts of decoder that are dependent on dimension of encoder tokens. Should be called when setting up MultiMAE. :p
src/croco/models/dpt_block.py:380
↓ 2 callersFunctioninv
Invert a torch or numpy matrix
cloud_opt/utils.py:335
↓ 2 callersFunctionis_dist_avail_and_initialized
()
src/croco/utils/misc.py:218
↓ 2 callersMethodis_other_pointcloud_overlapping
Check if a viewpoint is valid and overlaps significantly with a reference one.
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:301
↓ 2 callersMethodjitter_one
( self, img, fn_idx, brightness_factor, contrast_factor, satur
src/croco/datasets/transforms.py:47
↓ 2 callersFunctionlinear_schedule
(t, lr_start, lr_end)
cloud_opt/utils.py:424
↓ 2 callersFunctionlistify
(elems)
demo_ga.py:46
↓ 2 callersFunctionlistify
(elems)
src/dust3r/utils/device.py:87
↓ 2 callersFunctionload_image
(impath)
src/croco/datasets/pairs_dataset.py:11
↓ 2 callersFunctionload_sfm
(sfm_dir, cam_type="dslr")
datasets_preprocess/preprocess_scannetpp.py:75
↓ 2 callersFunctionload_timestamps
(time_file, traj_format="replica")
eval/relpose/evo_utils.py:141
↓ 2 callersFunctionload_traj
Read trajectory format. Return in TUM-RGBD format. Returns: traj_tum (N, 7): camera to world poses in (x,y,z,qx,qy,qz,qw) timestam
eval/relpose/evo_utils.py:102
↓ 2 callersFunctionlook_at_for_habitat
(eye, center, up, return_cam2world=True)
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:119
↓ 2 callersFunctionmake_batch_symmetric
(batch)
src/dust3r/inference.py:49
↓ 2 callersFunctionmerge_chunk_dict
(old_dict, curr_dict, add_number)
src/dust3r/inference.py:20
↓ 2 callersFunctionnorm
(p)
eval/mv_recon/dataset_utils/corr.py:96
↓ 2 callersFunctionnorm
(p)
src/dust3r/datasets/utils/corr.py:102
↓ 2 callersFunctionnormalize_pointcloud_t
( pts, norm_mode="avg_dis", valids=None, fix_first=True, gt=False )
eval/mv_recon/criterion.py:208
↓ 2 callersFunctionopencv_to_colmap_intrinsics
Convert OpenCV intrinsics (with pixel centers at (0, 0)) to COLMAP convention.
datasets_preprocess/preprocess_mvimgnet.py:90
↓ 2 callersFunctionopencv_to_colmap_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)
src/dust3r/utils/geometry.py:248
↓ 2 callersFunctionparse_extrinsics
Parse the extrinsics file to extract the intrinsics and pose matrices. Args: file_path (str): The path to the file containing the extrin
datasets_preprocess/preprocess_unreal4k.py:34
↓ 2 callersMethodparse_pc_data
( self, pc, color, conf=None, edge_color=[0.251, 0.702, 0.902],
viser_utils.py:622
↓ 2 callersFunctionplot_trajectory
( pred_traj, gt_traj=None, title="", filename="", align=True, correct_scale=True )
eval/relpose/evo_utils.py:333
↓ 2 callersFunctionprepare_crop
(img, rect, rot=0)
src/croco/datasets/crops/extract_crops_from_images.py:141
↓ 2 callersFunctionreadNpyFlow
read numpy array from file. filepath: file to read from returns: numpy array
datasets_preprocess/flow_IO.py:225
↓ 2 callersFunctionreadPfmFile
adapted from https://lmb.informatik.uni-freiburg.de/resources/datasets/SceneFlowDatasets.en.html
datasets_preprocess/flow_IO.py:273
↓ 2 callersFunctionread_model
(path, ext="")
datasets_preprocess/read_write_model.py:488
↓ 2 callersFunctionrelative_pose_absT_quatR
Compute the relative translation and quaternion between two poses.
src/dust3r/utils/camera.py:453
↓ 2 callersFunctionrender
( intrinsics: torch.Tensor, pts3d: torch.Tensor, rgbs: torch.Tensor | None = None, scale: floa
src/dust3r/utils/render.py:6
↓ 2 callersMethodreorg
(self, ls_b, masks_b)
src/dust3r/losses.py:888
↓ 2 callersFunctionreproject_view
(pts3d, view2)
eval/mv_recon/dataset_utils/corr.py:11
↓ 2 callersFunctionreproject_view
(pts3d, view2)
src/dust3r/datasets/utils/corr.py:12
↓ 2 callersFunctionrescale_flow
Resize an optical flow field to a new resolution and scale its vectors accordingly. Args: flow (np.ndarray): Flow array of shape [H,
datasets_preprocess/preprocess_spring.py:34
↓ 2 callersMethodresize
(self, *args, **kwargs)
datasets_preprocess/preprocess_mvimgnet.py:73
↓ 2 callersMethodrun
(self)
viser_utils.py:774
↓ 2 callersMethodsample_random_viewpoint
Sample a random viewpoint using the navmesh
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:259
↓ 2 callersMethodselect_scene
(self, scene, *instances, opposite=False)
src/dust3r/datasets/megadepth.py:44
↓ 2 callersFunctionselout
(ar)
src/dust3r/utils/misc.py:82
↓ 2 callersMethodsetup
(self, croconet)
src/croco/models/head_downstream.py:44
↓ 2 callersFunctionsetup_for_distributed
This function disables printing when not in master process
src/croco/utils/misc.py:201
↓ 2 callersFunctionshow_raw_pointcloud
(pts3d, colors, point_size=2)
src/dust3r/viz.py:45
↓ 2 callersFunctionshuffle_deque
(dq, seed=None)
eval/mv_recon/data.py:13
↓ 2 callersFunctionsplit_prediction_conf
(predictions, with_conf=False)
src/croco/stereoflow/engine.py:18
↓ 2 callersMethodsynchronize_between_processes
(self, accelerator)
src/croco/utils/misc.py:124
↓ 2 callersFunctiontiled_pred
( model, criterion, img1, img2, gt, overlap=0.5, bad_crop_thr=0.05, downscale=
src/croco/stereoflow/engine.py:243
↓ 2 callersMethodto_pil
(self)
datasets_preprocess/preprocess_mvimgnet.py:64
↓ 2 callersFunctiontransposed
(dic)
src/dust3r/utils/misc.py:105
↓ 2 callersFunctionuint8
(colors)
src/dust3r/viz.py:412
↓ 2 callersFunctionunreal2cv2
(points)
datasets_preprocess/preprocess_bedlam.py:53
↓ 2 callersMethodupdate
(self, value, n=1)
src/croco/utils/misc.py:43
↓ 2 callersFunctionwriteFlowKitti
(filename, uv)
src/croco/stereoflow/datasets_flow.py:756
↓ 2 callersFunctionwriteNpyFile
write numpy array to file. arr: numpy array to write filepath: file to write to
datasets_preprocess/flow_IO.py:233
↓ 2 callersFunctionwritePFM
(file, image, scale=1)
src/croco/stereoflow/datasets_stereo.py:924
↓ 2 callersFunctionwrite_model
(cameras, images, points3D, path, ext=".bin")
datasets_preprocess/read_write_model.py:510
↓ 1 callersFunctionSum
(losses, masks, conf=None)
eval/mv_recon/criterion.py:154
↓ 1 callersMethod__init__
(self, net, has_conf=False, has_rgb=False, has_pose=False)
src/dust3r/heads/dpt_head.py:139
↓ 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_i
src/croco/models/croco_downstream.py:78
↓ 1 callersMethod__init__
(self, config: CrocoConfig)
src/croco/models/croco.py:64
↓ 1 callersMethod__init__
(self, *args, **kwargs)
cloud_opt/base_opt.py:23
↓ 1 callersMethod__init__
(self, *args, **kwargs)
cloud_opt/dust3r_opt/base_opt.py:92
↓ 1 callersFunction_append_camera_parameters
Add camera parameters to the observation dictionnary produced by Habitat-Sim In-place modifications.
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:80
↓ 1 callersMethod_build_cache
(self)
src/croco/stereoflow/datasets_flow.py:440
↓ 1 callersMethod_build_cache
(self)
src/croco/stereoflow/datasets_stereo.py:287
↓ 1 callersFunction_check_edges
(edges)
cloud_opt/utils.py:68
↓ 1 callersMethod_check_edges
(self)
cloud_opt/dust3r_opt/base_opt.py:212
↓ 1 callersFunction_compute_img_conf
(imshapes, device, edges, edge2conf_i, edge2conf_j)
cloud_opt/utils.py:135
↓ 1 callersMethod_compute_img_conf
(self, pred1_conf, pred2_conf)
cloud_opt/dust3r_opt/base_opt.py:218
↓ 1 callersMethod_compute_metrics
(self)
src/croco/stereoflow/criterion.py:220
↓ 1 callersMethod_compute_metrics
(self)
src/croco/stereoflow/criterion.py:333
↓ 1 callersFunction_crop_image
(intrinsics_in, color_image_in, depthmap_in, resolution_out=(800, 800))
datasets_preprocess/preprocess_blendedmvs.py:107
↓ 1 callersMethod_decoder
(self, f_state, pos_state, f_img, pos_img, f_pose, pos_pose)
src/dust3r/model.py:660
↓ 1 callersFunction_downscale_image
(camera_intrinsics, image, depthmap, resolution_out=(512, 384))
datasets_preprocess/preprocess_megadepth.py:122
↓ 1 callersMethod_encode_state
(self, image_tokens, image_pos)
src/dust3r/model.py:538
↓ 1 callersFunction_fast_depthmap_to_pts3d
(depth, pixel_grid, focal, pp)
cloud_opt/dust3r_opt/optimizer.py:253
← previousnext →201–300 of 1,487, ranked by callers