SYMBOL INDEX (764 symbols across 76 files)

FILE: eval.py
  function eval (line 13) | def eval(args):

FILE: hubconf.py
  class MVSAnywhereInference (line 17) | class MVSAnywhereInference(nn.Module):
    method __init__ (line 19) | def __init__(
    method forward (line 64) | def forward(
    method preprocess_and_run (line 162) | def preprocess_and_run(
  function mvsanywhere (line 280) | def mvsanywhere(

FILE: scripts/create_visibility_volume.py
  class SimpleScanNetDataset (line 30) | class SimpleScanNetDataset(torch.utils.data.Dataset):
    method __init__ (line 33) | def __init__(
    method _get_available_frames (line 48) | def _get_available_frames(self, frame_interval: int):
    method load_rendered_depth (line 60) | def load_rendered_depth(self, frame_ind: int) -> torch.Tensor:
    method load_pose (line 73) | def load_pose(self, frame_ind) -> dict[str, torch.Tensor]:
    method load_intrinsics (line 87) | def load_intrinsics(self) -> dict[str, torch.Tensor]:
    method load_mesh (line 105) | def load_mesh(self) -> o3d.geometry.TriangleMesh:
    method __len__ (line 112) | def __len__(self):
    method __getitem__ (line 116) | def __getitem__(self, idx):
  function create_all_occlusion_masks (line 131) | def create_all_occlusion_masks(
  function cli (line 296) | def cli(

FILE: scripts/data_scripts/fix_nerfcapture_filenames.py
  function fix_nerfcapture_filenames (line 6) | def fix_nerfcapture_filenames(

FILE: scripts/data_scripts/generate_hypersim_planar_depths.py
  function crawl_subprocess (line 38) | def crawl_subprocess(opts, scan, count, progress):
  function crawl (line 76) | def crawl(opts, scans):

FILE: scripts/data_scripts/generate_test_tuples.py
  function compute_offline_tuple (line 63) | def compute_offline_tuple(
  function default_dvmvs_tuples (line 159) | def default_dvmvs_tuples(scan, poses, dists_to_last_valid, n_measurement...
  function offline_dvmvs_tuples (line 213) | def offline_dvmvs_tuples(scan, poses, n_measurement_frames):
  function dense_dvmvs_tuples (line 262) | def dense_dvmvs_tuples(scan, poses, n_measurement_frames):
  function offline_dense_dvmvs_tuples (line 336) | def offline_dense_dvmvs_tuples(scan, poses, n_measurement_frames):
  function crawl_subprocess_long (line 374) | def crawl_subprocess_long(opts_temp_filepath, scan, count, progress):
  function crawl (line 517) | def crawl(opts_temp_filepath, opts, scans):

FILE: scripts/data_scripts/generate_train_tuples.py
  function gather_pairs_train (line 54) | def gather_pairs_train(
  function crawl_subprocess_short (line 137) | def crawl_subprocess_short(opts_temp_filepath, scan, count, progress):
  function crawl_subprocess_long (line 228) | def crawl_subprocess_long(opts_temp_filepath, scan, count, progress):
  function crawl (line 392) | def crawl(opts_temp_filepath, opts, scans):

FILE: scripts/data_scripts/generate_train_tuples_geometry.py
  function crawl_subprocess_long (line 54) | def crawl_subprocess_long(opts_temp_filepath, scan, count, progress):
  function crawl (line 222) | def crawl(opts_temp_filepath, opts, scans):

FILE: scripts/data_scripts/precompute_valid_frames.py
  function process_scan (line 44) | def process_scan(opts_temp_filepath, scan, count, progress):
  function multi_process_scans (line 91) | def multi_process_scans(opts_temp_filepath, opts, scans):

FILE: scripts/data_scripts/scannet_wrangling_scripts/SensorData.py
  function print_array_on_one_line (line 15) | def print_array_on_one_line():
  class RGBDFrame (line 23) | class RGBDFrame:
    method load (line 24) | def load(self, file_handle):
    method decompress_depth (line 39) | def decompress_depth(self, compression_type):
    method decompress_depth_zlib (line 45) | def decompress_depth_zlib(self):
    method decompress_color (line 48) | def decompress_color(self, compression_type):
    method dump_color_to_file (line 54) | def dump_color_to_file(self, compression_type, filepath):
    method decompress_color_jpeg (line 63) | def decompress_color_jpeg(self):
  class SensorData (line 67) | class SensorData:
    method __init__ (line 68) | def __init__(self, filename):
    method load (line 72) | def load(self, filename):
    method export_depth_images (line 105) | def export_depth_images(self, output_path, image_size=None, frame_skip...
    method export_color_images (line 130) | def export_color_images(self, output_path, image_size=None, frame_skip...
    method save_mat_to_file (line 149) | def save_mat_to_file(self, matrix, filename):
    method export_poses (line 154) | def export_poses(self, output_path, frame_skip=1):
    method export_intrinsics (line 163) | def export_intrinsics(self, output_path, scan_name):

FILE: scripts/data_scripts/scannet_wrangling_scripts/download_scannet.py
  function get_release_scans (line 53) | def get_release_scans(release_file):
  function download_release (line 62) | def download_release(release_scans, out_dir, file_types, use_v1_sens):
  function download_file (line 72) | def download_file(url, out_file):
  function download_scan (line 87) | def download_scan(scan_id, out_dir, file_types, use_v1_sens):
  function download_task_data (line 103) | def download_task_data(out_dir):
  function download_tfrecords (line 122) | def download_tfrecords(in_dir, out_dir):
  function download_label_map (line 141) | def download_label_map(out_dir):
  function main (line 154) | def main():

FILE: scripts/data_scripts/scannet_wrangling_scripts/reader.py
  function process_scan (line 38) | def process_scan(opt, scan_job, count=None, progress=None):
  function main (line 65) | def main():

FILE: scripts/data_scripts/undistort_nerfstudio_data.py
  function undistort_nerfstudio_data (line 11) | def undistort_nerfstudio_data(

FILE: scripts/dust3r_waymo_preprocess.py
  function colmap_to_opencv_intrinsics (line 38) | def colmap_to_opencv_intrinsics(K):
  function opencv_to_colmap_intrinsics (line 51) | def opencv_to_colmap_intrinsics(K):
  function camera_matrix_of_crop (line 63) | def camera_matrix_of_crop(input_camera_matrix, input_resolution, output_...
  function geotrf (line 78) | def geotrf(Trf, pts, ncol=None, norm=False):
  function inv (line 142) | def inv(mat):
  function imread_cv2 (line 153) | def imread_cv2(path, options=cv2.IMREAD_COLOR):
  class ImageList (line 166) | class ImageList:
    method __init__ (line 170) | def __init__(self, images):
    method __len__ (line 179) | def __len__(self):
    method to_pil (line 182) | def to_pil(self):
    method size (line 186) | def size(self):
    method resize (line 191) | def resize(self, *args, **kwargs):
    method crop (line 194) | def crop(self, *args, **kwargs):
    method _dispatch (line 197) | def _dispatch(self, func, *args, **kwargs):
  function rescale_image_depthmap (line 201) | def rescale_image_depthmap(image, depthmap, camera_intrinsics, output_re...
  function starcall (line 237) | def starcall(args):
  function starstarcall (line 243) | def starstarcall(args):
  function parallel_map (line 249) | def parallel_map(function, args, workers=0, star_args=False, kw_args=Fal...
  function parallel_processes (line 299) | def parallel_processes(*args, **kwargs):
  function show_raw_pointcloud (line 308) | def show_raw_pointcloud(pts3d, colors, point_size=2):
  function main (line 317) | def main(waymo_root, output_dir, subslices, do_extract_frames, workers=1):
  function _list_sequences (line 326) | def _list_sequences(db_root):
  function make_subsequences (line 332) | def make_subsequences(input_list):
  function get_subsliced_sequences (line 339) | def get_subsliced_sequences(db_root, subslices):
  function extract_frames (line 349) | def extract_frames(db_root, output_dir, sequences, workers=8):
  function process_one_seq (line 358) | def process_one_seq(db_root, output_dir, seq):
  function extract_frames_one_seq (line 387) | def extract_frames_one_seq(filename):
  function make_crops (line 450) | def make_crops(output_dir, sequences, workers=16, **kw):
  function crop_one_seq (line 457) | def crop_one_seq(input_dir, output_dir, seq, resolution=640):

FILE: scripts/evals/mesh_eval.py
  function main (line 30) | def main():

FILE: scripts/evaluation.py
  function run (line 9) | def run():
  function incremental (line 26) | def incremental(checkpoint: str, output_dir: Path):

FILE: scripts/render_scripts/render_meshes.py
  class SimpleScanNetDataset (line 55) | class SimpleScanNetDataset(torch.utils.data.Dataset):
    method __init__ (line 58) | def __init__(self, scan_name: str, scan_data_root: Path, tuple_filepat...
    method _get_available_frames (line 70) | def _get_available_frames(self, tuple_filepath: str):
    method load_pose (line 80) | def load_pose(self, frame_ind) -> dict[str, torch.Tensor]:
    method load_intrinsics (line 94) | def load_intrinsics(self) -> dict[str, torch.Tensor]:
    method load_mesh (line 112) | def load_mesh(self) -> o3d.geometry.TriangleMesh:
    method __len__ (line 118) | def __len__(self):
    method __getitem__ (line 122) | def __getitem__(self, idx):
  function render_scene_meshes (line 136) | def render_scene_meshes(
  function render_scene_meshes_partial (line 256) | def render_scene_meshes_partial(
  function render_scenes (line 380) | def render_scenes(

FILE: simple_demo.py
  function main (line 8) | def main():

FILE: src/mvsanywhere/datasets/blendedmvg.py
  class BlendedMVGDataset (line 11) | class BlendedMVGDataset(GenericMVSDataset):
    method __init__ (line 27) | def __init__(
    method get_sub_folder_dir (line 125) | def get_sub_folder_dir(split):
    method get_frame_id_string (line 129) | def get_frame_id_string(self, frame_id):
    method get_color_filepath (line 138) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 155) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 172) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 187) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 204) | def get_pose_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 217) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 288) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 331) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 360) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/change_of_basis.py
  class ChangeOfBasis (line 4) | class ChangeOfBasis:
    method convert_matrix_to_vision_convention (line 51) | def convert_matrix_to_vision_convention(cls, pose: np.ndarray) -> np.n...
    method convert_arkit_to_vision_convention (line 57) | def convert_arkit_to_vision_convention(cls, pose: np.ndarray) -> np.nd...
    method convert_landscape_to_portrait (line 63) | def convert_landscape_to_portrait(cls, pose: np.ndarray) -> np.ndarray:
    method convert_portrait_to_landscape (line 69) | def convert_portrait_to_landscape(cls, pose: np.ndarray) -> np.ndarray:
    method convert_ned_to_vision_convention (line 75) | def convert_ned_to_vision_convention(cls, pose: np.ndarray) -> np.ndar...

FILE: src/mvsanywhere/datasets/colmap_dataset.py
  class ColmapDataset (line 24) | class ColmapDataset(GenericMVSDataset):
    method __init__ (line 63) | def __init__(
    method get_frame_id_string (line 157) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 166) | def get_valid_frame_path(self, split, scan):
    method get_valid_frame_ids (line 181) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method load_pose (line 258) | def load_pose(self, scan_id, frame_id):
    method load_intrinsics (line 289) | def load_intrinsics(self, scan_id, frame_id=None, flip=None):
    method load_capture_poses (line 382) | def load_capture_poses(self, scan_id):
    method load_target_size_depth_and_mask (line 408) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 424) | def load_full_res_depth_and_mask(self, scan_id, frame_id, crop=None):
    method get_color_filepath (line 440) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 456) | def get_high_res_color_filepath(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/dynamic_replica.py
  class DynamicReplicaFrameAnnotation (line 24) | class DynamicReplicaFrameAnnotation(ImplicitronFrameAnnotation):
  class DynamicReplicaDataset (line 30) | class DynamicReplicaDataset(GenericMVSDataset):
    method __init__ (line 41) | def __init__(
    method get_sub_folder_dir (line 149) | def get_sub_folder_dir(split):
    method get_frame_id_string (line 153) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 162) | def get_valid_frame_path(self, split, scan):
    method _get_frame_ids (line 177) | def _get_frame_ids(self, split, scan):
    method get_valid_frame_ids (line 183) | def get_valid_frame_ids(self, split, scan, store_computed=False):
    method get_color_filepath (line 270) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 297) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 314) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 331) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method _get_opencv_camera (line 349) | def _get_opencv_camera(self, frame_annotation):
    method load_intrinsics (line 398) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 464) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 510) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 543) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/generic_mvs_dataset.py
  class GenericMVSDataset (line 21) | class GenericMVSDataset(Dataset):
    method __init__ (line 49) | def __init__(
    method __len__ (line 233) | def __len__(self):
    method get_sub_folder_dir (line 237) | def get_sub_folder_dir(split):
    method get_valid_frame_path (line 241) | def get_valid_frame_path(self, split, scan):
    method get_valid_frame_ids (line 247) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method get_color_filepath (line 269) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 285) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 302) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 317) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 333) | def get_pose_filepath(self, scan_id, frame_id):
    method get_frame_id_string (line 346) | def get_frame_id_string(self, frame_id):
    method get_gt_mesh_path (line 355) | def get_gt_mesh_path(dataset_path, split, scan_id):
    method load_intrinsics (line 361) | def load_intrinsics(self, scan_id, frame_id=None, flip=None):
    method load_target_size_depth_and_mask (line 383) | def load_target_size_depth_and_mask(self, scan_id, frame_id):
    method load_full_res_depth_and_mask (line 404) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 422) | def load_pose(self, scan_id, frame_id):
    method load_color (line 438) | def load_color(self, scan_id, frame_id, crop=None):
    method load_high_res_color (line 469) | def load_high_res_color(self, scan_id, frame_id):
    method get_frame (line 497) | def get_frame(self, scan_id, frame_id, load_depth, flip=False):
    method stack_src_data (line 676) | def stack_src_data(self, src_data):
    method __getitem__ (line 689) | def __getitem__(self, idx):

FILE: src/mvsanywhere/datasets/hypersim.py
  class HypersimDataset (line 17) | class HypersimDataset(GenericMVSDataset):
    method __init__ (line 28) | def __init__(
    method get_frame_id_string (line 125) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 134) | def get_valid_frame_path(self, split, scan):
    method _get_frame_ids (line 151) | def _get_frame_ids(self, split, scan):
    method _check_hypersim_img_not_anomalous (line 165) | def _check_hypersim_img_not_anomalous(self, img, threshold=0.3):
    method get_valid_frame_ids (line 186) | def get_valid_frame_ids(self, split, scan, store_computed=False):
    method get_color_filepath (line 271) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 310) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 336) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 353) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_full_res_distance_filepath (line 380) | def get_full_res_distance_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 406) | def get_pose_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 426) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 533) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 583) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method _get_cam_position (line 615) | def _get_cam_position(self, pose_path, frame):
    method _get_cam_orientation (line 632) | def _get_cam_orientation(self, pose_path, frame):
    method load_pose (line 649) | def load_pose(self, scan_id, frame_id):
    method _get_M_cam_from_uv (line 715) | def _get_M_cam_from_uv(self, scan_id):
    method _get_rays_hypersim_torch (line 749) | def _get_rays_hypersim_torch(self, intrinsics_inv, H, W):
    method _get_prependicular_depths (line 786) | def _get_prependicular_depths(self, scan_id, frame_id):
    method _save_prependicular_depths_to_disk (line 815) | def _save_prependicular_depths_to_disk(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/matrix_city.py
  class MatrixCityDataset (line 18) | class MatrixCityDataset(GenericMVSDataset):
    method __init__ (line 29) | def __init__(
    method get_frame_id_string (line 137) | def get_frame_id_string(self, frame_id):
    method format_frame_id (line 146) | def format_frame_id(self, scan, frame_id):
    method get_valid_frame_path (line 158) | def get_valid_frame_path(self, split, scan):
    method _load_transforms_file (line 173) | def _load_transforms_file(self, scan):
    method _get_frame_ids (line 176) | def _get_frame_ids(self, split, scan):
    method get_valid_frame_ids (line 182) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method get_color_filepath (line 266) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 298) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 319) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 336) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 361) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method _load_depth (line 434) | def _load_depth(depth_path):
    method load_target_size_depth_and_mask (line 441) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 493) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 528) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/mvssynth.py
  class MVSSynthDataset (line 13) | class MVSSynthDataset(GenericMVSDataset):
    method __init__ (line 24) | def __init__(
    method get_valid_frame_path (line 123) | def get_valid_frame_path(self, split, scan):
    method get_valid_frame_ids (line 128) | def get_valid_frame_ids(self, split, scan, store_computed=False):
    method get_color_filepath (line 174) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 190) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 204) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 219) | def get_pose_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 233) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 311) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_pose (line 351) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/nerf_dataset.py
  class NeRFDataset (line 14) | class NeRFDataset(GenericMVSDataset):
    method __init__ (line 25) | def __init__(
    method get_frame_id_string (line 117) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 126) | def get_valid_frame_path(self, split, scan):
    method get_valid_frame_ids (line 131) | def get_valid_frame_ids(self, split, scan, store_computed=False):
    method get_color_filepath (line 190) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 206) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 222) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 236) | def get_pose_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 250) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 328) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 358) | def load_full_res_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_pose (line 388) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/nerfstudio_dataset.py
  class NerfStudioDataset (line 15) | class NerfStudioDataset(GenericMVSDataset):
    method __init__ (line 26) | def __init__(
    method get_frame_id_string (line 129) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 138) | def get_valid_frame_path(self, split, scan):
    method _get_frame_ids (line 153) | def _get_frame_ids(self, split, scan):
    method get_valid_frame_ids (line 161) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method load_pose (line 232) | def load_pose(self, scan_id, frame_id):
    method load_intrinsics (line 260) | def load_intrinsics(self, scan_id, frame_id, flip=None):
    method load_capture_metadata (line 343) | def load_capture_metadata(self, scan_id):
    method get_cached_depth_filepath (line 376) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_cached_confidence_filepath (line 392) | def get_cached_confidence_filepath(self, scan_id, frame_id, crop=None):
    method get_full_res_depth_filepath (line 408) | def get_full_res_depth_filepath(self, scan_id, frame_id, crop=None):
    method get_full_res_confidence_filepath (line 425) | def get_full_res_confidence_filepath(self, scan_id, frame_id):
    method load_full_res_depth_and_mask (line 441) | def load_full_res_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_target_size_depth_and_mask (line 464) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method get_color_filepath (line 485) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 509) | def get_high_res_color_filepath(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/read_write_colmap_model.py
  class Image (line 52) | class Image(BaseImage):
    method qvec2rotmat (line 53) | def qvec2rotmat(self):
  function read_next_bytes (line 78) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact...
  function write_next_bytes (line 90) | def write_next_bytes(fid, data, format_char_sequence, endian_character="...
  function read_cameras_text (line 106) | def read_cameras_text(path):
  function read_cameras_binary (line 136) | def read_cameras_binary(path_to_model_file):
  function write_cameras_text (line 171) | def write_cameras_text(cameras, path):
  function write_cameras_binary (line 190) | def write_cameras_binary(cameras, path_to_model_file):
  function read_images_text (line 207) | def read_images_text(path):
  function read_images_binary (line 247) | def read_images_binary(path_to_model_file):
  function write_images_text (line 297) | def write_images_text(images, path):
  function write_images_binary (line 337) | def write_images_binary(images, path_to_model_file):
  function read_points3D_text (line 358) | def read_points3D_text(path):
  function read_points3D_binary (line 390) | def read_points3D_binary(path_to_model_file):
  function write_points3D_text (line 428) | def write_points3D_text(points3D, path):
  function write_points3D_binary (line 459) | def write_points3D_binary(points3D, path_to_model_file):
  function detect_model_format (line 478) | def detect_model_format(path, ext):
  function read_model (line 490) | def read_model(path, ext=""):
  function write_model (line 512) | def write_model(cameras, images, points3D, path, ext=".bin"):
  function qvec2rotmat (line 524) | def qvec2rotmat(qvec):
  function rotmat2qvec (line 546) | def rotmat2qvec(R):

FILE: src/mvsanywhere/datasets/sailvos3d.py
  class SAILVOS3DDataset (line 18) | class SAILVOS3DDataset(GenericMVSDataset):
    method __init__ (line 29) | def __init__(
    method get_frame_id_string (line 130) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 139) | def get_valid_frame_path(self, split, scan):
    method _get_frame_ids (line 149) | def _get_frame_ids(self, split, scan):
    method get_valid_frame_ids (line 153) | def get_valid_frame_ids(self, split, scan, store_computed=False):
    method get_color_filepath (line 239) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 267) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 283) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 300) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 316) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method _load_rage_matrices (line 391) | def _load_rage_matrices(self, scan_id, frame_id):
    method convert_ndc_depth_to_cam (line 411) | def convert_ndc_depth_to_cam(
    method load_target_size_depth_and_mask (line 466) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method pixels_to_ndcs (line 523) | def pixels_to_ndcs(
    method load_full_res_depth_and_mask (line 552) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 588) | def load_pose(self, scan_id, frame_id):
    method compute_intrinsics_from_P (line 625) | def compute_intrinsics_from_P(P: np.ndarray, image_width: int, image_h...

FILE: src/mvsanywhere/datasets/scannet_dataset.py
  class ScannetDataset (line 11) | class ScannetDataset(GenericMVSDataset):
    method __init__ (line 77) | def __init__(
    method get_sub_folder_dir (line 175) | def get_sub_folder_dir(split):
    method get_frame_id_string (line 182) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 191) | def get_valid_frame_path(self, split, scan):
    method get_valid_frame_ids (line 199) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method get_gt_mesh_path (line 297) | def get_gt_mesh_path(dataset_path, split, scan_id):
    method get_color_filepath (line 309) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 336) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 364) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 387) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 406) | def get_pose_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 423) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 494) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 537) | def load_full_res_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_pose (line 576) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/tartanair.py
  class TartanAirDataset (line 16) | class TartanAirDataset(GenericMVSDataset):
    method __init__ (line 27) | def __init__(
    method get_frame_id_string (line 124) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 133) | def get_valid_frame_path(self, split, scan):
    method _get_frame_ids (line 148) | def _get_frame_ids(self, split, scan):
    method get_valid_frame_ids (line 155) | def get_valid_frame_ids(self, split, scan, store_computed=False):
    method get_color_filepath (line 238) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 271) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 292) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 309) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_full_res_classgt_filepath (line 330) | def get_full_res_classgt_filepath(self, scan_id, frame_id):
    method get_pose_filepath (line 352) | def get_pose_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 367) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method load_target_size_depth_and_mask (line 435) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 499) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 531) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/vdr_dataset.py
  class VDRDataset (line 18) | class VDRDataset(GenericMVSDataset):
    method __init__ (line 31) | def __init__(
    method get_sub_folder_dir (line 89) | def get_sub_folder_dir(split):
    method get_frame_id_string (line 92) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 101) | def get_valid_frame_path(self, split, scan):
    method get_valid_frame_ids (line 108) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method load_pose (line 178) | def load_pose(self, scan_id, frame_id):
    method load_intrinsics (line 214) | def load_intrinsics(self, scan_id, frame_id, flip=None):
    method load_capture_metadata (line 294) | def load_capture_metadata(self, scan_id):
    method get_cached_depth_filepath (line 322) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_cached_confidence_filepath (line 355) | def get_cached_confidence_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 388) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_full_res_confidence_filepath (line 410) | def get_full_res_confidence_filepath(self, scan_id, frame_id):
    method load_full_res_depth_and_mask (line 431) | def load_full_res_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_target_size_depth_and_mask (line 474) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method get_color_filepath (line 548) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 573) | def get_high_res_color_filepath(self, scan_id, frame_id):

