Functions — github.com/Inception3D/TTT3R

hub / github.com/Inception3D/TTT3R / functions

Functions1,064 in github.com/Inception3D/TTT3R

⨍ Functions 1,064◇ Types & classes 166

↓ 165 callersFunctionprint

(*args, **kwargs)

src/croco/utils/misc.py:207

↓ 64 callersMethodsize

(self)

src/dust3r/datasets/utils/cropping.py:45

↓ 53 callersFunctionimread_cv2

Open an image or a depthmap with opencv-python.

src/dust3r/utils/image.py:34

↓ 53 callersFunctionprint

(*args, **kwargs)

src/train.py:68

↓ 32 callersMethod_crop_resize_if_necessary

This function: - first downsizes the image with LANCZOS inteprolation, which is better than bilinear interpolation in

src/dust3r/datasets/base/base_multiview_dataset.py:443

↓ 32 callersMethodupdate

(self, **kwargs)

src/croco/utils/misc.py:96

↓ 26 callersFunctionadd_arg

( name_or_flags, default=None, default_stereo=None, default_flow=None, **kwargs )

src/croco/stereoflow/train.py:51

↓ 24 callersMethodmax

(self)

src/croco/utils/misc.py:74

↓ 22 callersMethodget_seq_from_start_id

args: num_views: number of views to return id_ref: the reference id (first id) ids_all: all the ids

src/dust3r/datasets/base/base_multiview_dataset.py:178

↓ 19 callersMethodget_img_and_ray_masks

(self, is_metric, v, rng, p=[0.8, 0.15, 0.05])

src/dust3r/datasets/base/base_multiview_dataset.py:291

↓ 18 callersMethodresize

(self, *args, **kwargs)

src/dust3r/datasets/utils/cropping.py:50

↓ 17 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 co

src/dust3r/utils/geometry.py:49

↓ 16 callersMethodmedian

(self)

src/croco/utils/misc.py:62

↓ 15 callersFunctioninvalid_to_nans

(arr, valid_mask, ndim=999)

src/dust3r/utils/misc.py:109

↓ 13 callersFunctionto_numpy

(x)

src/dust3r/utils/device.py:42

↓ 12 callersMethod_set_root

(self)

src/croco/stereoflow/datasets_stereo.py:138

↓ 10 callersFunctionbatch_append

(original_list, new_list)

src/train.py:647

↓ 10 callersFunctioninv

Invert a torch or numpy matrix

src/dust3r/utils/geometry.py:118

↓ 10 callersMethodresize

(self, *args, **kwargs)

eval/mv_recon/dataset_utils/cropping.py:49

↓ 9 callersMethod__init__

( self, in_features, hidden_features=None, out_features=None, act_laye

src/dust3r/blocks.py:62

↓ 9 callersMethod__init__

(self)

src/dust3r/losses.py:167

↓ 9 callersFunctiondepth_evaluation

Evaluate the depth map using various metrics and return a depth error parity map, with an option for least squares alignment. Args:

eval/video_depth/tools.py:126

↓ 8 callersMethodcrop

(self, *args, **kwargs)

src/dust3r/datasets/utils/cropping.py:53

↓ 7 callersMethod__init__

(self, max_gtnorm=None)

src/croco/stereoflow/criterion.py:25

↓ 7 callersMethod_set_root

(self)

src/croco/stereoflow/datasets_flow.py:120

↓ 7 callersFunctionmake_traj

(args)

eval/relpose/evo_utils.py:188

↓ 7 callersFunctionpostprocess

extract 3D points/confidence from prediction head output

src/dust3r/heads/postprocess.py:11

↓ 7 callersMethodstate_dict

(self)

src/croco/utils/misc.py:311

↓ 6 callersMethod__init__

( self, in_features, hidden_features=None, out_features=None, act_laye

src/croco/models/blocks.py:71

↓ 6 callersMethod_crop_resize_if_necessary

This function: - first downsizes the image with LANCZOS inteprolation, which is better than bilinear interpolation in

eval/mv_recon/base.py:167

↓ 6 callersFunction_resize_pil_image

(img, long_edge_size)

src/dust3r/utils/image.py:65

↓ 6 callersFunctioncolorize

turn a grayscale image into a color image :param x: torch.Tensor, grayscale image, [H, W] or [B, H, W] :param mask: torch.Tensor or None,

src/dust3r/viz.py:567

↓ 6 callersFunctiongroup_by_directory

Groups the file paths based on the second-to-last directory in their paths. Parameters: - pathes (list): List of file paths. Return

eval/video_depth/tools.py:14

↓ 6 callersMethodload_state_dict

(self, state_dict)

src/croco/utils/misc.py:317

↓ 6 callersFunctionprocess_scannet

(args, img_path)

eval/relpose/metadata.py:242

↓ 6 callersMethodset_epoch

(self, epoch)

src/dust3r/datasets/base/easy_dataset.py:166

↓ 5 callersMethod_downstream_head

(self, decout, img_shape, **kwargs)

src/dust3r/model.py:713

↓ 5 callersMethod_encode_ray_map

(self, ray_map, true_shape)

src/dust3r/model.py:536

↓ 5 callersMethod_get_img_level_feat

(self, feat)

src/dust3r/model.py:747

↓ 5 callersMethod_recurrent_rollout

( self, state_feat, state_pos, current_feat, current_pos, pose

src/dust3r/model.py:727

↓ 5 callersMethodclose

(self)

src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:250

↓ 5 callersFunctiondepth2disparity

(depth, return_mask=False)

eval/video_depth/tools.py:34

↓ 5 callersMethodinquire

(self, query, mem, return_attn=False)

src/dust3r/model.py:224

↓ 5 callersFunctioninvalid_to_zeros

(arr, valid_mask, ndim=999)

src/dust3r/utils/misc.py:118

↓ 5 callersMethodlog_every

( self, iterable, print_freq, accelerator: Accelerator, header=None, max_iter=None )

src/croco/utils/misc.py:131

↓ 5 callersFunctionpose_encoding_to_camera

Args: pose_encoding: A tensor of shape `BxC`, containing a batch of `B` `C`-dimensional pose encodings. p

src/dust3r/utils/camera.py:396

↓ 5 callersFunctionreg_dense_conf

extract confidence from prediction head output

src/dust3r/heads/postprocess.py:142

↓ 5 callersMethodrender_viewpoint

(self, viewpoint_position, viewpoint_orientation)

src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:332

↓ 5 callersFunctionto_numpy

(x)

viser_utils.py:56

↓ 5 callersMethodupdate_mem

mem_k: [B, size, C] mem_v: [B, size, C] feat_k: [B, 1, C] global_img_feat feat_v: [B, 1, C] out_pose_feat

src/dust3r/model.py:208

↓ 4 callersMethod__init__

(self, criterion, norm_mode="avg_dis", gt_scale=False, fix_first=True)

eval/mv_recon/criterion.py:277

↓ 4 callersFunction_check_input

(value, center=1, bound=(0, float("inf")), clip_first_on_zero=True)

eval/mv_recon/dataset_utils/transforms.py:13

↓ 4 callersFunction_check_input

(value, center=1, bound=(0, float("inf")), clip_first_on_zero=True)

src/dust3r/datasets/utils/transforms.py:14

↓ 4 callersFunction_get_gtnorm

(gt)

src/croco/stereoflow/criterion.py:13

↓ 4 callersMethod_init_state

Current Version: input the first frame img feature and pose to initialize the state feature and pose # [1, 768, 768] [1, 768, 2]

src/dust3r/model.py:718

↓ 4 callersFunction_read_img

(filename)

src/croco/stereoflow/datasets_stereo.py:826

↓ 4 callersFunctionadd_path_to_dust3r

(ckpt)

add_ckpt_path.py:6

↓ 4 callersFunctioncompute_pointcloud

Return a 3D point cloud corresponding to valid pixels of the depth map

src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:45

↓ 4 callersFunctionestimate_focal_knowing_depth

Reprojection method, for when the absolute depth is known: 1) estimate the camera focal using a robust estimator 2) reproject points onto true

src/dust3r/post_process.py:12

↓ 4 callersFunctionfloat2uint8

(x)

src/dust3r/viz.py:31

↓ 4 callersMethodfrom_pretrained

(cls, pretrained_model_name_or_path, **kw)

src/dust3r/model.py:315

↓ 4 callersMethodget_all_pts3d_t

(self, gts, preds, dist_clip=None)

eval/mv_recon/criterion.py:283

↓ 4 callersFunctionget_group_pointcloud_center_scale

(pts, valid_masks=None, z_only=False, center=True)

src/dust3r/utils/geometry.py:474

↓ 4 callersMethodget_norm_factor_point_cloud

( self, pts_self, pts_cross, valids, conf_self, conf_cross, norm_self_only=False )

src/dust3r/losses.py:319

↓ 4 callersMethodget_norm_factor_poses

(self, gt_trans, pr_trans, not_metric_mask)

src/dust3r/losses.py:335

↓ 4 callersFunctionget_render_results

(gts, preds, self_view=False)

src/dust3r/utils/render.py:54

↓ 4 callersFunctionimg_to_tensor

(img)

src/croco/stereoflow/datasets_stereo.py:45

↓ 4 callersMethodinit

(self, dim_tokens_enc=768)

src/dust3r/heads/dpt_head.py:32

↓ 4 callersFunctionlist_scenes

Generic method iterating through a base_path folder to find scenes.

src/croco/datasets/habitat_sim/paths.py:97

↓ 4 callersFunctionmake_fusion_block

(features, use_bn, width_ratio=1)

src/croco/models/dpt_block.py:235

↓ 4 callersFunctionmy_formatting

(val)

src/croco/datasets/habitat_sim/generate_multiview_images.py:165

↓ 4 callersMethodnormalize

(self, x, mask)

src/dust3r/losses.py:239

↓ 4 callersFunctionnormalize_pointcloud_group

renorm pointmaps pts1, pts2 with norm_mode

src/dust3r/utils/geometry.py:329

↓ 4 callersFunctionpostprocess_pose

extract pose from prediction head output

src/dust3r/heads/postprocess.py:30

↓ 4 callersFunctionpostprocess_rgb

(out, eps=1e-6)

src/dust3r/heads/postprocess.py:23

↓ 4 callersFunctionsave_model

(epoch, fname, best_so_far)

src/train.py:248

↓ 4 callersFunctionto_cpu

(x)

src/dust3r/utils/device.py:46

↓ 4 callersFunctionunravel_xy

(pos, shape)

eval/mv_recon/dataset_utils/corr.py:38

↓ 4 callersFunctionunravel_xy

(pos, shape)

src/dust3r/datasets/utils/corr.py:41

↓ 4 callersFunctionview_name

(view, batch_index=None)

eval/mv_recon/base.py:245

↓ 4 callersFunctionview_name

(view, batch_index=None)

src/dust3r/datasets/base/base_multiview_dataset.py:507

↓ 3 callersMethod__init__

( self, hidden_size=768, mlp_ratio=4, pose_encoding_type="absT_quaR", )

src/dust3r/utils/camera.py:14

↓ 3 callersMethod__init__

( self, net, has_conf=False, has_depth=False, has_rgb=False, has_pose_conf=False )

src/dust3r/heads/linear_head.py:31

↓ 3 callersMethod__init__

Init. Args: scale_factor (float): scaling mode (str): interpolation mode

src/croco/models/dpt_block.py:250

↓ 3 callersFunction_crop

(img, sy, sx)

src/croco/stereoflow/engine.py:360

↓ 3 callersMethod_encode_image

(self, image, true_shape)

src/dust3r/model.py:525

↓ 3 callersFunction_read_png_disp

(filename, coef=1.0)

src/croco/stereoflow/datasets_stereo.py:838

↓ 3 callersMethodadd_camera

(self, step)

viser_utils.py:834

↓ 3 callersMethodadd_meter

(self, name, meter)

src/croco/utils/misc.py:128

↓ 3 callersFunctionadd_scene_cam

( scene, pose_c2w, edge_color, image=None, focal=None, imsize=None, screen_width=0

src/dust3r/viz.py:309

↓ 3 callersMethodbackward

(ctx, grad_res)

src/croco/models/curope/curope2d.py:25

↓ 3 callersFunctioncamera_to_pose_encoding

Inverse to pose_encoding_to_camera camera: opencv, cam2world

src/dust3r/utils/camera.py:345

↓ 3 callersFunctioncompute_camera_pose_opencv_convention

(camera_position, camera_orientation)

src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:26

↓ 3 callersFunctiondepth_read

Read depth data from file, return as numpy array.

eval/video_depth/eval_depth.py:41

↓ 3 callersFunctiondepthmap_to_absolute_camera_coordinates

Args: - depthmap (HxW array): - camera_intrinsics: a 3x3 matrix - camera_pose: a 4x3 or 4x4 cam2world matrix Returns:

src/dust3r/utils/geometry.py:209

↓ 3 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

src/dust3r/utils/geometry.py:127

↓ 3 callersMethodget_all_pts3d

( self, gts, preds, dist_clip=None, norm_self_only=False, norm

src/dust3r/losses.py:377

↓ 3 callersFunctionget_video_results

()

eval/video_depth/eval_depth.py:99

↓ 3 callersFunctioninference_recurrent_lighter

(groups, model, device, verbose=True)

src/dust3r/inference.py:291

↓ 3 callersFunctionload_images

open and convert all images in a list or folder to proper input format for DUSt3R

src/dust3r/utils/image.py:75

next →1–100 of 1,064, ranked by callers