SYMBOL INDEX (512 symbols across 64 files)

FILE: datasets_preprocess/habitat/find_scenes.py
  function find_all_scenes (line 16) | def find_all_scenes(habitat_root, n_scenes=[100000]):

FILE: datasets_preprocess/habitat/habitat_renderer/habitat_sim_envmaps_renderer.py
  class NoNaviguableSpaceError (line 25) | class NoNaviguableSpaceError(RuntimeError):
    method __init__ (line 26) | def __init__(self, *args):
  class HabitatEnvironmentMapRenderer (line 29) | class HabitatEnvironmentMapRenderer:
    method __init__ (line 30) | def __init__(self,
    method _lazy_initialization (line 64) | def _lazy_initialization(self):
    method close (line 141) | def close(self):
    method __del__ (line 145) | def __del__(self):
    method render_viewpoint (line 148) | def render_viewpoint(self, viewpoint_position):
    method up_direction (line 166) | def up_direction(self):
    method R_cam_to_world (line 169) | def R_cam_to_world(self):

FILE: datasets_preprocess/habitat/habitat_renderer/multiview_crop_generator.py
  class HabitatMultiviewCrops (line 17) | class HabitatMultiviewCrops:
    method __init__ (line 18) | def __init__(self,
    method compute_pointmap (line 48) | def compute_pointmap(self, distancemap, position):
    method render_viewpoint_data (line 52) | def render_viewpoint_data(self, position):
    method extract_cropped_camera (line 59) | def extract_cropped_camera(self, projection, color_image, distancemap,...
  function perspective_projection_to_dict (line 76) | def perspective_projection_to_dict(persp_projection, position):
  function dict_to_perspective_projection (line 86) | def dict_to_perspective_projection(camera_params):

FILE: datasets_preprocess/habitat/habitat_renderer/projections.py
  class EquirectangularProjection (line 9) | class EquirectangularProjection:
    method __init__ (line 18) | def __init__(self, height, width):
    method unproject (line 24) | def unproject(self, u, v):
    method project (line 41) | def project(self, rays):
  class PerspectiveProjection (line 59) | class PerspectiveProjection:
    method __init__ (line 72) | def __init__(self, K, height, width):
    method project (line 78) | def project(self, rays):
    method unproject (line 83) | def unproject(self, u, v):
  class RotatedProjection (line 89) | class RotatedProjection:
    method __init__ (line 90) | def __init__(self, base_projection, R_to_base_projection):
    method width (line 95) | def width(self):
    method height (line 99) | def height(self):
    method project (line 102) | def project(self, rays):
    method unproject (line 107) | def unproject(self, u, v):
  function get_projection_rays (line 113) | def get_projection_rays(projection, noise_level=0):
  function compute_camera_intrinsics (line 124) | def compute_camera_intrinsics(height, width, hfov):
  function colmap_to_opencv_intrinsics (line 129) | def colmap_to_opencv_intrinsics(K):
  function opencv_to_colmap_intrinsics (line 141) | def opencv_to_colmap_intrinsics(K):

FILE: datasets_preprocess/habitat/habitat_renderer/projections_conversions.py
  class RemapProjection (line 11) | class RemapProjection:
    method __init__ (line 12) | def __init__(self, input_projection, output_projection, pixel_jitterin...
    method convert (line 34) | def convert(self, img, interpolation=cv2.INTER_LINEAR, borderMode=cv2....

FILE: datasets_preprocess/habitat/preprocess_habitat.py
  function preprocess_metadata (line 26) | def preprocess_metadata(metadata_filename,

FILE: datasets_preprocess/preprocess_arkitscenes.py
  function get_parser (line 24) | def get_parser():
  function value_to_decimal (line 32) | def value_to_decimal(value, decimal_places):
  function closest (line 37) | def closest(value, sorted_list):
  function get_up_vectors (line 52) | def get_up_vectors(pose_device_to_world):
  function get_right_vectors (line 56) | def get_right_vectors(pose_device_to_world):
  function read_traj (line 60) | def read_traj(traj_path):
  function main (line 93) | def main(rootdir, pairsdir, outdir):
  function convert_scene_metadata (line 269) | def convert_scene_metadata(scene_subdir, intrinsics_dir,
  function find_scene_orientation (line 308) | def find_scene_orientation(poses_cam_to_world):

FILE: datasets_preprocess/preprocess_blendedMVS.py
  function get_parser (line 27) | def get_parser():
  function main (line 36) | def main(db_root, pairs_path, output_dir):
  function load_crop_and_save (line 63) | def load_crop_and_save(root, img, out_dir):
  function _crop_image (line 89) | def _crop_image(intrinsics_in, color_image_in, depthmap_in, resolution_o...
  function _load_pose (line 96) | def _load_pose(path, ret_44=False):
  function load_pfm_file (line 110) | def load_pfm_file(file_path):

FILE: datasets_preprocess/preprocess_co3d.py
  function get_parser (line 47) | def get_parser():
  function convert_ndc_to_pinhole (line 63) | def convert_ndc_to_pinhole(focal_length, principal_point, image_size):
  function opencv_from_cameras_projection (line 77) | def opencv_from_cameras_projection(R, T, focal, p0, image_size):
  function get_set_list (line 112) | def get_set_list(category_dir, split, is_single_sequence_subset=False):
  function prepare_sequences (line 129) | def prepare_sequences(category, co3d_dir, output_dir, img_size, split, m...

FILE: datasets_preprocess/preprocess_megadepth.py
  function get_parser (line 23) | def get_parser():
  function main (line 32) | def main(db_root, pairs_path, output_dir):
  function resize_one_image (line 65) | def resize_one_image(root, tag, K_pre_rectif, pose_w2cam, out_dir):
  function _downscale_image (line 95) | def _downscale_image(camera_intrinsics, image, depthmap, resolution_out=...
  function _load_kpts_and_poses (line 106) | def _load_kpts_and_poses(root, scene_id, subscene, z_only=False, intrins...
  function colmap_raw_pose_to_principal_axis (line 153) | def colmap_raw_pose_to_principal_axis(image_pose):
  function colmap_raw_pose_to_RT (line 165) | def colmap_raw_pose_to_RT(image_pose):

FILE: datasets_preprocess/preprocess_scannetpp.py
  function get_parser (line 43) | def get_parser():
  function pose_from_qwxyz_txyz (line 53) | def pose_from_qwxyz_txyz(elems):
  function get_frame_number (line 61) | def get_frame_number(name, cam_type='dslr'):
  function load_sfm (line 76) | def load_sfm(sfm_dir, cam_type='dslr'):
  function subsample_img_infos (line 134) | def subsample_img_infos(img_infos, num_images, allowed_name_subset=None):
  function undistort_images (line 146) | def undistort_images(intrinsics, rgb, mask):
  function process_scenes (line 191) | def process_scenes(root, pairsdir, output_dir, target_resolution):

FILE: datasets_preprocess/preprocess_staticthings3d.py
  function get_parser (line 27) | def get_parser():
  function main (line 36) | def main(db_root, pairs_path, output_dir):
  function load_crop_and_save (line 57) | def load_crop_and_save(db_root, relpath_, camera, num, out_dir):
  function _crop_image (line 88) | def _crop_image(intrinsics_in, color_image_in, depthmap_in, resolution_o...
  function _list_all_scenes (line 95) | def _list_all_scenes(path):
  function readFloat (line 108) | def readFloat(name):

FILE: datasets_preprocess/preprocess_waymo.py
  function get_parser (line 37) | def get_parser():
  function main (line 47) | def main(waymo_root, pairs_path, output_dir, workers=1):
  function _list_sequences (line 66) | def _list_sequences(db_root):
  function extract_frames (line 73) | def extract_frames(db_root, output_dir, workers=8):
  function process_one_seq (line 81) | def process_one_seq(db_root, output_dir, seq):
  function extract_frames_one_seq (line 105) | def extract_frames_one_seq(filename):
  function make_crops (line 170) | def make_crops(output_dir, workers=16, **kw):
  function crop_one_seq (line 177) | def crop_one_seq(input_dir, output_dir, seq, resolution=512):

FILE: datasets_preprocess/preprocess_wildrgbd.py
  function get_parser (line 29) | def get_parser():
  function get_set_list (line 43) | def get_set_list(category_dir, split):
  function prepare_sequences (line 60) | def prepare_sequences(category, wildrgbd_dir, output_dir, img_size, spli...

FILE: dust3r/cloud_opt/__init__.py
  class GlobalAlignerMode (line 14) | class GlobalAlignerMode(Enum):
  function global_aligner (line 20) | def global_aligner(dust3r_output, device, mode=GlobalAlignerMode.PointCl...

FILE: dust3r/cloud_opt/base_opt.py
  class BasePCOptimizer (line 27) | class BasePCOptimizer (nn.Module):
    method __init__ (line 33) | def __init__(self, *args, **kwargs):
    method _init_from_views (line 44) | def _init_from_views(self, view1, view2, pred1, pred2,
    method n_edges (line 108) | def n_edges(self):
    method str_edges (line 112) | def str_edges(self):
    method imsizes (line 116) | def imsizes(self):
    method device (line 120) | def device(self):
    method state_dict (line 123) | def state_dict(self, trainable=True):
    method load_state_dict (line 127) | def load_state_dict(self, data):
    method _check_edges (line 130) | def _check_edges(self):
    method _compute_img_conf (line 136) | def _compute_img_conf(self, pred1_conf, pred2_conf):
    method get_adaptors (line 143) | def get_adaptors(self):
    method _get_poses (line 150) | def _get_poses(self, poses):
    method _set_pose (line 157) | def _set_pose(self, poses, idx, R, T=None, scale=None, force=False):
    method get_pw_norm_scale_factor (line 178) | def get_pw_norm_scale_factor(self):
    method get_pw_scale (line 186) | def get_pw_scale(self):
    method get_pw_poses (line 191) | def get_pw_poses(self):  # cam to world
    method get_masks (line 197) | def get_masks(self):
    method depth_to_pts3d (line 200) | def depth_to_pts3d(self):
    method get_pts3d (line 203) | def get_pts3d(self, raw=False):
    method _set_focal (line 209) | def _set_focal(self, idx, focal, force=False):
    method get_focals (line 212) | def get_focals(self):
    method get_known_focal_mask (line 215) | def get_known_focal_mask(self):
    method get_principal_points (line 218) | def get_principal_points(self):
    method get_conf (line 221) | def get_conf(self, mode=None):
    method get_im_poses (line 225) | def get_im_poses(self):
    method _set_depthmap (line 228) | def _set_depthmap(self, idx, depth, force=False):
    method get_depthmaps (line 231) | def get_depthmaps(self, raw=False):
    method clean_pointcloud (line 234) | def clean_pointcloud(self, **kw):
    method forward (line 246) | def forward(self, ret_details=False):
    method compute_global_alignment (line 276) | def compute_global_alignment(self, init=None, niter_PnP=10, **kw):
    method mask_sky (line 290) | def mask_sky(self):
    method show (line 297) | def show(self, show_pw_cams=False, show_pw_pts3d=False, cam_size=None,...
  function global_alignment_loop (line 326) | def global_alignment_loop(net, lr=0.01, niter=300, schedule='cosine', lr...
  function global_alignment_iter (line 352) | def global_alignment_iter(net, cur_iter, niter, lr_base, lr_min, optimiz...
  function clean_pointcloud (line 370) | def clean_pointcloud( im_confs, K, cams, depthmaps, all_pts3d,

FILE: dust3r/cloud_opt/commons.py
  function edge_str (line 12) | def edge_str(i, j):
  function i_j_ij (line 16) | def i_j_ij(ij):
  function edge_conf (line 20) | def edge_conf(conf_i, conf_j, edge):
  function compute_edge_scores (line 24) | def compute_edge_scores(edges, conf_i, conf_j):
  function NoGradParamDict (line 28) | def NoGradParamDict(x):
  function get_imshapes (line 33) | def get_imshapes(edges, pred_i, pred_j):
  function get_conf_trf (line 48) | def get_conf_trf(mode):
  function l2_dist (line 62) | def l2_dist(a, b, weight):
  function l1_dist (line 66) | def l1_dist(a, b, weight):
  function signed_log1p (line 73) | def signed_log1p(x):
  function signed_expm1 (line 78) | def signed_expm1(x):
  function cosine_schedule (line 83) | def cosine_schedule(t, lr_start, lr_end):
  function linear_schedule (line 88) | def linear_schedule(t, lr_start, lr_end):

FILE: dust3r/cloud_opt/init_im_poses.py
  function init_from_known_poses (line 24) | def init_from_known_poses(self, niter_PnP=10, min_conf_thr=3):
  function init_minimum_spanning_tree (line 67) | def init_minimum_spanning_tree(self, **kw):
  function init_from_pts3d (line 80) | def init_from_pts3d(self, pts3d, im_focals, im_poses):
  function minimum_spanning_tree (line 123) | def minimum_spanning_tree(imshapes, edges, pred_i, pred_j, conf_i, conf_...
  function dict_to_sparse_graph (line 212) | def dict_to_sparse_graph(dic):
  function rigid_points_registration (line 220) | def rigid_points_registration(pts1, pts2, conf):
  function sRT_to_4x4 (line 226) | def sRT_to_4x4(scale, R, T, device):
  function estimate_focal (line 233) | def estimate_focal(pts3d_i, pp=None):
  function pixel_grid (line 243) | def pixel_grid(H, W):
  function fast_pnp (line 247) | def fast_pnp(pts3d, focal, msk, device, pp=None, niter_PnP=10):
  function get_known_poses (line 290) | def get_known_poses(self):
  function get_known_focals (line 299) | def get_known_focals(self):
  function align_multiple_poses (line 308) | def align_multiple_poses(src_poses, target_poses):

FILE: dust3r/cloud_opt/modular_optimizer.py
  class ModularPointCloudOptimizer (line 17) | class ModularPointCloudOptimizer (BasePCOptimizer):
    method __init__ (line 24) | def __init__(self, *args, optimize_pp=False, fx_and_fy=False, focal_br...
    method preset_pose (line 38) | def preset_pose(self, known_poses, pose_msk=None):  # cam-to-world
    method preset_intrinsics (line 50) | def preset_intrinsics(self, known_intrinsics, msk=None):
    method preset_focal (line 58) | def preset_focal(self, known_focals, msk=None):
    method preset_principal_point (line 64) | def preset_principal_point(self, known_pp, msk=None):
    method _no_grad (line 70) | def _no_grad(self, tensor):
    method _get_msk_indices (line 73) | def _get_msk_indices(self, msk):
    method _set_focal (line 88) | def _set_focal(self, idx, focal, force=False):
    method get_focals (line 94) | def get_focals(self):
    method _set_principal_point (line 98) | def _set_principal_point(self, idx, pp, force=False):
    method get_principal_points (line 105) | def get_principal_points(self):
    method get_intrinsics (line 108) | def get_intrinsics(self):
    method get_im_poses (line 117) | def get_im_poses(self):  # cam to world
    method _set_depthmap (line 121) | def _set_depthmap(self, idx, depth, force=False):
    method get_depthmaps (line 127) | def get_depthmaps(self):
    method depth_to_pts3d (line 130) | def depth_to_pts3d(self):
    method get_pts3d (line 144) | def get_pts3d(self):

FILE: dust3r/cloud_opt/optimizer.py
  class PointCloudOptimizer (line 16) | class PointCloudOptimizer(BasePCOptimizer):
    method __init__ (line 22) | def __init__(self, *args, optimize_pp=False, focal_break=20, **kwargs):
    method _check_all_imgs_are_selected (line 63) | def _check_all_imgs_are_selected(self, msk):
    method preset_pose (line 66) | def preset_pose(self, known_poses, pose_msk=None):  # cam-to-world
    method preset_focal (line 83) | def preset_focal(self, known_focals, msk=None):
    method preset_principal_point (line 93) | def preset_principal_point(self, known_pp, msk=None):
    method _get_msk_indices (line 103) | def _get_msk_indices(self, msk):
    method _no_grad (line 118) | def _no_grad(self, tensor):
    method _set_focal (line 121) | def _set_focal(self, idx, focal, force=False):
    method get_focals (line 127) | def get_focals(self):
    method get_known_focal_mask (line 131) | def get_known_focal_mask(self):
    method _set_principal_point (line 134) | def _set_principal_point(self, idx, pp, force=False):
    method get_principal_points (line 141) | def get_principal_points(self):
    method get_intrinsics (line 144) | def get_intrinsics(self):
    method get_im_poses (line 152) | def get_im_poses(self):  # cam to world
    method _set_depthmap (line 156) | def _set_depthmap(self, idx, depth, force=False):
    method get_depthmaps (line 164) | def get_depthmaps(self, raw=False):
    method depth_to_pts3d (line 170) | def depth_to_pts3d(self):
    method get_pts3d (line 182) | def get_pts3d(self, raw=False):
    method forward (line 188) | def forward(self):
  function _fast_depthmap_to_pts3d (line 204) | def _fast_depthmap_to_pts3d(depth, pixel_grid, focal, pp):
  function ParameterStack (line 214) | def ParameterStack(params, keys=None, is_param=None, fill=0):
  function _ravel_hw (line 231) | def _ravel_hw(tensor, fill=0):
  function acceptable_focal_range (line 240) | def acceptable_focal_range(H, W, minf=0.5, maxf=3.5):
  function apply_mask (line 245) | def apply_mask(img, msk):

FILE: dust3r/cloud_opt/pair_viewer.py
  class PairViewer (line 18) | class PairViewer (BasePCOptimizer):
    method __init__ (line 24) | def __init__(self, *args, **kwargs):
    method _set_depthmap (line 83) | def _set_depthmap(self, idx, depth, force=False):
    method get_depthmaps (line 88) | def get_depthmaps(self, raw=False):
    method _set_focal (line 92) | def _set_focal(self, idx, focal, force=False):
    method get_focals (line 95) | def get_focals(self):
    method get_known_focal_mask (line 98) | def get_known_focal_mask(self):
    method get_principal_points (line 101) | def get_principal_points(self):
    method get_intrinsics (line 104) | def get_intrinsics(self):
    method get_im_poses (line 114) | def get_im_poses(self):
    method depth_to_pts3d (line 117) | def depth_to_pts3d(self):
    method forward (line 126) | def forward(self):

FILE: dust3r/datasets/__init__.py
  function get_data_loader (line 16) | def get_data_loader(dataset, batch_size, num_workers=8, shuffle=True, dr...

FILE: dust3r/datasets/arkitscenes.py
  class ARKitScenes (line 17) | class ARKitScenes(BaseStereoViewDataset):
    method __init__ (line 18) | def __init__(self, *args, split, ROOT, **kwargs):
    method _load_data (line 30) | def _load_data(self, split):
    method __len__ (line 39) | def __len__(self):
    method _get_views (line 42) | def _get_views(self, idx, resolution, rng):

FILE: dust3r/datasets/base/base_stereo_view_dataset.py
  class BaseStereoViewDataset (line 17) | class BaseStereoViewDataset (EasyDataset):
    method __init__ (line 29) | def __init__(self, *,  # only keyword arguments
    method __len__ (line 46) | def __len__(self):
    method get_stats (line 49) | def get_stats(self):
    method __repr__ (line 52) | def __repr__(self):
    method _get_views (line 60) | def _get_views(self, idx, resolution, rng):
    method __getitem__ (line 63) | def __getitem__(self, idx):
    method _set_resolutions (line 120) | def _set_resolutions(self, resolutions):
    method _crop_resize_if_necessary (line 137) | def _crop_resize_if_necessary(self, image, depthmap, intrinsics, resol...
  function is_good_type (line 184) | def is_good_type(key, v):
  function view_name (line 194) | def view_name(view, batch_index=None):
  function transpose_to_landscape (line 202) | def transpose_to_landscape(view):

FILE: dust3r/datasets/base/batched_sampler.py
  class BatchedRandomSampler (line 11) | class BatchedRandomSampler:
    method __init__ (line 21) | def __init__(self, dataset, batch_size, pool_size, world_size=1, rank=...
    method __len__ (line 34) | def __len__(self):
    method set_epoch (line 37) | def set_epoch(self, epoch):
    method __iter__ (line 40) | def __iter__(self):
  function round_by (line 71) | def round_by(total, multiple, up=False):

FILE: dust3r/datasets/base/easy_dataset.py
  class EasyDataset (line 11) | class EasyDataset:
    method __add__ (line 22) | def __add__(self, other):
    method __rmul__ (line 25) | def __rmul__(self, factor):
    method __rmatmul__ (line 28) | def __rmatmul__(self, factor):
    method set_epoch (line 31) | def set_epoch(self, epoch):
    method make_sampler (line 34) | def make_sampler(self, batch_size, shuffle=True, world_size=1, rank=0,...
  class MulDataset (line 41) | class MulDataset (EasyDataset):
    method __init__ (line 46) | def __init__(self, multiplicator, dataset):
    method __len__ (line 51) | def __len__(self):
    method __repr__ (line 54) | def __repr__(self):
    method __getitem__ (line 57) | def __getitem__(self, idx):
    method _resolutions (line 65) | def _resolutions(self):
  class ResizedDataset (line 69) | class ResizedDataset (EasyDataset):
    method __init__ (line 74) | def __init__(self, new_size, dataset):
    method __len__ (line 79) | def __len__(self):
    method __repr__ (line 82) | def __repr__(self):
    method set_epoch (line 89) | def set_epoch(self, epoch):
    method __getitem__ (line 102) | def __getitem__(self, idx):
    method _resolutions (line 111) | def _resolutions(self):
  class CatDataset (line 115) | class CatDataset (EasyDataset):
    method __init__ (line 119) | def __init__(self, datasets):
    method __len__ (line 125) | def __len__(self):
    method __repr__ (line 128) | def __repr__(self):
    method set_epoch (line 132) | def set_epoch(self, epoch):
    method __getitem__ (line 136) | def __getitem__(self, idx):
    method _resolutions (line 153) | def _resolutions(self):

FILE: dust3r/datasets/blendedmvs.py
  class BlendedMVS (line 16) | class BlendedMVS (BaseStereoViewDataset):
    method __init__ (line 20) | def __init__(self, *args, ROOT, split=None, **kwargs):
    method _load_data (line 25) | def _load_data(self, split):
    method __len__ (line 40) | def __len__(self):
    method get_stats (line 43) | def get_stats(self):
    method _get_views (line 46) | def _get_views(self, pair_idx, resolution, rng):

FILE: dust3r/datasets/co3d.py
  class Co3d (line 21) | class Co3d(BaseStereoViewDataset):
    method __init__ (line 22) | def __init__(self, mask_bg=True, *args, ROOT, **kwargs):
    method __len__ (line 45) | def __len__(self):
    method _get_metadatapath (line 48) | def _get_metadatapath(self, obj, instance, view_idx):
    method _get_impath (line 51) | def _get_impath(self, obj, instance, view_idx):
    method _get_depthpath (line 54) | def _get_depthpath(self, obj, instance, view_idx):
    method _get_maskpath (line 57) | def _get_maskpath(self, obj, instance, view_idx):
    method _read_depthmap (line 60) | def _read_depthmap(self, depthpath, input_metadata):
    method _get_views (line 65) | def _get_views(self, idx, resolution, rng):

FILE: dust3r/datasets/habitat.py
  class Habitat (line 20) | class Habitat(BaseStereoViewDataset):
    method __init__ (line 21) | def __init__(self, size, *args, ROOT, **kwargs):
    method filter_scene (line 30) | def filter_scene(self, label, instance=None):
    method _get_views (line 39) | def _get_views(self, idx, resolution, rng):
    method _load_one_view (line 60) | def _load_one_view(self, data_path, key, view_index, resolution, rng):

FILE: dust3r/datasets/megadepth.py
  class MegaDepth (line 16) | class MegaDepth(BaseStereoViewDataset):
    method __init__ (line 17) | def __init__(self, *args, split, ROOT, **kwargs):
    method _load_data (line 31) | def _load_data(self, split):
    method __len__ (line 37) | def __len__(self):
    method get_stats (line 40) | def get_stats(self):
    method select_scene (line 43) | def select_scene(self, scene, *instances, opposite=False):
    method _get_views (line 64) | def _get_views(self, pair_idx, resolution, rng):

FILE: dust3r/datasets/scannetpp.py
  class ScanNetpp (line 18) | class ScanNetpp(BaseStereoViewDataset):
    method __init__ (line 19) | def __init__(self, *args, ROOT, **kwargs):
    method _load_data (line 25) | def _load_data(self):
    method __len__ (line 34) | def __len__(self):
    method _get_views (line 37) | def _get_views(self, idx, resolution, rng):

FILE: dust3r/datasets/staticthings3d.py
  class StaticThings3D (line 16) | class StaticThings3D (BaseStereoViewDataset):
    method __init__ (line 19) | def __init__(self, ROOT, *args, mask_bg='rand', **kwargs):
    method __len__ (line 30) | def __len__(self):
    method get_stats (line 33) | def get_stats(self):
    method _get_views (line 36) | def _get_views(self, pair_idx, resolution, rng):

FILE: dust3r/datasets/utils/cropping.py
  class ImageList (line 21) | class ImageList:
    method __init__ (line 25) | def __init__(self, images):
    method __len__ (line 34) | def __len__(self):
    method to_pil (line 37) | def to_pil(self):
    method size (line 41) | def size(self):
    method resize (line 46) | def resize(self, *args, **kwargs):
    method crop (line 49) | def crop(self, *args, **kwargs):
    method _dispatch (line 52) | def _dispatch(self, func, *args, **kwargs):
  function rescale_image_depthmap (line 56) | def rescale_image_depthmap(image, depthmap, camera_intrinsics, output_re...
  function camera_matrix_of_crop (line 87) | def camera_matrix_of_crop(input_camera_matrix, input_resolution, output_...
  function crop_image_depthmap (line 103) | def crop_image_depthmap(image, depthmap, camera_intrinsics, crop_bbox):
  function bbox_from_intrinsics_in_out (line 120) | def bbox_from_intrinsics_in_out(input_camera_matrix, output_camera_matri...

FILE: dust3r/datasets/waymo.py
  class Waymo (line 16) | class Waymo (BaseStereoViewDataset):
    method __init__ (line 20) | def __init__(self, *args, ROOT, **kwargs):
    method _load_data (line 25) | def _load_data(self):
    method __len__ (line 33) | def __len__(self):
    method get_stats (line 36) | def get_stats(self):
    method _get_views (line 39) | def _get_views(self, pair_idx, resolution, rng):

FILE: dust3r/datasets/wildrgbd.py
  class WildRGBD (line 18) | class WildRGBD(Co3d):
    method __init__ (line 19) | def __init__(self, mask_bg=True, *args, ROOT, **kwargs):
    method _get_metadatapath (line 23) | def _get_metadatapath(self, obj, instance, view_idx):
    method _get_impath (line 26) | def _get_impath(self, obj, instance, view_idx):
    method _get_depthpath (line 29) | def _get_depthpath(self, obj, instance, view_idx):
    method _get_maskpath (line 32) | def _get_maskpath(self, obj, instance, view_idx):
    method _read_depthmap (line 35) | def _read_depthmap(self, depthpath, input_metadata):

FILE: dust3r/demo.py
  function get_args_parser (line 30) | def get_args_parser():
  function set_print_with_timestamp (line 53) | def set_print_with_timestamp(time_format="%Y-%m-%d %H:%M:%S"):
  function _convert_scene_output_to_glb (line 66) | def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams...
  function get_3D_model_from_scene (line 110) | def get_3D_model_from_scene(outdir, silent, scene, min_conf_thr=3, as_po...
  function get_reconstructed_scene (line 135) | def get_reconstructed_scene(outdir, model, device, silent, image_size, f...
  function set_scenegraph_options (line 189) | def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
  function main_demo (line 210) | def main_demo(tmpdirname, model, device, image_size, server_name, server...

FILE: dust3r/heads/__init__.py
  function head_factory (line 11) | def head_factory(head_type, output_mode, net, has_conf=False):

FILE: dust3r/heads/dpt_head.py
  class DPTOutputAdapter_fix (line 20) | class DPTOutputAdapter_fix(DPTOutputAdapter):
    method init (line 26) | def init(self, dim_tokens_enc=768):
    method forward (line 34) | def forward(self, encoder_tokens: List[torch.Tensor], image_size=None):
  class PixelwiseTaskWithDPT (line 68) | class PixelwiseTaskWithDPT(nn.Module):
    method __init__ (line 71) | def __init__(self, *, n_cls_token=0, hooks_idx=None, dim_tokens=None,
    method forward (line 89) | def forward(self, x, img_info):
  function create_dpt_head (line 96) | def create_dpt_head(net, has_conf=False):

FILE: dust3r/heads/linear_head.py
  class LinearPts3d (line 12) | class LinearPts3d (nn.Module):
    method __init__ (line 18) | def __init__(self, net, has_conf=False):
    method setup (line 27) | def setup(self, croconet):
    method forward (line 30) | def forward(self, decout, img_shape):

FILE: dust3r/heads/postprocess.py
  function postprocess (line 10) | def postprocess(out, depth_mode, conf_mode):
  function reg_dense_depth (line 22) | def reg_dense_depth(xyz, mode):
  function reg_dense_conf (line 49) | def reg_dense_conf(x, mode):

FILE: dust3r/image_pairs.py
  function make_pairs (line 11) | def make_pairs(imgs, scene_graph='complete', prefilter=None, symmetrize=...
  function sel (line 71) | def sel(x, kept):
  function _filter_edges_seq (line 80) | def _filter_edges_seq(edges, seq_dis_thr, cyclic=False):
  function filter_pairs_seq (line 94) | def filter_pairs_seq(pairs, seq_dis_thr, cyclic=False):
  function filter_edges_seq (line 100) | def filter_edges_seq(view1, view2, pred1, pred2, seq_dis_thr, cyclic=Fal...

FILE: dust3r/inference.py
  function _interleave_imgs (line 14) | def _interleave_imgs(img1, img2):
  function make_batch_symmetric (line 26) | def make_batch_symmetric(batch):
  function loss_of_one_batch (line 32) | def loss_of_one_batch(batch, model, criterion, device, symmetrize_batch=...
  function inference (line 56) | def inference(pairs, model, device, batch_size=8, verbose=True):
  function check_if_same_size (line 75) | def check_if_same_size(pairs):
  function get_pred_pts3d (line 81) | def get_pred_pts3d(gt, pred, use_pose=False):
  function find_opt_scaling (line 106) | def find_opt_scaling(gt_pts1, gt_pts2, pr_pts1, pr_pts2=None, fit_mode='...

FILE: dust3r/losses.py
  function Sum (line 16) | def Sum(*losses_and_masks):
  class BaseCriterion (line 28) | class BaseCriterion(nn.Module):
    method __init__ (line 29) | def __init__(self, reduction='mean'):
  class LLoss (line 34) | class LLoss (BaseCriterion):
    method forward (line 38) | def forward(self, a, b):
    method distance (line 50) | def distance(self, a, b):
  class L21Loss (line 54) | class L21Loss (LLoss):
    method distance (line 57) | def distance(self, a, b):
  class Criterion (line 64) | class Criterion (nn.Module):
    method __init__ (line 65) | def __init__(self, criterion=None):
    method get_name (line 70) | def get_name(self):
    method with_reduction (line 73) | def with_reduction(self, mode='none'):
  class MultiLoss (line 82) | class MultiLoss (nn.Module):
    method __init__ (line 89) | def __init__(self):
    method compute_loss (line 94) | def compute_loss(self, *args, **kwargs):
    method get_name (line 97) | def get_name(self):
    method __mul__ (line 100) | def __mul__(self, alpha):
    method __add__ (line 107) | def __add__(self, loss2):
    method __repr__ (line 116) | def __repr__(self):
    method forward (line 124) | def forward(self, *args, **kwargs):
  class Regr3D (line 142) | class Regr3D (Criterion, MultiLoss):
    method __init__ (line 153) | def __init__(self, criterion, norm_mode='avg_dis', gt_scale=False):
    method get_all_pts3d (line 158) | def get_all_pts3d(self, gt1, gt2, pred1, pred2, dist_clip=None):
    method compute_loss (line 185) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw):
  class ConfLoss (line 197) | class ConfLoss (MultiLoss):
    method __init__ (line 208) | def __init__(self, pixel_loss, alpha=1):
    method get_name (line 214) | def get_name(self):
    method get_conf_log (line 217) | def get_conf_log(self, x):
    method compute_loss (line 220) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw):
  class Regr3D_ShiftInv (line 241) | class Regr3D_ShiftInv (Regr3D):
    method get_all_pts3d (line 245) | def get_all_pts3d(self, gt1, gt2, pred1, pred2):
  class Regr3D_ScaleInv (line 266) | class Regr3D_ScaleInv (Regr3D):
    method get_all_pts3d (line 271) | def get_all_pts3d(self, gt1, gt2, pred1, pred2):
  class Regr3D_ScaleShiftInv (line 297) | class Regr3D_ScaleShiftInv (Regr3D_ScaleInv, Regr3D_ShiftInv):

FILE: dust3r/model.py
  function load_model (line 27) | def load_model(model_path, device, verbose=True):
  class AsymmetricCroCo3DStereo (line 46) | class AsymmetricCroCo3DStereo (
    method __init__ (line 58) | def __init__(self,
    method from_pretrained (line 77) | def from_pretrained(cls, pretrained_model_name_or_path, **kw):
    method _set_patch_embed (line 87) | def _set_patch_embed(self, img_size=224, patch_size=16, enc_embed_dim=...
    method load_state_dict (line 91) | def load_state_dict(self, ckpt, **kw):
    method set_freeze (line 100) | def set_freeze(self, freeze):  # this is for use by downstream models
    method _set_prediction_head (line 109) | def _set_prediction_head(self, *args, **kwargs):
    method set_downstream_head (line 113) | def set_downstream_head(self, output_mode, head_type, landscape_only, ...
    method _encode_image (line 128) | def _encode_image(self, image, true_shape):
    method _encode_image_pairs (line 142) | def _encode_image_pairs(self, img1, img2, true_shape1, true_shape2):
    method _encode_symmetrized (line 153) | def _encode_symmetrized(self, view1, view2):
    method _decoder (line 172) | def _decoder(self, f1, pos1, f2, pos2):
    method _downstream_head (line 193) | def _downstream_head(self, head_num, decout, img_shape):
    method forward (line 199) | def forward(self, view1, view2):

FILE: dust3r/optim_factory.py
  function adjust_learning_rate_by_lr (line 9) | def adjust_learning_rate_by_lr(optimizer, lr):

FILE: dust3r/patch_embed.py
  function get_patch_embed (line 13) | def get_patch_embed(patch_embed_cls, img_size, patch_size, enc_embed_dim):
  class PatchEmbedDust3R (line 19) | class PatchEmbedDust3R(PatchEmbed):
    method forward (line 20) | def forward(self, x, **kw):
  class ManyAR_PatchEmbed (line 32) | class ManyAR_PatchEmbed (PatchEmbed):
    method __init__ (line 38) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=...
    method forward (line 42) | def forward(self, img, true_shape):

FILE: dust3r/post_process.py
  function estimate_focal_knowing_depth (line 12) | def estimate_focal_knowing_depth(pts3d, pp, focal_mode='median', min_foc...

FILE: dust3r/training.py
  function get_args_parser (line 39) | def get_args_parser():
  function train (line 92) | def train(args):
  function save_final_model (line 239) | def save_final_model(args, epoch, model_without_ddp, best_so_far=None):
  function build_dataset (line 253) | def build_dataset(dataset, batch_size, num_workers, test=False):
  function train_one_epoch (line 267) | def train_one_epoch(model: torch.nn.Module, criterion: torch.nn.Module,
  function test_one_epoch (line 342) | def test_one_epoch(model: torch.nn.Module, criterion: torch.nn.Module,

FILE: dust3r/utils/device.py
  function todevice (line 11) | def todevice(batch, device, callback=None, non_blocking=False):
  function to_numpy (line 42) | def to_numpy(x): return todevice(x, 'numpy')
  function to_cpu (line 43) | def to_cpu(x): return todevice(x, 'cpu')
  function to_cuda (line 44) | def to_cuda(x): return todevice(x, 'cuda')
  function collate_with_cat (line 47) | def collate_with_cat(whatever, lists=False):
  function listify (line 75) | def listify(elems):

FILE: dust3r/utils/geometry.py
  function xy_grid (line 15) | def xy_grid(W, H, device=None, origin=(0, 0), unsqueeze=None, cat_dim=-1...
  function geotrf (line 40) | def geotrf(Trf, pts, ncol=None, norm=False):
  function inv (line 104) | def inv(mat):
  function depthmap_to_pts3d (line 114) | def depthmap_to_pts3d(depth, pseudo_focal, pp=None, **_):
  function depthmap_to_camera_coordinates (line 165) | def depthmap_to_camera_coordinates(depthmap, camera_intrinsics, pseudo_f...
  function depthmap_to_absolute_camera_coordinates (line 200) | def depthmap_to_absolute_camera_coordinates(depthmap, camera_intrinsics,...
  function colmap_to_opencv_intrinsics (line 223) | def colmap_to_opencv_intrinsics(K):
  function opencv_to_colmap_intrinsics (line 236) | def opencv_to_colmap_intrinsics(K):
  function normalize_pointcloud (line 249) | def normalize_pointcloud(pts1, pts2, norm_mode='avg_dis', valid1=None, v...
  function get_joint_pointcloud_depth (line 313) | def get_joint_pointcloud_depth(z1, z2, valid_mask1, valid_mask2=None, qu...
  function get_joint_pointcloud_center_scale (line 328) | def get_joint_pointcloud_center_scale(pts1, pts2, valid_mask1=None, vali...
  function find_reciprocal_matches (line 345) | def find_reciprocal_matches(P1, P2):
  function get_med_dist_between_poses (line 364) | def get_med_dist_between_poses(poses):

FILE: dust3r/utils/image.py
  function img_to_arr (line 26) | def img_to_arr(img):
  function imread_cv2 (line 32) | def imread_cv2(path, options=cv2.IMREAD_COLOR):
  function rgb (line 45) | def rgb(ftensor, true_shape=None):
  function _resize_pil_image (line 64) | def _resize_pil_image(img, long_edge_size):
  function load_images (line 74) | def load_images(folder_or_list, size, square_ok=False, verbose=True, pat...

FILE: dust3r/utils/misc.py
  function fill_default_args (line 10) | def fill_default_args(kwargs, func):
  function freeze_all_params (line 22) | def freeze_all_params(modules):
  function is_symmetrized (line 32) | def is_symmetrized(gt1, gt2):
  function flip (line 43) | def flip(tensor):
  function interleave (line 48) | def interleave(tensor1, tensor2):
  function transpose_to_landscape (line 54) | def transpose_to_landscape(head, activate=True):
  function transposed (line 99) | def transposed(dic):
  function invalid_to_nans (line 103) | def invalid_to_nans(arr, valid_mask, ndim=999):
  function invalid_to_zeros (line 112) | def invalid_to_zeros(arr, valid_mask, ndim=999):

FILE: dust3r/utils/parallel.py
  function parallel_threads (line 12) | def parallel_threads(function, args, workers=0, star_args=False, kw_args...
  function parallel_processes (line 62) | def parallel_processes(*args, **kwargs):
  function starcall (line 70) | def starcall(args):
  function starstarcall (line 76) | def starstarcall(args):

FILE: dust3r/viz.py
  function cat_3d (line 23) | def cat_3d(vecs):
  function show_raw_pointcloud (line 29) | def show_raw_pointcloud(pts3d, colors, point_size=2):
  function pts3d_to_trimesh (line 38) | def pts3d_to_trimesh(img, pts3d, valid=None):
  function cat_meshes (line 78) | def cat_meshes(meshes):
  function show_duster_pairs (line 90) | def show_duster_pairs(view1, view2, pred1, pred2):
  function auto_cam_size (line 115) | def auto_cam_size(im_poses):
  class SceneViz (line 119) | class SceneViz:
    method __init__ (line 120) | def __init__(self):
    method add_rgbd (line 123) | def add_rgbd(self, image, depth, intrinsics=None, cam2world=None, zfar...
    method add_pointcloud (line 137) | def add_pointcloud(self, pts3d, color=(0,0,0), mask=None, denoise=False):
    method add_rgbd (line 173) | def add_rgbd(self, image, depth, intrinsics=None, cam2world=None, zfar...
    method add_camera (line 190) | def add_camera(self, pose_c2w, focal=None, color=(0, 0, 0), image=None...
    method add_cameras (line 202) | def add_cameras(self, poses, focals=None, images=None, imsizes=None, c...
    method show (line 208) | def show(self, point_size=2):
  function show_raw_pointcloud_with_cams (line 212) | def show_raw_pointcloud_with_cams(imgs, pts3d, mask, focals, cams2world,
  function add_scene_cam (line 246) | def add_scene_cam(scene, pose_c2w, edge_color, image=None, focal=None, i...
  function cat (line 322) | def cat(a, b):
  function uint8 (line 336) | def uint8(colors):
  function segment_sky (line 345) | def segment_sky(image):

FILE: dust3r_visloc/datasets/aachen_day_night.py
  class VislocAachenDayNight (line 11) | class VislocAachenDayNight(BaseVislocColmapDataset):
    method __init__ (line 12) | def __init__(self, root, subscene, pairsfile, topk=1, cache_sfm=False):

FILE: dust3r_visloc/datasets/base_colmap.py
  function kapture_to_opencv_intrinsics (line 29) | def kapture_to_opencv_intrinsics(sensor):
  function K_from_colmap (line 79) | def K_from_colmap(elems):
  function pose_from_qwxyz_txyz (line 88) | def pose_from_qwxyz_txyz(elems):
  class BaseVislocColmapDataset (line 96) | class BaseVislocColmapDataset(BaseVislocDataset):
    method __init__ (line 97) | def __init__(self, image_path, map_path, query_path, pairsfile_path, t...
    method _load_sfm (line 116) | def _load_sfm(self, sfm_dir):
    method __len__ (line 171) | def __len__(self):
    method _get_view_query (line 174) | def _get_view_query(self, imgname):
    method _get_view_map (line 220) | def _get_view_map(self, imgname, idx):
    method __getitem__ (line 274) | def __getitem__(self, idx):

FILE: dust3r_visloc/datasets/base_dataset.py
  class BaseVislocDataset (line 7) | class BaseVislocDataset:
    method __init__ (line 8) | def __init__(self):
    method set_resolution (line 11) | def set_resolution(self, model):
    method __len__ (line 15) | def __len__(self):
    method __getitem__ (line 18) | def __getitem__(self, idx):

FILE: dust3r_visloc/datasets/cambridge_landmarks.py
  class VislocCambridgeLandmarks (line 11) | class VislocCambridgeLandmarks (BaseVislocColmapDataset):
    method __init__ (line 12) | def __init__(self, root, subscene, pairsfile, topk=1, cache_sfm=False):

FILE: dust3r_visloc/datasets/inloc.py
  function read_alignments (line 23) | def read_alignments(path_to_alignment):
  class VislocInLoc (line 47) | class VislocInLoc(BaseVislocDataset):
    method __init__ (line 48) | def __init__(self, root, pairsfile, topk=1):
    method __len__ (line 88) | def __len__(self):
    method __getitem__ (line 91) | def __getitem__(self, idx):

FILE: dust3r_visloc/datasets/sevenscenes.py
  class VislocSevenScenes (line 23) | class VislocSevenScenes(BaseVislocDataset):
    method __init__ (line 24) | def __init__(self, root, subscene, pairsfile, topk=1):
    method __len__ (line 54) | def __len__(self):
    method __getitem__ (line 57) | def __getitem__(self, idx):

FILE: dust3r_visloc/datasets/utils.py
  function cam_to_world_from_kapture (line 13) | def cam_to_world_from_kapture(kdata, timestamp, camera_id):
  function get_HW_resolution (line 27) | def get_HW_resolution(H, W, maxdim, patchsize=16):
  function get_resize_function (line 51) | def get_resize_function(maxdim, patch_size, H, W, is_mask=False):
  function rescale_points3d (line 93) | def rescale_points3d(pts2d, pts3d, to_resize, HR, WR):

FILE: dust3r_visloc/evaluation.py
  function aggregate_stats (line 15) | def aggregate_stats(info_str, pose_errors, angular_errors):
  function get_pose_error (line 31) | def get_pose_error(pr_camtoworld, gt_cam_to_world):
  function export_results (line 38) | def export_results(output_dir, xp_label, query_names, poses_pred):

FILE: dust3r_visloc/localization.py
  function run_pnp (line 30) | def run_pnp(pts2D, pts3D, K, distortion = None, mode='cv2', reprojection...

FILE: visloc.py
  function get_args_parser (line 23) | def get_args_parser():