SYMBOL INDEX (213 symbols across 30 files)

FILE: datasets/colmap_misc.py
  function is_computed (line 11) | def is_computed(sparse_dir):
  function read_colmap_pose (line 20) | def read_colmap_pose(image):
  function read_camera_params (line 30) | def read_camera_params(camera):
  function load_sparse_pcl_colmap (line 42) | def load_sparse_pcl_colmap(dir_recon):
  function get_sparse_depth (line 65) | def get_sparse_depth(T_w2c, img_size, crop_margin, sparse_pcl, frame_idx):

FILE: datasets/colmap_utils.py
  class Image (line 50) | class Image(BaseImage):
    method qvec2rotmat (line 51) | def qvec2rotmat(self):
  function read_next_bytes (line 74) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact...
  function write_next_bytes (line 86) | def write_next_bytes(fid, data, format_char_sequence, endian_character="...
  function read_cameras_text (line 102) | def read_cameras_text(path):
  function read_cameras_binary (line 128) | def read_cameras_binary(path_to_model_file):
  function write_cameras_text (line 157) | def write_cameras_text(cameras, path):
  function write_cameras_binary (line 174) | def write_cameras_binary(cameras, path_to_model_file):
  function read_images_text (line 194) | def read_images_text(path):
  function read_images_binary (line 225) | def read_images_binary(path_to_model_file):
  function write_images_text (line 260) | def write_images_text(images, path):
  function write_images_binary (line 288) | def write_images_binary(images, path_to_model_file):
  function read_points3D_text (line 309) | def read_points3D_text(path):
  function read_points3d_binary (line 336) | def read_points3d_binary(path_to_model_file):
  function write_points3D_text (line 366) | def write_points3D_text(points3D, path):
  function write_points3d_binary (line 391) | def write_points3d_binary(points3D, path_to_model_file):
  function detect_model_format (line 410) | def detect_model_format(path, ext):
  function read_model (line 419) | def read_model(path, ext=""):
  function write_model (line 441) | def write_model(cameras, images, points3D, path, ext=".bin"):
  function qvec2rotmat (line 453) | def qvec2rotmat(qvec):
  function rotmat2qvec (line 466) | def rotmat2qvec(R):
  function main (line 480) | def main():

FILE: datasets/data.py
  function process_projs (line 6) | def process_projs(proj):
  function pose_to_4x4 (line 17) | def pose_to_4x4(w2c):
  function data_to_c2w (line 24) | def data_to_c2w(w2c):
  function pil_loader (line 30) | def pil_loader(path):
  function get_sparse_depth (line 38) | def get_sparse_depth(pose_data, orig_size, sparse_pcl, frame_idx):

FILE: datasets/download_realestate10k.py
  class Data (line 17) | class Data:
    method __init__ (line 18) | def __init__(self, url, seqname, list_timestamps):
    method add (line 26) | def add(self, seqname, list_timestamps):
    method __len__ (line 30) | def __len__(self):
  function process (line 34) | def process(data, seq_id, videoname, output_root):
  function wrap_process (line 60) | def wrap_process(list_args):
  class DataDownloader (line 64) | class DataDownloader:
    method __init__ (line 65) | def __init__(self, data_path: Path, out_path: Path, tmp_path: Path, mo...
    method run (line 102) | def run(self):
    method show (line 143) | def show(self):
  function main (line 157) | def main():

FILE: datasets/kitti.py
  function readlines (line 23) | def readlines(filename):
  class KITTIDataset (line 30) | class KITTIDataset(data.Dataset):
    method __init__ (line 31) | def __init__(self,
    method __len__ (line 113) | def __len__(self):
    method get_color (line 116) | def get_color(self, folder, frame_index, side, do_flip):
    method get_depth_anything (line 125) | def get_depth_anything(self, folder, frame_index, side, do_flip):
    method get_image_path (line 137) | def get_image_path(self, folder, frame_index, side):
    method preprocess (line 143) | def preprocess(self, inputs, color_aug):
    method _get_sequences (line 168) | def _get_sequences(data_path):
    method _load_poses (line 184) | def _load_poses(pose_path, sequences):
    method _load_calibs (line 212) | def _load_calibs(data_path, target_image_size, keep_aspect_ratio):
    method __getitem__ (line 326) | def __getitem__(self, index):

FILE: datasets/nyu/camera.py
  function camera_params (line 4) | def camera_params():
  function make_K (line 21) | def make_K(fx, fy, cx, cy):

FILE: datasets/nyu/compute_colmap.py
  function get_sub_dirs (line 20) | def get_sub_dirs(dir_colmap):
  function get_sub_dirs (line 27) | def get_sub_dirs(dir_colmap):
  function get_nyu_test_sequences (line 34) | def get_nyu_test_sequences():
  function run_colmap (line 42) | def run_colmap(dir_colmap, cuda_device):
  function add_camera (line 115) | def add_camera(db):
  function compute_sfm (line 139) | def compute_sfm(seq_name):
  function main (line 191) | def main():

FILE: datasets/nyu/dataset.py
  class NYUv2Dataset (line 23) | class NYUv2Dataset(data.Dataset):
    method __init__ (line 24) | def __init__(self,
    method _load_split_indices (line 73) | def _load_split_indices(index_path):
    method load_image (line 89) | def load_image(self, seq_key, filename, color_aug_fn):
    method get_colmap_dir (line 135) | def get_colmap_dir(self, seq_key):
    method get_image_files (line 138) | def get_image_files(self, seq_key):
    method get_full_sequence (line 144) | def get_full_sequence(self, seq_key, src_idx):
    method get_inputs_and_targets (line 150) | def get_inputs_and_targets(self, seq_key, src_and_tgt_frame_idxs, fram...
    method __getitem__ (line 219) | def __getitem__(self, index):
    method __len__ (line 229) | def __len__(self) -> int:

FILE: datasets/preprocess_realestate10k.py
  function main (line 10) | def main():

FILE: datasets/re10k.py
  function load_seq_data (line 21) | def load_seq_data(data_path, split):
  class Re10KDataset (line 28) | class Re10KDataset(data.Dataset):
    method __init__ (line 29) | def __init__(self,
    method __len__ (line 131) | def __len__(self) -> int:
    method _load_seq_data (line 134) | def _load_seq_data(self, split):
    method _full_index (line 137) | def _full_index(self, seq_keys, seq_data, left_offset, extra_frames):
    method _load_sparse_pcl (line 153) | def _load_sparse_pcl(self, seq_key):
    method _load_image (line 165) | def _load_image(self, key, id):
    method _load_depth (line 176) | def _load_depth(self, key, id):
    method _load_split_indices (line 193) | def _load_split_indices(index_path):
    method get_frame_data (line 210) | def get_frame_data(self, seq_key, frame_idx, pose_data, color_aug_fn):
    method __getitem__ (line 257) | def __getitem__(self, index):

FILE: datasets/tardataset.py
  class TarDataset (line 15) | class TarDataset(Dataset):
    method __init__ (line 39) | def __init__(self, archive, transform=to_tensor, extensions=('.png', '...
    method filter_samples (line 67) | def filter_samples(self, is_valid_file=None, extensions=('.png', '.jpg...
    method __getitem__ (line 87) | def __getitem__(self, index):
    method __len__ (line 110) | def __len__(self):
    method get_image (line 119) | def get_image(self, name, pil=False):
    method get_text_file (line 135) | def get_text_file(self, name, encoding='utf-8'):
    method get_file (line 148) | def get_file(self, name):
    method __del__ (line 167) | def __del__(self):
    method __getstate__ (line 173) | def __getstate__(self):

FILE: datasets/util.py
  function create_datasets (line 10) | def create_datasets(cfg, split="val"):
  function custom_collate (line 40) | def custom_collate(batch):

FILE: evaluate.py
  function get_model_instance (line 21) | def get_model_instance(model):
  function evaluate (line 27) | def evaluate(model, cfg, evaluator, dataloader, device=None, save_vis=Fa...
  function main (line 134) | def main(cfg: DictConfig):

FILE: evaluation/evaluator.py
  class Evaluator (line 10) | class Evaluator(nn.Module):
    method __init__ (line 11) | def __init__(self, crop_border=True):
    method norm (line 28) | def norm(img):
    method metric_names (line 31) | def metric_names(self):
    method forward (line 34) | def forward(self, img_pred, img_gt):

FILE: misc/depth.py
  function estimate_depth_scale_kitti (line 10) | def estimate_depth_scale_kitti(depth, depth_gt):
  function estimate_depth_scale (line 26) | def estimate_depth_scale(depth, sparse_depth):
  function estimate_depth_scale_ransac (line 53) | def estimate_depth_scale_ransac(depth, sparse_depth, num_iterations=1000...
  function gray2rgb (line 99) | def gray2rgb(im, cmap=CMAP_DEFAULT):
  function normalize_depth_for_display (line 107) | def normalize_depth_for_display(depth, pc=95, crop_percent=0, normalizer...

FILE: misc/download_pretrained_models.py
  function main (line 6) | def main():

FILE: misc/localstorage.py
  function get_local_dir (line 9) | def get_local_dir():
  function local_storage_path (line 18) | def local_storage_path(filename):
  function copy_to_local_storage (line 21) | def copy_to_local_storage(filename, rank=None):
  function extract_tar (line 35) | def extract_tar(fn, unzip_dir):

FILE: misc/logger.py
  function get_all_keys (line 12) | def get_all_keys(d):
  function setup_logging_dir (line 19) | def setup_logging_dir():
  class NeptuneLogger (line 29) | class NeptuneLogger:
    method __init__ (line 30) | def __init__(self, cfg):
    method _setup (line 49) | def _setup(cfg):
    method log (line 68) | def log(self, values, step):
    method log3d (line 72) | def log3d(self, kv, step):
    method upload_file (line 75) | def upload_file(self, key, filename):
    method upload_image (line 78) | def upload_image(self, key, image):
    method log_image (line 81) | def log_image(self, key, image, step=0):
  function setup_logger (line 85) | def setup_logger(cfg):

FILE: misc/util.py
  function get_source_frame_ids (line 3) | def get_source_frame_ids():
  function add_source_frame_id (line 6) | def add_source_frame_id(novel_frames):
  function sec_to_hm (line 9) | def sec_to_hm(t):
  function sec_to_hm_str (line 20) | def sec_to_hm_str(t):

FILE: misc/visualise_3d.py
  function depth_to_img (line 16) | def depth_to_img(d):
  function vis_2d_offsets (line 22) | def vis_2d_offsets(model, inputs, outputs, out_dir, frame_id):
  function construct_list_of_attributes (line 80) | def construct_list_of_attributes(num_rest: int) -> list[str]:
  function export_ply (line 94) | def export_ply(
  function save_ply (line 162) | def save_ply(outputs, path, gaussians_per_pixel=3):

FILE: models/decoder/gauss_util.py
  function getProjectionMatrix (line 18) | def getProjectionMatrix(znear, zfar, fovX, fovY, pX=0.0, pY=0.0):
  function fov2focal (line 46) | def fov2focal(fov, pixels):
  function focal2fov (line 50) | def focal2fov(focal, pixels):
  function K_to_NDC_pp (line 53) | def K_to_NDC_pp(Kx, Ky, H, W):
  function render_predicted (line 63) | def render_predicted(cfg,

FILE: models/decoder/gaussian_decoder.py
  function get_splits_and_inits (line 6) | def get_splits_and_inits(cfg):
  class GaussianDecoder (line 39) | class GaussianDecoder(nn.Module):
    method __init__ (line 41) | def __init__(self, cfg):
    method forward (line 51) | def forward(self, x, split_dimensions=[3,1,3,4,3,9]):

FILE: models/decoder/resnet_decoder.py
  class ResnetDecoder (line 10) | class ResnetDecoder(nn.Module):
    method __init__ (line 12) | def __init__(self, cfg, num_ch_enc, use_skips=True):
    method forward (line 54) | def forward(self, input_features):
  class ResnetDepthDecoder (line 70) | class ResnetDepthDecoder(nn.Module):
    method __init__ (line 72) | def __init__(self, cfg, num_ch_enc, use_skips=True):
    method forward (line 114) | def forward(self, input_features):

FILE: models/encoder/layers.py
  function disp_to_depth (line 6) | def disp_to_depth(disp, min_depth, max_depth):
  function upsample (line 18) | def upsample(x, mode="nearest"):
  class Conv3x3 (line 24) | class Conv3x3(nn.Module):
    method __init__ (line 27) | def __init__(self, in_channels, out_channels, use_refl=True):
    method forward (line 36) | def forward(self, x):
  class ConvBlock (line 42) | class ConvBlock(nn.Module):
    method __init__ (line 45) | def __init__(self, in_channels, out_channels):
    method forward (line 51) | def forward(self, x):
  class BackprojectDepth (line 57) | class BackprojectDepth(nn.Module):
    method __init__ (line 60) | def __init__(self, batch_size, height, width, shift_rays_half_pixel=0):
    method forward (line 84) | def forward(self, depth, inv_K):
  class Project3D (line 92) | class Project3D(nn.Module):
    method __init__ (line 95) | def __init__(self, batch_size, height, width, eps=1e-7):
    method forward (line 103) | def forward(self, points, K, T=None):
  class Project3DSimple (line 121) | class Project3DSimple(nn.Module):
    method __init__ (line 124) | def __init__(self, batch_size, height, width, eps=1e-7):
    method forward (line 132) | def forward(self, points, K):
  class SSIM (line 143) | class SSIM(nn.Module):
    method __init__ (line 146) | def __init__(self):
    method forward (line 159) | def forward(self, x, y):

FILE: models/encoder/resnet_encoder.py
  class ResNetMultiImageInput (line 18) | class ResNetMultiImageInput(models.ResNet):
    method __init__ (line 22) | def __init__(self, block, layers, num_classes=1000, num_input_images=1):
  function resnet_multiimage_input (line 43) | def resnet_multiimage_input(num_layers, pretrained=False, num_input_imag...
  class ResnetEncoder (line 64) | class ResnetEncoder(nn.Module):
    method __init__ (line 67) | def __init__(self, num_layers, pretrained, bn_order, num_input_images=1):
    method forward (line 85) | def forward(self, input_image):

FILE: models/encoder/unidepth_encoder.py
  class UniDepthExtended (line 9) | class UniDepthExtended(nn.Module):
    method __init__ (line 10) | def __init__(self, cfg):
    method get_parameter_groups (line 48) | def get_parameter_groups(self):
    method forward (line 52) | def forward(self, inputs):

FILE: models/model.py
  function default_param_group (line 14) | def default_param_group(model):
  function to_device (line 18) | def to_device(inputs, device):
  class GaussianPredictor (line 25) | class GaussianPredictor(nn.Module):
    method __init__ (line 26) | def __init__(self, cfg):
    method set_backproject (line 45) | def set_backproject(self):
    method target_frame_ids (line 70) | def target_frame_ids(self, inputs):
    method all_frame_ids (line 73) | def all_frame_ids(self, inputs):
    method set_train (line 76) | def set_train(self):
    method set_eval (line 83) | def set_eval(self):
    method is_train (line 90) | def is_train(self):
    method forward (line 93) | def forward(self, inputs):
    method compute_gauss_means (line 106) | def compute_gauss_means(self, inputs, outputs):
    method process_gt_poses (line 130) | def process_gt_poses(self, inputs, outputs):
    method render_images (line 189) | def render_images(self, inputs, outputs):
    method checkpoint_dir (line 281) | def checkpoint_dir(self):
    method save_model (line 284) | def save_model(self, optimiser, step, ema=None):
    method load_model (line 307) | def load_model(self, weights_path, optimiser=None, device="cpu", ckpt_...

FILE: splits/re10k_pixelsplat/preprocess_2_frames_split.py
  function transform_json_to_txt (line 4) | def transform_json_to_txt(json_data, seq_data, src_idx=0):
  function main (line 38) | def main():

FILE: train.py
  function run_epoch (line 21) | def run_epoch(fabric,
  function main (line 83) | def main(cfg: DictConfig):

FILE: trainer.py
  class Trainer (line 17) | class Trainer(nn.Module):
    method __init__ (line 18) | def __init__(self, cfg):
    method set_logger (line 30) | def set_logger(self, logger):
    method forward (line 33) | def forward(self, inputs):
    method compute_reconstruction_loss (line 38) | def compute_reconstruction_loss(self, pred, target, losses):
    method compute_losses (line 66) | def compute_losses(self, inputs, outputs):
    method log_time (line 113) | def log_time(self, batch_idx, duration, loss):
    method log_scalars (line 125) | def log_scalars(self, mode, outputs, losses, lr):
    method log (line 145) | def log(self, mode, inputs, outputs):
    method validate (line 203) | def validate(self, model, evaluator, val_loader, device):