Functions — github.com/Inception3D/TTT3R

hub / github.com/Inception3D/TTT3R / functions

Functions1,064 in github.com/Inception3D/TTT3R

⨍ Functions 1,064◇ Types & classes 166

Function_read_crestereo_disp

(filename)

src/croco/stereoflow/datasets_stereo.py:857

Method_read_depthmap

(self, depthpath, input_metadata)

src/dust3r/datasets/wildrgbd.py:44

Function_read_hdf5_disp

(filename)

src/croco/stereoflow/datasets_stereo.py:876

Function_read_hdf5_flow

(filename)

src/croco/stereoflow/datasets_flow.py:768

Function_read_kitti_disp

(filename)

src/croco/stereoflow/datasets_stereo.py:865

Function_read_kitti_flow

(filename)

src/croco/stereoflow/datasets_flow.py:742

Function_read_middlebury20052006_disp

(filename)

src/croco/stereoflow/datasets_stereo.py:861

Function_read_npy_disp

(filename)

src/croco/stereoflow/datasets_stereo.py:853

Function_read_numpy_flow

(filename)

src/croco/stereoflow/datasets_flow.py:641

Function_read_pfm_disp

(filename)

src/croco/stereoflow/datasets_stereo.py:845

Function_read_pfm_flow

(filename)

src/croco/stereoflow/datasets_flow.py:645

Method_resolutions

(self)

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

Method_resolutions

(self)

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

Method_resolutions

(self)

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

Method_set_decoder

( self, enc_embed_dim, dec_embed_dim, dec_num_heads, dec_depth,

src/dust3r/model.py:337

Method_set_decoder

No decoder

src/croco/models/croco_downstream.py:53

Method_set_mask_generator

No mask generator

src/croco/models/croco_downstream.py:44

Method_set_mask_generator

No mask generator

src/croco/models/croco_downstream.py:88

Method_set_mask_token

No mask token

src/croco/models/croco_downstream.py:48

Method_set_mask_token

No mask token

src/croco/models/croco_downstream.py:92

Method_set_mask_token

(self, dec_embed_dim)

src/croco/models/croco.py:153

Method_set_patch_embed

(self, img_size=224, patch_size=16, enc_embed_dim=768)

src/dust3r/model.py:329

Method_set_prediction_head

No prediction head

src/dust3r/model.py:483

Method_set_prediction_head

No 'prediction head' for downstream tasks.

src/croco/models/croco_downstream.py:57

Method_set_prediction_head

No prediction head for downstream tasks, define your own head

src/croco/models/croco_downstream.py:97

Functionabsolute_error_loss

(params, predicted_depth, ground_truth_depth)

eval/video_depth/tools.py:47

Methodadd_batch

(self, predictions, gt)

src/croco/stereoflow/criterion.py:189

Methodadd_cameras

( self, poses, focals=None, images=None, imsizes=None, colors=None, **kw )

src/dust3r/viz.py:251

Methodadd_rgbd

( self, image, depth, intrinsics=None, cam2world=None, zfar=np.inf, mask=None )

src/dust3r/viz.py:149

Functionadjust_learning_rate

Decay the learning rate with half-cycle cosine after warmup

src/croco/utils/misc.py:578

Functionall_reduce_mean

Use accelerator to all-reduce and compute mean.

src/croco/utils/misc.py:442

Functionauto_cam_size

(im_poses)

src/dust3r/viz.py:141

Methodavg

(self)

src/croco/utils/misc.py:66

Functionbbox_from_intrinsics_in_out

( input_camera_matrix, output_camera_matrix, output_resolution )

eval/mv_recon/dataset_utils/cropping.py:136

Functionbbox_from_intrinsics_in_out

( input_camera_matrix, output_camera_matrix, output_resolution )

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

Functioncheck_if_same_size

(pairs)

src/dust3r/inference.py:302

Functioncollate_with_cat

(whatever, lists=False)

src/dust3r/utils/device.py:54

FunctioncolorTest

flow_utils.colorTest(): display an example of image showing the color encoding scheme

src/croco/stereoflow/datasets_flow.py:784

Functioncolorize

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,

viser_utils.py:294

Functioncolorize_feature_map

Args: x: torch.Tensor, [B, H, W, D] Returns: torch.Tensor, [B, H, W, 3]

src/dust3r/viz.py:1059

Functioncompletion_ratio

(gt_points, rec_points, dist_th=0.05)

eval/mv_recon/utils.py:6

Methodcompute_frame_loss

(self, gts, preds, **kw)

eval/mv_recon/criterion.py:422

Functioncompute_iou

(pred_vox, target_vox)

eval/mv_recon/utils.py:44

Methodcompute_loss

(self, gts, preds, **kw)

src/dust3r/losses.py:205

Methodcompute_loss

(self, gts, preds, **kw)

src/dust3r/losses.py:762

Methodcompute_loss

(self, gts, preds, **kw)

src/dust3r/losses.py:898

Methodcompute_loss

(self, gts, preds, **kw)

src/dust3r/losses.py:1070

Methodcompute_relative_pose_loss

( self, gt_trans, gt_quats, pr_trans, pr_quats, masks=None )

src/dust3r/losses.py:697

Functionconvert_scene_output_to_glb

( outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05, show_cam=True,

viser_utils.py:101

Functioncrop_image_depthmap

Return a crop of the input view.

eval/mv_recon/dataset_utils/cropping.py:119

Functioncrop_image_depthmap

Return a crop of the input view.

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

Functioncustom_sort_key

(key)

src/dust3r/inference.py:11

Functiondepth_read

(filename)

datasets_preprocess/long_prepare_kitti.py:5

Methoddistance

(self, a, b)

eval/mv_recon/criterion.py:120

Methoddistance

(self, a, b)

src/dust3r/losses.py:68

Methoddistance

(self, a, b)

src/dust3r/losses.py:76

Functiondraw_correspondences

draw correspondences between two images :param img1: tensor [B, H, W, 3] :param img2: tensor [B, H, W, 3] :param coord1: tensor [B, N

src/dust3r/viz.py:598

Functiondraw_correspondences_lines

draw correspondences between two images :param img1: tensor [B, H, W, 3] :param img2: tensor [B, H, W, 3] :param coord1: tensor [B, N

src/dust3r/viz.py:630

Methodefficient_random_intervals

( start, num_elements, interval_range, fixed_interval_prob=0.8, weight

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

Functioneval_metrics_first_pose_align_last_pose

( pred_traj, gt_traj=None, seq="", filename="", figpath="", sample_stride=1 )

eval/relpose/evo_utils.py:288

Methodextra_repr

(self)

src/dust3r/blocks.py:55

Methodextra_repr

(self)

src/croco/models/blocks.py:64

Functionextract_correspondences_from_pts3d

( view1, view2, target_n_corres, rng=np.random, ret_xy=True, nneg=0 )

eval/mv_recon/dataset_utils/corr.py:52

Functionextract_number

(filename)

src/dust3r/datasets/omniobject3d.py:16

Functionextract_number

(filename)

src/dust3r/datasets/uasol.py:16

Functionfilename

transform a python obj or cmd into a proper filename. - \1 gets replaced by slash '/' - \2 gets replaced by comma ','

src/croco/utils/misc.py:469

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_flow.py:383

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_flow.py:574

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_flow.py:630

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_stereo.py:509

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_stereo.py:580

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_stereo.py:706

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_stereo.py:758

Methodfinalize_submission

(self, outdir)

src/croco/stereoflow/datasets_stereo.py:815

Functionfind_opt_scaling

( gt_pts1, gt_pts2, pr_pts1, pr_pts2=None, fit_mode="weiszfeld_stop_grad", valid1=None

src/dust3r/inference.py:339

Functionfind_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

src/dust3r/utils/geometry.py:491

Functionflip

flip so that tensor[0::2] <=> tensor[1::2]

src/dust3r/utils/misc.py:44

Methodforward

(self, *args, **kwargs)

eval/mv_recon/criterion.py:78

Methodforward

(self, a, b)

eval/mv_recon/criterion.py:99

Methodforward

(self, x)

src/dust3r/blocks.py:52

Methodforward

(self, x)

src/dust3r/blocks.py:83

Methodforward

(self, x, xpos, return_attn=False)

src/dust3r/blocks.py:102

Methodforward

(self, x, xpos)

src/dust3r/blocks.py:184

Methodforward

(self, query, key, value, qpos, kpos, return_attn=False)

src/dust3r/blocks.py:209

Methodforward

(self, x, y, xpos, ypos, return_attn=False)

src/dust3r/blocks.py:319

Methodforward

(self, x, y, z, xpos, ypos)

src/dust3r/blocks.py:382

Methodforward

(self, x: torch.Tensor, mod: torch.Tensor)

src/dust3r/blocks.py:412

Methodforward

(self, x, mod, xpos)

src/dust3r/blocks.py:452

Methodforward

(self, x)

src/dust3r/blocks.py:501

Methodforward

(self, a, b)

src/dust3r/losses.py:48

Methodforward

(self, *args, **kwargs)

src/dust3r/losses.py:146

Methodforward

(self, x, y)

src/dust3r/losses.py:180

Methodforward

(self, pred, gt, mask)

src/dust3r/losses.py:227

Methodforward

(self, pred, gt, mask)

src/dust3r/losses.py:263

Methodforward

(self, x, **kw)

src/dust3r/patch_embed.py:19

Methodforward

(self, img, true_shape)

src/dust3r/patch_embed.py:53

Methodforward

(self, views, ret_state=False)

src/dust3r/model.py:931

Methodforward

pose_feat: BxC preliminary_cameras: cameras in opencv coordinate.

src/dust3r/utils/camera.py:33

Methodforward

(self, camera)

src/dust3r/utils/camera.py:74

Methodforward

Args: x: tensor of shape [..., dim] diag_cov: An optional tensor of shape `(..., dim)` representing t

src/dust3r/utils/camera.py:194

← previous next →801–900 of 1,064, ranked by callers