FILE: src/mvsanywhere/datasets/vkitti.py
  class VirtualKITTIDataset (line 15) | class VirtualKITTIDataset(GenericMVSDataset):
    method __init__ (line 26) | def __init__(
    method get_frame_id_string (line 123) | def get_frame_id_string(self, frame_id):
    method get_valid_frame_path (line 132) | def get_valid_frame_path(self, split, scan):
    method _get_frame_ids (line 147) | def _get_frame_ids(self, split, scan):
    method get_valid_frame_ids (line 156) | def get_valid_frame_ids(self, split, scan, store_computed=True):
    method get_color_filepath (line 238) | def get_color_filepath(self, scan_id, frame_id):
    method get_high_res_color_filepath (line 275) | def get_high_res_color_filepath(self, scan_id, frame_id):
    method get_cached_depth_filepath (line 298) | def get_cached_depth_filepath(self, scan_id, frame_id):
    method get_full_res_depth_filepath (line 315) | def get_full_res_depth_filepath(self, scan_id, frame_id):
    method get_full_res_classgt_filepath (line 338) | def get_full_res_classgt_filepath(self, scan_id, frame_id):
    method load_intrinsics (line 361) | def load_intrinsics(self, scan_id, frame_id=None, flip=False):
    method _load_depth (line 433) | def _load_depth(depth_path, is_float16=True):
    method _load_classgt (line 438) | def _load_classgt(classgt_path, is_float16=True):
    method load_target_size_depth_and_mask (line 442) | def load_target_size_depth_and_mask(self, scan_id, frame_id, crop=None):
    method load_full_res_depth_and_mask (line 498) | def load_full_res_depth_and_mask(self, scan_id, frame_id):
    method load_pose (line 528) | def load_pose(self, scan_id, frame_id):

FILE: src/mvsanywhere/experiment_modules/rmvd_mvsa.py
  class MVSA_Wrapped (line 17) | class MVSA_Wrapped(nn.Module):
    method __init__ (line 18) | def __init__(self, opts, use_refinement=False, **kwargs):
    method input_adapter (line 38) | def input_adapter(self, images, keyview_idx, poses=None, intrinsics=No...
    method forward (line 97) | def forward(
    method output_adapter (line 159) | def output_adapter(self, model_output):

FILE: src/mvsanywhere/experiment_modules/sr_depth_model.py
  class DepthModel (line 44) | class DepthModel(pl.LightningModule):
    method __init__ (line 92) | def __init__(self, opts):
    method compute_matching_feats (line 266) | def compute_matching_feats(
    method get_heuristic_depth_range (line 321) | def get_heuristic_depth_range(self, cur_data, src_data):
    method refine_depth_range (line 339) | def refine_depth_range(self, cur_data, src_data, outputs):
    method forward (line 370) | def forward(
    method compute_losses (line 593) | def compute_losses(self, cur_data, src_data, outputs):
    method step (line 702) | def step(self, phase, batch, batch_idx):
    method training_step (line 888) | def training_step(self, batch, batch_idx):
    method validation_step (line 892) | def validation_step(self, batch, batch_idx):
    method configure_optimizers (line 896) | def configure_optimizers(self):

FILE: src/mvsanywhere/losses.py
  class MSGradientLoss (line 11) | class MSGradientLoss(nn.Module):
    method __init__ (line 12) | def __init__(self, num_scales: int = 4):
    method forward (line 17) | def forward(self, depth_gt: Tensor, depth_pred: Tensor) -> Tensor:
  class ScaleInvariantLoss (line 39) | class ScaleInvariantLoss(jit.ScriptModule):
    method __init__ (line 40) | def __init__(self, si_lambda: float = 0.85):
    method forward (line 46) | def forward(self, log_depth_gt: Tensor, log_depth_pred: Tensor) -> Ten...
  class NormalsLoss (line 54) | class NormalsLoss(nn.Module):
    method forward (line 55) | def forward(self, normals_gt_b3hw: Tensor, normals_pred_b3hw: Tensor) ...
  class MVDepthLoss (line 78) | class MVDepthLoss(nn.Module):
    method __init__ (line 79) | def __init__(self, height, width):
    method _check_warped_image (line 88) | def _check_warped_image(
    method get_valid_mask (line 134) | def get_valid_mask(
    method get_error_for_pair (line 174) | def get_error_for_pair(
    method forward (line 211) | def forward(

FILE: src/mvsanywhere/modules/cost_volume.py
  class CostVolumeManager (line 10) | class CostVolumeManager(nn.Module):
    method __init__ (line 23) | def __init__(
    method initialise_for_projection (line 52) | def initialise_for_projection(self, device=None):
    method get_mask (line 74) | def get_mask(self, pix_coords_bk2hw):
    method generate_depth_planes (line 97) | def generate_depth_planes(
    method warp_features (line 133) | def warp_features(
    method build_cost_volume (line 219) | def build_cost_volume(
    method indices_to_disparity (line 317) | def indices_to_disparity(self, indices, depth_planes_bdhw):
    method forward (line 322) | def forward(
    method to_fast (line 365) | def to_fast(self) -> "FastCostVolumeManager":
  class FastCostVolumeManager (line 374) | class FastCostVolumeManager(CostVolumeManager):
    method __init__ (line 375) | def __init__(self, matching_height, matching_width, num_depth_bins=64,...
    method warp_features (line 380) | def warp_features(
    method build_cost_volume (line 468) | def build_cost_volume(

FILE: src/mvsanywhere/modules/depth_anything_blocks.py
  function _make_scratch (line 13) | def _make_scratch(in_shape, out_shape, groups=1, expand=False):
  function _make_fusion_block (line 46) | def _make_fusion_block(features, use_bn, size = None):
  class ResidualConvUnit (line 58) | class ResidualConvUnit(nn.Module):
    method __init__ (line 62) | def __init__(self, features, activation, bn):
    method forward (line 90) | def forward(self, x):
  class FeatureFusionBlock (line 116) | class FeatureFusionBlock(nn.Module):
    method __init__ (line 120) | def __init__(self, features, activation, deconv=False, bn=False, expan...
    method forward (line 147) | def forward(self, *xs, size=None):
  class DPTHead (line 177) | class DPTHead(nn.Module):
    method __init__ (line 178) | def __init__(
    method set_prediction_scale (line 279) | def set_prediction_scale(self, prediction_scale):
    method forward (line 285) | def forward(self, out_features, patch_h, patch_w):
    method load_da_weights (line 328) | def load_da_weights(self, weights_path):

FILE: src/mvsanywhere/modules/feature_volume.py
  class FeatureVolumeManager (line 12) | class FeatureVolumeManager(CostVolumeManager):
    method __init__ (line 26) | def __init__(
    method build_cost_volume (line 81) | def build_cost_volume(
    method to_fast (line 358) | def to_fast(self) -> "FastFeatureVolumeManager":
  class FastFeatureVolumeManager (line 368) | class FastFeatureVolumeManager(FeatureVolumeManager):
    method __init__ (line 377) | def __init__(
    method warp_features (line 432) | def warp_features(
    method build_cost_volume (line 586) | def build_cost_volume(

FILE: src/mvsanywhere/modules/layers.py
  function conv3x3 (line 7) | def conv3x3(
  function conv1x1 (line 28) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1, bias: bool...
  class BasicBlock (line 33) | class BasicBlock(nn.Module):
    method __init__ (line 36) | def __init__(
    method forward (line 77) | def forward(self, x: Tensor) -> Tensor:
  class TensorFormatter (line 97) | class TensorFormatter(nn.Module):
    method __init__ (line 105) | def __init__(self):
    method _expand_batch_with_channels (line 111) | def _expand_batch_with_channels(self, x):
    method _reduce_batch_to_channels (line 120) | def _reduce_batch_to_channels(self, x):
    method forward (line 130) | def forward(self, x, apply_func):

FILE: src/mvsanywhere/modules/networks.py
  function double_basic_block (line 13) | def double_basic_block(num_ch_in, num_ch_out, num_repeats=2):
  class DepthDecoderPP (line 20) | class DepthDecoderPP(nn.Module):
    method __init__ (line 21) | def __init__(self, num_ch_enc, scales=range(4), num_output_channels=1,...
    method forward (line 65) | def forward(self, input_features):
  class CVEncoder (line 88) | class CVEncoder(nn.Module):
    method __init__ (line 89) | def __init__(self, num_ch_cv, num_ch_enc, num_ch_outs):
    method forward (line 110) | def forward(self, x, img_feats):
  class MLP (line 120) | class MLP(nn.Module):
    method __init__ (line 121) | def __init__(self, channel_list, disable_final_activation=False):
    method forward (line 134) | def forward(self, x):
  class ResnetMatchingEncoder (line 138) | class ResnetMatchingEncoder(nn.Module):
    method __init__ (line 141) | def __init__(
    method forward (line 188) | def forward(self, input_image):
  class UNetMatchingEncoder (line 192) | class UNetMatchingEncoder(nn.Module):
    method __init__ (line 193) | def __init__(self):
    method forward (line 211) | def forward(self, x):

FILE: src/mvsanywhere/modules/networks_fast.py
  class ConvBlock (line 6) | class ConvBlock(nn.Module):
    method __init__ (line 7) | def __init__(self, in_ch, out_ch, use_elu=True, use_bn=False):
    method forward (line 17) | def forward(self, x):
  class ConvUpsampleAndConcatBlock (line 27) | class ConvUpsampleAndConcatBlock(nn.Module):
    method __init__ (line 28) | def __init__(self, in_ch, out_ch, skip_chns, use_elu=True, use_bn=False):
    method forward (line 36) | def forward(self, x, cat_feats):
  class SkipDecoder (line 45) | class SkipDecoder(nn.Module):
    method __init__ (line 46) | def __init__(self, input_channels, use_bn=False):
    method forward (line 79) | def forward(self, features):
  class SkipDecoderRegression (line 98) | class SkipDecoderRegression(SkipDecoder):
    method __init__ (line 99) | def __init__(self, input_channels, use_bn=False):
    method forward (line 134) | def forward(self, features):

FILE: src/mvsanywhere/modules/view_agnostic_feature_volume.py
  class ViewAgnosticFeatureVolumeManager (line 12) | class ViewAgnosticFeatureVolumeManager(CostVolumeManager):
    method __init__ (line 26) | def __init__(
    method build_cost_volume (line 82) | def build_cost_volume(
    method to_fast (line 381) | def to_fast(self) -> "FastViewAgnosticFeatureVolumeManager":
  class FastViewAgnosticFeatureVolumeManager (line 391) | class FastViewAgnosticFeatureVolumeManager(ViewAgnosticFeatureVolumeMana...
    method __init__ (line 400) | def __init__(
    method warp_features (line 430) | def warp_features(
    method build_cost_volume (line 584) | def build_cost_volume(

FILE: src/mvsanywhere/modules/vit_modules.py
  class Attention (line 18) | class Attention(nn.Module):
    method __init__ (line 19) | def __init__(
    method forward (line 38) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class PytorchMemEffAttention (line 53) | class PytorchMemEffAttention(Attention):
    method forward (line 54) | def forward(self, x: torch.Tensor, attn_bias=None) -> torch.Tensor:
  function use_memeffattn_in_model (line 67) | def use_memeffattn_in_model(model):
  class DINOv2 (line 87) | class DINOv2(nn.Module):
    method __init__ (line 95) | def __init__(
    method forward (line 108) | def forward(self, x):
    method load_da_weights (line 123) | def load_da_weights(self, weights_path):
  class CostVolumePatchEmbed (line 133) | class CostVolumePatchEmbed(nn.Module):
    method __init__ (line 135) | def __init__(
    method forward (line 189) | def forward(self, x, img_feats):
    method patch_embed (line 210) | def patch_embed(self, x):
  class ViTCVEncoder (line 216) | class ViTCVEncoder(nn.Module):
    method __init__ (line 218) | def __init__(
    method forward (line 250) | def forward(self, x, img_feats):
    method prepare_tokens_with_masks (line 273) | def prepare_tokens_with_masks(self, x, img_features, masks=None):

FILE: src/mvsanywhere/options.py
  class DataOptions (line 10) | class DataOptions:
  class Options (line 38) | class Options:
  class OptionsHandler (line 257) | class OptionsHandler:
    method __init__ (line 294) | def __init__(self, required_flags=[]):
    method parse_and_merge_options (line 309) | def parse_and_merge_options(self, config_filepaths=None, ignore_cl_arg...
    method populate_argparse (line 376) | def populate_argparse(self):
    method check_required_items (line 390) | def check_required_items(self):
    method merge_config_options (line 396) | def merge_config_options(self, config_options):
    method merge_cl_args (line 404) | def merge_cl_args(self, cl_args):
    method pretty_print_options (line 423) | def pretty_print_options(self):
    method load_options_from_yaml (line 432) | def load_options_from_yaml(config_filepath):
    method save_options_as_yaml (line 437) | def save_options_as_yaml(config_filepath, options):
  function handle_backwards_compat (line 442) | def handle_backwards_compat(opts):

FILE: src/mvsanywhere/run_demo.py
  function prepare_scan_files (line 149) | def prepare_scan_files(opts):
  function init_model (line 208) | def init_model(opts):
  function main (line 216) | def main(opts):

FILE: src/mvsanywhere/test.py
  function main (line 129) | def main(opts):

FILE: src/mvsanywhere/test_rmvd.py
  function main (line 16) | def main(opts):

FILE: src/mvsanywhere/tools/fusers_helper.py
  class DepthFuser (line 14) | class DepthFuser:
    method __init__ (line 15) | def __init__(self, gt_path="", fusion_resolution=0.04, max_fusion_dept...
  class OurFuser (line 20) | class OurFuser(DepthFuser):
    method __init__ (line 33) | def __init__(
    method fuse_frames (line 64) | def fuse_frames(self, depths_b1hw, K_b44, cam_T_world_b44, color_b3hw):
    method export_mesh (line 72) | def export_mesh(self, path, export_single_mesh=True):
    method save_tsdf (line 78) | def save_tsdf(self, path):
    method sample_tsdf (line 81) | def sample_tsdf(self, world_points_N3, what_to_sample="tsdf", sampling...
    method get_mesh (line 98) | def get_mesh(self, export_single_mesh=True, convert_to_trimesh=True):
    method get_mesh_pytorch3d (line 101) | def get_mesh_pytorch3d(self, scale_to_world=True, min_bounds_3=None, m...
  class Open3DFuser (line 107) | class Open3DFuser(DepthFuser):
    method __init__ (line 114) | def __init__(
    method fuse_frames (line 141) | def fuse_frames(
    method export_mesh (line 198) | def export_mesh(self, path, use_marching_cubes_mask=None):
    method get_mesh (line 201) | def get_mesh(self, export_single_mesh=None, convert_to_trimesh=False):
    method save_tsdf (line 209) | def save_tsdf(self, path):
  function get_fuser (line 213) | def get_fuser(opts, scan):
  class CustomOpen3dFuser (line 259) | class CustomOpen3dFuser(Open3DFuser):
    method __init__ (line 260) | def __init__(self, extended_neg_truncation=False, *args, **kwargs):
    method fuse_frames (line 286) | def fuse_frames(
    method update_tsdf_for_voxels (line 372) | def update_tsdf_for_voxels(
    method export_mesh (line 452) | def export_mesh(self, path, use_marching_cubes_mask=None, trim_tsdf_us...
    method get_mesh (line 458) | def get_mesh(
    method get_mesh_pytorch3d (line 491) | def get_mesh_pytorch3d(self, scale_to_world=True):

FILE: src/mvsanywhere/tools/keyframe_buffer.py
  class DVMVS_Config (line 12) | class DVMVS_Config:
  class DVMVS_TartanAir_Config (line 24) | class DVMVS_TartanAir_Config:
  class DVMVS_MatrixCity_Config (line 36) | class DVMVS_MatrixCity_Config:
  function is_pose_available (line 49) | def is_pose_available(pose):
  function is_valid_pair (line 59) | def is_valid_pair(
  function pose_distance (line 81) | def pose_distance(reference_pose, measurement_pose):
  class KeyframeBuffer (line 100) | class KeyframeBuffer:
    method __init__ (line 101) | def __init__(
    method calculate_penalty (line 117) | def calculate_penalty(self, t_score, R_score):
    method try_new_keyframe (line 127) | def try_new_keyframe(self, pose, image, dist_to_last_valid=None, index...
    method get_best_measurement_frames (line 193) | def get_best_measurement_frames(self, n_requested_measurement_frames):
  class SimpleBuffer (line 220) | class SimpleBuffer:
    method __init__ (line 221) | def __init__(
    method try_new_keyframe (line 231) | def try_new_keyframe(self, pose, image, index=None):
    method get_measurement_frames (line 274) | def get_measurement_frames(self):
  class OfflineKeyframeBuffer (line 279) | class OfflineKeyframeBuffer:
    method __init__ (line 280) | def __init__(
    method calculate_penalty (line 297) | def calculate_penalty(self, t_score, R_score):
    method try_new_keyframe (line 307) | def try_new_keyframe(self, pose, image, index=None):
    method get_best_measurement_frames (line 368) | def get_best_measurement_frames(self, n_requested_measurement_frames):
    method get_best_measurement_frames_for_0index (line 394) | def get_best_measurement_frames_for_0index(self, n_requested_measureme...

FILE: src/mvsanywhere/tools/marching_cubes/ext.cpp
  function PYBIND11_MODULE (line 4) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: src/mvsanywhere/tools/marching_cubes/marching_cubes_cpu.cpp
  function MarchingCubesCpu (line 29) | std::tuple<at::Tensor, at::Tensor, at::Tensor> MarchingCubesCpu(

FILE: src/mvsanywhere/tools/marching_cubes/marching_cubes_utils.h
  type Vertex (line 21) | struct Vertex {
  function operator (line 34) | bool operator==(const Vertex& xyz) const {
  function operator (line 39) | bool operator!=(const Vertex& xyz) const {
  type Cube (line 48) | struct Cube {
  function Vertex (line 101) | Vertex VertexInterp(
  function HashVpair (line 138) | int64_t HashVpair(const int edge, int W, int H, int D) {

FILE: src/mvsanywhere/tools/mesh_renderer.py
  class Renderer (line 31) | class Renderer:
    method __init__ (line 37) | def __init__(self, height=480, width=640, flat_render=False, ambient_l...
    method render (line 42) | def render(
    method __call__ (line 76) | def __call__(self, height, width, intrinsics, pose, meshes):
    method fix_pose (line 79) | def fix_pose(self, pose):
    method mesh_opengl (line 89) | def mesh_opengl(self, mesh, mesh_material=None):
    method delete (line 95) | def delete(self):
    method render_mesh (line 98) | def render_mesh(
    method render_mesh_cull_composite (line 141) | def render_mesh_cull_composite(self, alpha, **kwargs):
  function render_colour (line 151) | def render_colour(renderer, meshes, world_T_cam, K, height=256, width=320):
  class SmoothBirdsEyeCamera (line 161) | class SmoothBirdsEyeCamera:
    method __init__ (line 166) | def __init__(
    method get_bird_eye_trans (line 183) | def get_bird_eye_trans(
  function get_cam_pose_from_lookat_and_loc (line 255) | def get_cam_pose_from_lookat_and_loc(
  function camera_marker (line 282) | def camera_marker(
  function get_image_box (line 470) | def get_image_box(
  function create_lights_above_mesh (line 537) | def create_lights_above_mesh(mesh: trimesh.Trimesh, light_intensity: flo...
  function create_light_array (line 553) | def create_light_array(light_type, center_loc, x_length=10.0, y_length=1...
  function transform_trimesh (line 576) | def transform_trimesh(mesh, transform):

FILE: src/mvsanywhere/tools/partial_fuser.py
  class PartialFuser (line 11) | class PartialFuser:
    method __init__ (line 12) | def __init__(
    method get_mesh (line 42) | def get_mesh(self, query_frame_id: int):
    method fuse_all_frames (line 85) | def fuse_all_frames(self):

FILE: src/mvsanywhere/tools/tsdf.py
  function get_frustum_bounds (line 15) | def get_frustum_bounds(
  class TSDF (line 53) | class TSDF:
    method __init__ (line 62) | def __init__(
    method from_file (line 87) | def from_file(cls, tsdf_file):
    method from_mesh (line 100) | def from_mesh(cls, mesh: trimesh.Trimesh, voxel_size: float):
    method from_bounds (line 123) | def from_bounds(cls, bounds: dict, voxel_size: float):
    method generate_voxel_coords (line 157) | def generate_voxel_coords(
    method cuda (line 168) | def cuda(self):
    method cpu (line 175) | def cpu(self):
    method to_mesh (line 182) | def to_mesh(self, scale_to_world=True, export_single_mesh=False):
    method to_mesh_pytorch3d (line 216) | def to_mesh_pytorch3d(self, scale_to_world=True, min_bounds_3=None, ma...
    method save_mesh (line 257) | def save_mesh(self, savepath, filename):
    method save_tsdf (line 267) | def save_tsdf(self, filepath):
    method sample_tsdf (line 277) | def sample_tsdf(self, world_points_N3, what_to_sample="tsdf", sampling...
  class TSDFFuser (line 342) | class TSDFFuser:
    method __init__ (line 347) | def __init__(self, tsdf, min_depth=0.5, max_depth=5.0, use_gpu=True):
    method voxel_coords_3hwd (line 374) | def voxel_coords_3hwd(self):
    method voxel_hashset (line 378) | def voxel_hashset(self):
    method tsdf_values (line 382) | def tsdf_values(self):
    method tsdf_weights (line 386) | def tsdf_weights(self):
    method voxel_size (line 390) | def voxel_size(self):
    method shape (line 394) | def shape(self):
    method truncation (line 398) | def truncation(self):
    method project_to_camera (line 401) | def project_to_camera(self, cam_T_world_T_144, K_144, valid_voxels_14N):
    method integrate_depth (line 414) | def integrate_depth(

FILE: src/mvsanywhere/tools/tuple_generator.py
  function compute_offline_tuple (line 63) | def compute_offline_tuple(
  function default_dvmvs_tuples (line 159) | def default_dvmvs_tuples(scan, poses, dists_to_last_valid, n_measurement...
  function offline_dvmvs_tuples (line 213) | def offline_dvmvs_tuples(scan, poses, n_measurement_frames):
  function dense_dvmvs_tuples (line 262) | def dense_dvmvs_tuples(scan, poses, n_measurement_frames):
  function offline_dense_dvmvs_tuples (line 336) | def offline_dense_dvmvs_tuples(scan, poses, n_measurement_frames):
  function crawl_subprocess_long (line 374) | def crawl_subprocess_long(opts, scan, count, progress):

FILE: src/mvsanywhere/train.py
  function prepare_dataloaders (line 37) | def prepare_dataloaders(opts: options.Options) -> Tuple[List[DataLoader]...
  function prepare_callbacks (line 114) | def prepare_callbacks(
  function prepare_model (line 143) | def prepare_model(opts: options.Options) -> torch.nn.Module:
  function prepare_ddp_strategy (line 185) | def prepare_ddp_strategy(opts: options.Options) -> Strategy:
  function prepare_trainer (line 199) | def prepare_trainer(
  function main (line 241) | def main(opts):

FILE: src/mvsanywhere/utils/augmentation_utils.py
  class CustomColorJitter (line 13) | class CustomColorJitter(torch.nn.Module):
    method __init__ (line 16) | def __init__(
    method forward (line 33) | def forward(self, x: torch.Tensor, denormalize_first=False) -> torch.T...

FILE: src/mvsanywhere/utils/cropping_utils.py
  function find_image_bounding_box (line 4) | def find_image_bounding_box(image: np.ndarray) -> tuple[int, int, int, i...
  function find_image_collection_bounding_box (line 47) | def find_image_collection_bounding_box(images: list[np.ndarray]) -> tupl...
  function tightly_crop_images (line 68) | def tightly_crop_images(images: list[np.ndarray]):

FILE: src/mvsanywhere/utils/dataset_utils.py
  function get_dataset (line 20) | def get_dataset(dataset_name, split_filepath, single_debug_scan_id=None,...

FILE: src/mvsanywhere/utils/generic_utils.py
  function copy_code_state (line 18) | def copy_code_state(path):
  function readlines (line 40) | def readlines(filepath):
  function normalize_depth_single (line 47) | def normalize_depth_single(depth_11hw, mask_11hw, robust=False):
  function normalize_depth (line 77) | def normalize_depth(depth_b1hw: torch.Tensor, mask_b1hw: torch.Tensor = ...
  function pyrdown (line 89) | def pyrdown(input_tensor: torch.Tensor, num_scales: int = 4):
  function upsample (line 98) | def upsample(x):
  function batched_trace (line 110) | def batched_trace(mat_bNN):
  function tensor_B_to_bM (line 114) | def tensor_B_to_bM(tensor_BS, batch_size, num_views):
  function tensor_bM_to_B (line 125) | def tensor_bM_to_B(tensor_bMS):
  function combine_dims (line 137) | def combine_dims(x, dim_begin, dim_end):
  function to_gpu (line 143) | def to_gpu(input_dict, key_ignores=[]):
  function imagenet_normalize (line 153) | def imagenet_normalize(image):
  function reverse_imagenet_normalize (line 159) | def reverse_imagenet_normalize(image):
  function fov_to_image_dimension (line 170) | def fov_to_image_dimension(fov_degrees: float, focal_length: float) -> f...
  function crop_or_pad (line 177) | def crop_or_pad(image_bchw, new_height, new_width, pad_mode="constant"):
  function read_image_file (line 224) | def read_image_file(
  function read_pfm_file (line 280) | def read_pfm_file(
  function crop_image_to_target_ratio (line 348) | def crop_image_to_target_ratio(image, target_aspect_ratio=4.0 / 3.0):
  function cache_model_outputs (line 380) | def cache_model_outputs(
  function get_generic_eps (line 431) | def get_generic_eps(tensor: Optional[Any] = None):

FILE: src/mvsanywhere/utils/geometry_utils.py
  function to_homogeneous (line 12) | def to_homogeneous(input_tensor: Tensor, dim: int = 0) -> Tensor:
  class BackprojectDepth (line 22) | class BackprojectDepth(jit.ScriptModule):
    method __init__ (line 28) | def __init__(self, height: int, width: int):
    method forward (line 55) | def forward(self, depth_b1hw: Tensor, invK_b44: Tensor) -> Tensor:
  class Project3D (line 66) | class Project3D(jit.ScriptModule):
    method __init__ (line 71) | def __init__(self, eps: float = 1e-8):
    method forward (line 77) | def forward(self, points_b4N: Tensor, K_b44: Tensor, cam_T_world_b44: ...
  class NormalGenerator (line 96) | class NormalGenerator(jit.ScriptModule):
    method __init__ (line 97) | def __init__(
    method forward (line 117) | def forward(self, depth_b1hw: Tensor, invK_b44: Tensor) -> Tensor:
  function get_angle_dif (line 149) | def get_angle_dif(matA_b33, matB_b33):
  function get_camera_rays (line 157) | def get_camera_rays(
  function pose_distance (line 191) | def pose_distance(pose_b44):
  function qvec2rotmat (line 206) | def qvec2rotmat(qvec):
  function rotx (line 231) | def rotx(t):
  function roty (line 240) | def roty(t):
  function rotz (line 249) | def rotz(t):

FILE: src/mvsanywhere/utils/metrics_utils.py
  function compute_depth_metrics (line 7) | def compute_depth_metrics(gt, pred, mult_a=False):
  function compute_depth_metrics_batched (line 51) | def compute_depth_metrics_batched(gt_bN, pred_bN, valid_masks_bN, mult_a...
  class ResultsAverager (line 122) | class ResultsAverager:
    method __init__ (line 127) | def __init__(self, exp_name, metrics_name):
    method update_results (line 142) | def update_results(self, elem_metrics):
    method print_sheets_friendly (line 162) | def print_sheets_friendly(
    method output_json (line 200) | def output_json(self, filepath, print_running_metrics=False):
    method load_scores (line 236) | def load_scores(self, filepath, print_running_metrics=False):
    method pretty_print_results (line 252) | def pretty_print_results(self, print_exp_name=True, print_running_metr...
    method compute_final_average (line 275) | def compute_final_average(self, ignore_nans=False):

FILE: src/mvsanywhere/utils/model_utils.py
  function get_model_class (line 10) | def get_model_class(opts):
  function load_model_inference (line 18) | def load_model_inference(opts, model_class_to_use):
  function load_model_training (line 62) | def load_model_training(opts, model_class_to_use):

FILE: src/mvsanywhere/utils/pytorch3d_extras.py
  class _marching_cubes (line 20) | class _marching_cubes(Function):
    method forward (line 28) | def forward(ctx, vol, isolevel, active_voxels, min_bounds, max_bounds):
    method backward (line 35) | def backward(ctx, grad_verts, grad_faces):
  function marching_cubes (line 39) | def marching_cubes(

FILE: src/mvsanywhere/utils/rendering_utils.py
  class PyTorch3DMeshDepthRenderer (line 9) | class PyTorch3DMeshDepthRenderer:
    method __init__ (line 10) | def __init__(self, height=192, width=256) -> None:
    method render (line 25) | def render(self, mesh, cam_T_world_b44, K_b44):
  function load_and_preprocess_mesh_for_rendering (line 50) | def load_and_preprocess_mesh_for_rendering(

FILE: src/mvsanywhere/utils/visualization_utils.py
  function colormap_image (line 15) | def colormap_image(
  function image_tensor3hw_to_numpyhw3 (line 75) | def image_tensor3hw_to_numpyhw3(
  function tile_images (line 93) | def tile_images(
  function save_viz_video_frames (line 200) | def save_viz_video_frames(frame_list, path, fps=30):
  function quick_viz_export (line 210) | def quick_viz_export(
  function load_and_merge_images (line 323) | def load_and_merge_images(frame_ids, quick_viz_directory, fps=30):

FILE: src/mvsanywhere/utils/volume_utils.py
  class SimpleVolume (line 10) | class SimpleVolume:
    method __init__ (line 18) | def __init__(
    method from_bounds (line 46) | def from_bounds(cls, bounds: dict, voxel_size: float):
    method generate_voxel_coords_3hwd (line 96) | def generate_voxel_coords_3hwd(
    method load (line 116) | def load(cls, filepath) -> Any:
    method save (line 127) | def save(self, filepath):
    method cuda (line 137) | def cuda(self):
    method cpu (line 144) | def cpu(self):
    method to_point_cloud (line 151) | def to_point_cloud(self, threshold: Optional[float] = None, num_points...
    method sample_volume (line 185) | def sample_volume(self, world_points_N3):
    method project_volume_to_camera (line 239) | def project_volume_to_camera(self, cam_T_world_b44: torch.Tensor, K_b4...
  class VisibilityAggregator (line 253) | class VisibilityAggregator:
    method __init__ (line 254) | def __init__(self, volume: SimpleVolume, additional_extent: float = 0.3):
    method integrate_into_volume (line 264) | def integrate_into_volume(

FILE: src/regsplatfacto/regsplatfacto/data/mvsanywhere_dataset.py
  class MVSAnywhereDataset (line 20) | class MVSAnywhereDataset(InputDataset):
    method __init__ (line 30) | def __init__(
    method _predict_all_depths_and_normals (line 56) | def _predict_all_depths_and_normals(self, depth_path: Path) -> tuple[n...
    method _get_depth_and_normal_estimate (line 185) | def _get_depth_and_normal_estimate(self, image_idx: int) -> dict[str, ...
    method get_data (line 227) | def get_data(self, image_idx: int, image_type: Literal["uint8", "float...

FILE: src/regsplatfacto/regsplatfacto/data/regsplatfacto_datamanager.py
  class RegSplatfactoDatamanagerConfig (line 13) | class RegSplatfactoDatamanagerConfig(FullImageDatamanagerConfig):
  class RegSplatfactoDatamanager (line 25) | class RegSplatfactoDatamanager(FullImageDatamanager):
    method __init__ (line 34) | def __init__(self, config: RegSplatfactoDatamanagerConfig, **kwargs):
    method _depth_model_to_dataset (line 38) | def _depth_model_to_dataset(self, depth_model: str):
    method dataset_type (line 45) | def dataset_type(self):
    method create_train_dataset (line 48) | def create_train_dataset(self):
    method create_eval_dataset (line 63) | def create_eval_dataset(self):

FILE: src/regsplatfacto/regsplatfacto/meshing.py
  function fuse_npz_folder (line 14) | def fuse_npz_folder(renders_path: Path, fuser: Open3DFuser, key_to_fuse_...
  function main (line 54) | def main(
  function entrypoint (line 129) | def entrypoint() -> None:

FILE: src/regsplatfacto/regsplatfacto/regsplatfacto_model.py
  function get_points_on_sphere (line 25) | def get_points_on_sphere(N: int, radius: float = 250.0) -> torch.Tensor:
  class RegSplatfactoModelConfig (line 49) | class RegSplatfactoModelConfig(SplatfactoModelConfig):
    method __post_init__ (line 101) | def __post_init__(self) -> None:
  class RegSplatfactoModel (line 127) | class RegSplatfactoModel(SplatfactoModel):
    method populate_modules (line 140) | def populate_modules(self):
    method get_skybox_param_groups (line 174) | def get_skybox_param_groups(self) -> Dict[str, List[Parameter]]:
    method get_param_groups (line 180) | def get_param_groups(self) -> Dict[str, List[Parameter]]:
    method skybox_means (line 193) | def skybox_means(self):
    method skybox_scales (line 197) | def skybox_scales(self):
    method skybox_quats (line 201) | def skybox_quats(self):
    method skybox_features_dc (line 205) | def skybox_features_dc(self):
    method skybox_opacities (line 209) | def skybox_opacities(self):
    method split_gaussians (line 212) | def split_gaussians(self, split_mask, samps) -> dict[str, torch.Tensor]:
    method get_rasterized_outputs (line 267) | def get_rasterized_outputs(self, camera: Cameras) -> dict[str, torch.T...
    method get_outputs (line 401) | def get_outputs(self, camera: Cameras) -> dict[str, torch.Tensor | lis...
    method get_loss_dict (line 434) | def get_loss_dict(
    method _get_normal_generator (line 526) | def _get_normal_generator(self, height: int, width: int) -> NormalGene...
    method _get_intrinsics (line 542) | def _get_intrinsics(self, camera: Cameras) -> torch.Tensor:
    method _get_implied_normal_from_depth (line 560) | def _get_implied_normal_from_depth(
    method compute_scale_regularisation_loss_median (line 590) | def compute_scale_regularisation_loss_median(self) -> torch.Tensor:
    method compute_flat_loss (line 628) | def compute_flat_loss(self) -> torch.Tensor:
    method compute_depth_loss (line 642) | def compute_depth_loss(self, pred_depth: torch.Tensor, gt_depth: torch...
    method compute_metric_depth_loss (line 659) | def compute_metric_depth_loss(
    method compute_scale_invariant_depth_loss (line 682) | def compute_scale_invariant_depth_loss(
    method compute_normal_loss (line 697) | def compute_normal_loss(
    method _get_gt_depth_and_normal (line 721) | def _get_gt_depth_and_normal(

FILE: src/regsplatfacto/regsplatfacto/render_for_meshing.py
  function compute_mapping_from_nerfstudio_to_world (line 57) | def compute_mapping_from_nerfstudio_to_world(
  class DatasetRenderMeshing (line 74) | class DatasetRenderMeshing(BaseRender):
    method main (line 94) | def main(self) -> None:
  function entrypoint (line 245) | def entrypoint():
  function get_parser_fn (line 255) | def get_parser_fn():

FILE: src/regsplatfacto/regsplatfacto/utils.py
  function to_homogeneous (line 18) | def to_homogeneous(input_tensor: torch.Tensor, dim: int = 0) -> torch.Te...
  class BackprojectDepth (line 28) | class BackprojectDepth(nn.Module):
    method __init__ (line 34) | def __init__(self, height: int, width: int) -> None:
    method forward (line 53) | def forward(self, depth_b1hw: torch.Tensor, invK_b44: torch.Tensor) ->...
  class Project3D (line 65) | class Project3D(nn.Module):
    method __init__ (line 70) | def __init__(self, eps: float = 1e-8):
    method forward (line 75) | def forward(
  class NormalGenerator (line 96) | class NormalGenerator(nn.Module):
    method __init__ (line 97) | def __init__(
    method forward (line 111) | def forward(self, depth_b1hw: torch.Tensor, invK_b44: torch.Tensor) ->...
  function force_to_hw3 (line 127) | def force_to_hw3(array: np.ndarray) -> np.ndarray: