MCPcopy Create free account

hub / github.com/Inception3D/TTT3R / functions

Functions1,064 in github.com/Inception3D/TTT3R

↓ 2 callersFunctionxy_grid
Output a (H,W,2) array of int32 with output[j,i,0] = i + origin[0] output[j,i,1] = j + origin[1]
src/dust3r/utils/geometry.py:15
↓ 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 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 callersMethod_compute_metrics
(self)
src/croco/stereoflow/criterion.py:220
↓ 1 callersMethod_compute_metrics
(self)
src/croco/stereoflow/criterion.py:333
↓ 1 callersMethod_decoder
(self, f_state, pos_state, f_img, pos_img, f_pose, pos_pose, return_attn)
src/dust3r/model.py:669
↓ 1 callersMethod_encode_state
(self, image_tokens, image_pos)
src/dust3r/model.py:547
↓ 1 callersMethod_forward_encoder
(self, views)
src/dust3r/model.py:751
↓ 1 callersMethod_get_depthpath
(self, obj, instance, view_idx)
src/dust3r/datasets/cop3d.py:24
↓ 1 callersMethod_get_depthpath
(self, obj, instance, view_idx)
src/dust3r/datasets/co3d.py:54
↓ 1 callersMethod_get_impath
(self, obj, instance, view_idx)
src/dust3r/datasets/cop3d.py:21
↓ 1 callersMethod_get_impath
(self, obj, instance, view_idx)
src/dust3r/datasets/co3d.py:51
↓ 1 callersMethod_get_maskpath
(self, obj, instance, view_idx)
src/dust3r/datasets/co3d.py:59
↓ 1 callersMethod_get_metadatapath
(self, obj, instance, view_idx)
src/dust3r/datasets/cop3d.py:18
↓ 1 callersMethod_get_metadatapath
(self, obj, instance, view_idx)
src/dust3r/datasets/co3d.py:48
↓ 1 callersFunction_get_num_layer_for_vit
(var_name, enc_depth, dec_depth)
src/croco/utils/misc.py:484
↓ 1 callersMethod_get_views
(self, idx, resolution, rng)
eval/mv_recon/base.py:68
↓ 1 callersMethod_get_views
(self, idx, resolution, rng, num_views)
src/dust3r/datasets/base/base_multiview_dataset.py:327
↓ 1 callersMethod_init_weights
(self, m)
src/croco/models/croco.py:200
↓ 1 callersMethod_lazy_initialization
(self)
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:195
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/scannetpp.py:24
↓ 1 callersMethod_load_data
(self, split)
src/dust3r/datasets/scannet.py:24
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/mvs_synth.py:23
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/eden.py:22
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/tartanair.py:27
↓ 1 callersMethod_load_data
(self, split=None)
src/dust3r/datasets/vkitti2.py:26
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/omniobject3d.py:32
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/uasol.py:32
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/urbansyn.py:22
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/smartportraits.py:22
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/blendedmvs.py:24
↓ 1 callersMethod_load_data
(self, split)
src/dust3r/datasets/megadepth.py:27
↓ 1 callersMethod_load_data
(self, split)
src/dust3r/datasets/dynamic_replica.py:24
↓ 1 callersMethod_load_data
(self, split)
src/dust3r/datasets/pointodyssey.py:60
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/mp3d.py:23
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/dl3dv.py:24
↓ 1 callersMethod_load_data
(self, split)
src/dust3r/datasets/arkitscenes.py:58
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/hypersim.py:24
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/threedkb.py:22
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/mvimgnet.py:24
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/irs.py:22
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/waymo.py:35
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/synscapes.py:22
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/realestate10k.py:23
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/bedlam.py:177
↓ 1 callersMethod_load_data
(self, split)
src/dust3r/datasets/arkitscenes_highres.py:31
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/mapfree.py:44
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/spring.py:24
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/unreal4k.py:30
↓ 1 callersMethod_load_data
(self)
src/dust3r/datasets/hoi4d.py:21
↓ 1 callersFunction_load_model_and_criterion
(model_path, do_load_metrics, device)
src/croco/stereoflow/test.py:71
↓ 1 callersMethod_load_or_build_cache
(self)
src/croco/stereoflow/datasets_flow.py:126
↓ 1 callersMethod_load_or_build_cache
(self)
src/croco/stereoflow/datasets_stereo.py:144
↓ 1 callersFunction_ntuple
(n)
src/dust3r/blocks.py:16
↓ 1 callersFunction_ntuple
(n)
src/croco/models/blocks.py:25
↓ 1 callersMethod_prepare_data
(self)
src/croco/stereoflow/datasets_flow.py:401
↓ 1 callersMethod_prepare_data
(self)
src/croco/stereoflow/datasets_stereo.py:273
↓ 1 callersFunction_print_repr_dataset
(d)
src/croco/stereoflow/train.py:299
↓ 1 callersMethod_random_color
(self, img1, img2)
src/croco/stereoflow/augmentor.py:205
↓ 1 callersMethod_random_crop
(self, img1, img2, disp)
src/croco/stereoflow/augmentor.py:122
↓ 1 callersMethod_random_rotate_shift_right
(self, img2)
src/croco/stereoflow/augmentor.py:141
↓ 1 callersMethod_random_scale
(self, img1, img2, disp)
src/croco/stereoflow/augmentor.py:59
↓ 1 callersMethod_random_vflip
(self, img1, img2, disp)
src/croco/stereoflow/augmentor.py:133
↓ 1 callersMethod_read_depthmap
(self, impath, input_metadata)
src/dust3r/datasets/cop3d.py:31
↓ 1 callersMethod_read_depthmap
(self, depthpath, input_metadata)
src/dust3r/datasets/co3d.py:62
↓ 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
src/croco/utils/misc.py:453
↓ 1 callersFunction_save_batch
( pred, gt, pairnames, dataset, task, save, outdir, time, submission_dir=None )
src/croco/stereoflow/test.py:103
↓ 1 callersMethod_set_decoder
( self, enc_embed_dim, dec_embed_dim, dec_num_heads, dec_depth,
src/croco/models/croco.py:156
↓ 1 callersMethod_set_mask_generator
(self, num_patches, mask_ratio)
src/croco/models/croco.py:150
↓ 1 callersMethod_set_patch_embed
(self, img_size=224, patch_size=16, enc_embed_dim=768)
src/croco/models/croco.py:147
↓ 1 callersMethod_set_prediction_head
(self, dec_embed_dim, patch_size)
src/croco/models/croco.py:188
↓ 1 callersMethod_set_resolutions
Set the resolution(s) of the dataset. Params: - resolutions: int or tuple or list of tuples
eval/mv_recon/base.py:142
↓ 1 callersMethod_set_resolutions
(self, resolutions)
src/dust3r/datasets/base/base_multiview_dataset.py:423
↓ 1 callersMethod_set_state_decoder
( self, enc_embed_dim, dec_embed_dim, dec_num_heads, dec_depth,
src/dust3r/model.py:366
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
src/dust3r/utils/camera.py:269
↓ 1 callersFunctionabsolute_value_scaling
(predicted_depth, ground_truth_depth, s=1, t=0)
eval/video_depth/tools.py:56
↓ 1 callersFunctionabsolute_value_scaling2
( predicted_depth, ground_truth_depth, s_init=1.0, t_init=0.0, lr=1e-4, max_iters=1000
eval/video_depth/tools.py:72
↓ 1 callersFunctionaccuracy
(gt_points, rec_points, gt_normals=None, rec_normals=None)
eval/mv_recon/utils.py:13
↓ 1 callersMethodadd_batch
(self, predictions, gt)
src/croco/stereoflow/criterion.py:255
↓ 1 callersMethodadd_camera
( self, pose_c2w, focal=None, color=(0, 0, 0), image=None, ims
src/dust3r/viz.py:222
↓ 1 callersMethodadd_pc
(self, step)
viser_utils.py:813
↓ 1 callersMethodadd_pc
(self, step)
src/dust3r/viz.py:906
↓ 1 callersMethodanimate
(self)
viser_utils.py:866
↓ 1 callersMethodanimate
(self)
src/dust3r/viz.py:973
↓ 1 callersFunctionarg_parser
()
src/croco/datasets/crops/extract_crops_from_images.py:17
↓ 1 callersFunctionassociate
Associate two dictionaries of (stamp, data). As the time stamps never match exactly, we aim to find the closest match for every input tuple.
datasets_preprocess/long_prepare_tum.py:28
↓ 1 callersFunctionbest_plotmode
(traj)
eval/relpose/evo_utils.py:364
↓ 1 callersMethodbuild_adjacency_list
(S, thresh=0.2)
src/dust3r/datasets/blendedmvs.py:56
↓ 1 callersFunctionc2w_to_tumpose
Convert a camera-to-world matrix to a tuple of translation and rotation input: c2w: 4x4 matrix output: tuple of translation and rotation
eval/relpose/utils.py:58
↓ 1 callersFunctioncalculate_averages
(results)
eval/relpose/evo_utils.py:451
↓ 1 callersFunctioncamera_matrix_of_crop
( input_camera_matrix, input_resolution, output_resolution, scaling=1, offset_factor=0.5,
eval/mv_recon/dataset_utils/cropping.py:97
↓ 1 callersFunctioncamera_matrix_of_crop
( input_camera_matrix, input_resolution, output_resolution, scaling=1, offset_factor=0.5,
src/dust3r/datasets/utils/cropping.py:101
↓ 1 callersFunctioncat_meshes
(meshes)
src/dust3r/viz.py:101
↓ 1 callersMethodcolor_transform
Photometric augmentation
src/croco/stereoflow/augmentor.py:272
↓ 1 callersFunctioncolorize_feature_map_np
Args: x: np.ndarray, [H, W, D]
src/dust3r/viz.py:1077
← previousnext →201–300 of 1,064, ranked by callers