SYMBOL INDEX (370 symbols across 37 files) FILE: data_scripts/7scenes_preprocessing.py function mkdir (line 34) | def mkdir(directory): function process_scene (line 62) | def process_scene(ds): FILE: data_scripts/generate_test_tuples.py function compute_offline_tuple (line 65) | 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 214) | def offline_dvmvs_tuples(scan, poses, n_measurement_frames): function dense_dvmvs_tuples (line 268) | def dense_dvmvs_tuples(scan, poses, n_measurement_frames): function offline_dense_dvmvs_tuples (line 345) | def offline_dense_dvmvs_tuples(scan, poses, n_measurement_frames): function crawl_subprocess_long (line 384) | def crawl_subprocess_long(opts_temp_filepath, scan, count, progress): function crawl (line 530) | def crawl(opts_temp_filepath, opts, scans): FILE: data_scripts/generate_train_tuples.py function gather_pairs_train (line 55) | def gather_pairs_train(poses, used_pairs, function crawl_subprocess_short (line 140) | def crawl_subprocess_short(opts_temp_filepath, scan, count, progress): function crawl_subprocess_long (line 221) | def crawl_subprocess_long(opts_temp_filepath, scan, count, progress): function crawl (line 378) | def crawl(opts_temp_filepath, opts, scans): FILE: data_scripts/precompute_valid_frames.py function process_scan (line 45) | def process_scan(opts_temp_filepath, scan, count, progress): function multi_process_scans (line 92) | def multi_process_scans(opts_temp_filepath, opts, scans): FILE: data_scripts/scannet_wrangling_scripts/SensorData.py function print_array_on_one_line (line 16) | def print_array_on_one_line(): class RGBDFrame (line 23) | class RGBDFrame(): method load (line 25) | def load(self, file_handle): method decompress_depth (line 35) | def decompress_depth(self, compression_type): method decompress_depth_zlib (line 42) | def decompress_depth_zlib(self): method decompress_color (line 46) | def decompress_color(self, compression_type): method dump_color_to_file (line 52) | def dump_color_to_file(self, compression_type, filepath): method decompress_color_jpeg (line 61) | def decompress_color_jpeg(self): class SensorData (line 65) | class SensorData: method __init__ (line 67) | def __init__(self, filename): method load (line 72) | def load(self, filename): method export_depth_images (line 98) | def export_depth_images(self, output_path, image_size=None, frame_skip... method export_color_images (line 117) | def export_color_images(self, output_path, image_size=None, frame_skip... method save_mat_to_file (line 133) | def save_mat_to_file(self, matrix, filename): method export_poses (line 139) | def export_poses(self, output_path, frame_skip=1): method export_intrinsics (line 147) | def export_intrinsics(self, output_path, scan_name): FILE: data_scripts/scannet_wrangling_scripts/download_scannet.py function get_release_scans (line 33) | def get_release_scans(release_file): function download_release (line 42) | def download_release(release_scans, out_dir, file_types, use_v1_sens): function download_file (line 52) | def download_file(url, out_file): function download_scan (line 66) | def download_scan(scan_id, out_dir, file_types, use_v1_sens): function download_task_data (line 78) | def download_task_data(out_dir): function download_tfrecords (line 94) | def download_tfrecords(in_dir, out_dir): function download_label_map (line 112) | def download_label_map(out_dir): function main (line 125) | def main(): FILE: data_scripts/scannet_wrangling_scripts/reader.py function process_scan (line 30) | def process_scan(opt, scan_job, count=None, progress=None): function main (line 54) | def main(): FILE: experiment_modules/depth_model.py class DepthModel (line 21) | class DepthModel(pl.LightningModule): method __init__ (line 68) | def __init__(self, opts): method compute_matching_feats (line 191) | def compute_matching_feats( method forward (line 247) | def forward( method compute_losses (line 409) | def compute_losses(self, cur_data, src_data, outputs): method step (line 502) | def step(self, phase, batch, batch_idx): method training_step (line 607) | def training_step(self, batch, batch_idx): method validation_step (line 611) | def validation_step(self, batch, batch_idx): method configure_optimizers (line 615) | def configure_optimizers(self): FILE: 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 57) | class NormalsLoss(nn.Module): method forward (line 58) | def forward(self, normals_gt_b3hw: Tensor, normals_pred_b3hw: Tensor) ... class MVDepthLoss (line 79) | class MVDepthLoss(nn.Module): method __init__ (line 80) | def __init__(self, height, width): method get_valid_mask (line 90) | def get_valid_mask( method get_error_for_pair (line 136) | def get_error_for_pair(self, method forward (line 175) | def forward( FILE: modules/cost_volume.py class CostVolumeManager (line 13) | class CostVolumeManager(nn.Module): method __init__ (line 27) | def __init__( method initialise_for_projection (line 58) | def initialise_for_projection(self): method get_mask (line 77) | def get_mask(self, pix_coords_bk2hw): method generate_depth_planes (line 100) | def generate_depth_planes(self, batch_size: int, method warp_features (line 139) | def warp_features( method build_cost_volume (line 237) | def build_cost_volume( method indices_to_disparity (line 338) | def indices_to_disparity(self, indices, depth_planes_bdhw): method forward (line 345) | def forward( class FeatureVolumeManager (line 383) | class FeatureVolumeManager(CostVolumeManager): method __init__ (line 398) | def __init__(self, method build_cost_volume (line 451) | def build_cost_volume(self, method to_fast (line 739) | def to_fast(self) -> 'FastFeatureVolumeManager': class FastFeatureVolumeManager (line 749) | class FastFeatureVolumeManager(FeatureVolumeManager): method __init__ (line 759) | def __init__(self, method warp_features (line 812) | def warp_features( method build_cost_volume (line 967) | def build_cost_volume( FILE: modules/layers.py function conv3x3 (line 7) | def conv3x3( function conv1x1 (line 20) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1, bias: bool... class BasicBlock (line 24) | class BasicBlock(nn.Module): method __init__ (line 27) | def __init__( method forward (line 68) | def forward(self, x: Tensor) -> Tensor: class TensorFormatter (line 87) | class TensorFormatter(nn.Module): method __init__ (line 95) | def __init__(self): method _expand_batch_with_channels (line 101) | def _expand_batch_with_channels(self, x): method _reduce_batch_to_channels (line 109) | def _reduce_batch_to_channels(self, x): method forward (line 117) | def forward(self, x, apply_func): FILE: 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__( method forward (line 75) | def forward(self, input_features): class CVEncoder (line 99) | class CVEncoder(nn.Module): method __init__ (line 100) | def __init__(self, num_ch_cv, num_ch_enc, num_ch_outs): method forward (line 120) | def forward(self, x, img_feats): class MLP (line 129) | class MLP(nn.Module): method __init__ (line 130) | def __init__(self, channel_list, disable_final_activation = False): method forward (line 146) | def forward(self, x): class ResnetMatchingEncoder (line 149) | class ResnetMatchingEncoder(nn.Module): method __init__ (line 152) | def __init__( method forward (line 204) | def forward(self, input_image): class UNetMatchingEncoder (line 207) | class UNetMatchingEncoder(nn.Module): method __init__ (line 208) | def __init__(self): method forward (line 226) | def forward(self, x): FILE: options.py class Options (line 10) | class Options(): class OptionsHandler (line 220) | class OptionsHandler(): method __init__ (line 256) | def __init__(self, required_flags=[]): method parse_and_merge_options (line 271) | def parse_and_merge_options(self, config_filepaths=None, ignore_cl_arg... method populate_argparse (line 336) | def populate_argparse(self): method check_required_items (line 350) | def check_required_items(self): method merge_config_options (line 356) | def merge_config_options(self, config_options): method merge_cl_args (line 364) | def merge_cl_args(self, cl_args): method pretty_print_options (line 384) | def pretty_print_options(self): method load_options_from_yaml (line 393) | def load_options_from_yaml(config_filepath): method save_options_as_yaml (line 398) | def save_options_as_yaml(config_filepath, options): function handle_backwards_compat (line 402) | def handle_backwards_compat(opts): FILE: pc_fusion.py function main (line 34) | def main(opts): FILE: rerun_visualize_live_meshing.py function to_device (line 32) | def to_device(input_dict, key_ignores=[], device="cuda"): function log_source_data (line 42) | def log_source_data(src_entity_path: str, src_data: Dict[str, Any]) -> N... function log_camera (line 55) | def log_camera( function log_image (line 76) | def log_image( function log_rerun (line 92) | def log_rerun( function main (line 168) | def main(opts): FILE: sr_datasets/arkit_dataset.py class ARKitDataset (line 16) | class ARKitDataset(GenericMVSDataset): method __init__ (line 28) | def __init__(self, method get_sub_folder_dir (line 125) | def get_sub_folder_dir(split): method get_frame_id_string (line 128) | def get_frame_id_string(self, frame_id): method get_valid_frame_path (line 137) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 145) | def get_valid_frame_ids(self, split, scan, store_computed=True): method get_color_filepath (line 240) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 265) | def get_high_res_color_filepath(self, scan_id, frame_id): method get_cached_depth_filepath (line 290) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 299) | def get_full_res_depth_filepath(self, scan_id, frame_id): method get_pose_filepath (line 307) | def get_pose_filepath(self, scan_id, frame_id): method load_target_size_depth_and_mask (line 321) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method load_full_res_depth_and_mask (line 335) | def load_full_res_depth_and_mask(self, scan_id, frame_id): method load_pose (line 350) | def load_pose(self, scan_id, frame_id): method load_intrinsics (line 372) | def load_intrinsics(self, scan_id, frame_id, flip=None): function process_data (line 421) | def process_data(data_path, data_source='ARKit', function path_parser (line 467) | def path_parser(root_path, data_source='TagBA'): function load_camera_pose (line 484) | def load_camera_pose(cam_pose_dir, use_homogenous=True, data_source='Tag... function load_camera_intrinsic (line 536) | def load_camera_intrinsic(cam_file, data_source='TagBA'): function extract_frames (line 597) | def extract_frames(video_path, out_folder, size): function sync_intrinsics_and_poses (line 608) | def sync_intrinsics_and_poses(cam_file, pose_file, out_file): FILE: sr_datasets/colmap_dataset.py class ColmapDataset (line 15) | class ColmapDataset(GenericMVSDataset): method __init__ (line 48) | def __init__( method get_sub_folder_dir (line 142) | def get_sub_folder_dir(split): method get_frame_id_string (line 145) | def get_frame_id_string(self, frame_id): method get_valid_frame_path (line 154) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 163) | def get_valid_frame_ids(self, split, scan, store_computed=True): method load_pose (line 233) | def load_pose(self, scan_id, frame_id): method load_intrinsics (line 268) | def load_intrinsics(self, scan_id, frame_id=None, flip=None): method load_capture_poses (line 398) | def load_capture_poses(self, scan_id): method load_target_size_depth_and_mask (line 448) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method load_full_res_depth_and_mask (line 462) | def load_full_res_depth_and_mask(self, scan_id, frame_id): method get_cached_depth_filepath (line 477) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 487) | def get_full_res_depth_filepath(self, scan_id, frame_id): method get_color_filepath (line 495) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 522) | def get_high_res_color_filepath(self, scan_id, frame_id): method load_color (line 543) | def load_color(self, scan_id, frame_id): method load_high_res_color (line 569) | def load_high_res_color(self, scan_id, frame_id): FILE: sr_datasets/generic_mvs_dataset.py class GenericMVSDataset (line 15) | class GenericMVSDataset(Dataset): method __init__ (line 43) | def __init__(self, method __len__ (line 197) | def __len__(self): method get_sub_folder_dir (line 201) | def get_sub_folder_dir(split): method get_valid_frame_path (line 205) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 211) | def get_valid_frame_ids(self, split, scan, store_computed=True): method get_color_filepath (line 233) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 249) | def get_high_res_color_filepath(self, scan_id, frame_id): method get_cached_depth_filepath (line 266) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 281) | def get_full_res_depth_filepath(self, scan_id, frame_id): method get_pose_filepath (line 297) | def get_pose_filepath(self, scan_id, frame_id): method get_frame_id_string (line 310) | def get_frame_id_string(self, frame_id): method get_gt_mesh_path (line 319) | def get_gt_mesh_path(dataset_path, split, scan_id): method load_intrinsics (line 325) | def load_intrinsics(self, scan_id, frame_id=None, flip=None): method load_target_size_depth_and_mask (line 347) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method load_full_res_depth_and_mask (line 368) | def load_full_res_depth_and_mask(self, scan_id, frame_id): method load_pose (line 386) | def load_pose(self, scan_id, frame_id): method load_color (line 402) | def load_color(self, scan_id, frame_id): method load_high_res_color (line 425) | def load_high_res_color(self, scan_id, frame_id): method get_frame (line 451) | def get_frame(self, scan_id, frame_id, load_depth, flip=False): method stack_src_data (line 587) | def stack_src_data(self, src_data): method __getitem__ (line 602) | def __getitem__(self, idx): FILE: sr_datasets/scannet_dataset.py class ScannetDataset (line 11) | class ScannetDataset(GenericMVSDataset): method __init__ (line 77) | def __init__( method get_sub_folder_dir (line 165) | def get_sub_folder_dir(split): method get_frame_id_string (line 172) | def get_frame_id_string(self, frame_id): method get_valid_frame_path (line 181) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 190) | def get_valid_frame_ids(self, split, scan, store_computed=True): method get_gt_mesh_path (line 292) | def get_gt_mesh_path(dataset_path, split, scan_id): method get_color_filepath (line 304) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 330) | def get_high_res_color_filepath(self, scan_id, frame_id): method get_cached_depth_filepath (line 357) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 379) | def get_full_res_depth_filepath(self, scan_id, frame_id): method get_pose_filepath (line 399) | def get_pose_filepath(self, scan_id, frame_id): method load_intrinsics (line 416) | def load_intrinsics(self, scan_id, frame_id=None, flip=False): method load_target_size_depth_and_mask (line 474) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method load_full_res_depth_and_mask (line 517) | def load_full_res_depth_and_mask(self, scan_id, frame_id): method load_pose (line 549) | def load_pose(self, scan_id, frame_id): FILE: sr_datasets/scanniverse_dataset.py class ScanniverseDataset (line 14) | class ScanniverseDataset(GenericMVSDataset): method __init__ (line 27) | def __init__( method get_sub_folder_dir (line 128) | def get_sub_folder_dir(split): method load_capture_metadata (line 131) | def load_capture_metadata(self, scan_id): method get_frame_id_string (line 239) | def get_frame_id_string(self, frame_id): method get_valid_frame_path (line 248) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 256) | def get_valid_frame_ids(self, split, scan, store_computed=True): method get_color_filepath (line 331) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 369) | def get_high_res_color_filepath(self, scan_id, frame_id): method get_cached_depth_filepath (line 387) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 397) | def get_full_res_depth_filepath(self, scan_id, frame_id): method load_pose (line 405) | def load_pose(self, scan_id, frame_id): method load_intrinsics (line 447) | def load_intrinsics(self, scan_id, frame_id, flip=None): method load_target_size_depth_and_mask (line 514) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method load_full_res_depth_and_mask (line 528) | def load_full_res_depth_and_mask(self, scan_id, frame_id): FILE: sr_datasets/seven_scenes_dataset.py class SevenScenesDataset (line 12) | class SevenScenesDataset(GenericMVSDataset): method __init__ (line 47) | def __init__( method get_sub_folder_dir (line 135) | def get_sub_folder_dir(split): method get_frame_id_string (line 139) | def get_frame_id_string(self, frame_id): method get_valid_frame_path (line 148) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 157) | def get_valid_frame_ids(self, split, scan, store_computed=True): method get_color_filepath (line 254) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 281) | def get_high_res_color_filepath(self, scan_id, frame_id): method get_cached_depth_filepath (line 308) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 328) | def get_full_res_depth_filepath(self, scan_id, frame_id): method get_pose_filepath (line 347) | def get_pose_filepath(self, scan_id, frame_id): method load_intrinsics (line 362) | def load_intrinsics(self, scan_id=None, frame_id=None, flip=None): method load_target_size_depth_and_mask (line 410) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method load_full_res_depth_and_mask (line 452) | def load_full_res_depth_and_mask(self, scan_id, frame_id): method load_pose (line 485) | def load_pose(self, scan_id, frame_id): FILE: sr_datasets/vdr_dataset.py class VDRDataset (line 17) | class VDRDataset(GenericMVSDataset): method __init__ (line 29) | def __init__( method get_sub_folder_dir (line 78) | def get_sub_folder_dir(split): method get_frame_id_string (line 81) | def get_frame_id_string(self, frame_id): method get_valid_frame_path (line 90) | def get_valid_frame_path(self, split, scan): method get_valid_frame_ids (line 98) | def get_valid_frame_ids(self, split, scan, store_computed=True): method load_pose (line 168) | def load_pose(self, scan_id, frame_id): method load_intrinsics (line 207) | def load_intrinsics(self, scan_id, frame_id, flip=None): method load_capture_metadata (line 266) | def load_capture_metadata(self, scan_id): method get_cached_depth_filepath (line 294) | def get_cached_depth_filepath(self, scan_id, frame_id): method get_cached_confidence_filepath (line 327) | def get_cached_confidence_filepath(self, scan_id, frame_id): method get_full_res_depth_filepath (line 361) | def get_full_res_depth_filepath(self, scan_id, frame_id): method get_full_res_confidence_filepath (line 383) | def get_full_res_confidence_filepath(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_target_size_depth_and_mask (line 444) | def load_target_size_depth_and_mask(self, scan_id, frame_id): method get_color_filepath (line 513) | def get_color_filepath(self, scan_id, frame_id): method get_high_res_color_filepath (line 541) | def get_high_res_color_filepath(self, scan_id, frame_id): FILE: test.py function main (line 128) | def main(opts): FILE: tools/fusers_helper.py class DepthFuser (line 11) | class DepthFuser(): method __init__ (line 12) | def __init__( class OurFuser (line 22) | class OurFuser(DepthFuser): method __init__ (line 34) | def __init__( method fuse_frames (line 64) | def fuse_frames(self, depths_b1hw, K_b44, method export_mesh (line 73) | def export_mesh(self, path, export_single_mesh=True): method get_mesh (line 80) | def get_mesh(self, export_single_mesh=True, convert_to_trimesh=True): class Open3DFuser (line 84) | class Open3DFuser(DepthFuser): method __init__ (line 90) | def __init__( method fuse_frames (line 116) | def fuse_frames( method export_mesh (line 177) | def export_mesh(self, path, use_marching_cubes_mask=None): method get_mesh (line 180) | def get_mesh(self, export_single_mesh=None, convert_to_trimesh=False): function get_fuser (line 188) | def get_fuser(opts, scan): FILE: tools/keyframe_buffer.py class DVMVS_Config (line 12) | class DVMVS_Config: function is_pose_available (line 24) | def is_pose_available(pose): function is_valid_pair (line 33) | def is_valid_pair( function pose_distance (line 54) | def pose_distance(reference_pose, measurement_pose): class KeyframeBuffer (line 72) | class KeyframeBuffer: method __init__ (line 73) | def __init__( method calculate_penalty (line 89) | def calculate_penalty(self, t_score, R_score): method try_new_keyframe (line 99) | def try_new_keyframe(self, pose, image, dist_to_last_valid=None, index... method get_best_measurement_frames (line 163) | def get_best_measurement_frames(self, n_requested_measurement_frames): class SimpleBuffer (line 189) | class SimpleBuffer: method __init__ (line 190) | def __init__( method try_new_keyframe (line 200) | def try_new_keyframe(self, pose, image, index=None): method get_measurement_frames (line 241) | def get_measurement_frames(self): class OfflineKeyframeBuffer (line 245) | class OfflineKeyframeBuffer: method __init__ (line 246) | def __init__( method calculate_penalty (line 263) | def calculate_penalty(self, t_score, R_score): method try_new_keyframe (line 273) | def try_new_keyframe(self, pose, image, index=None): method get_best_measurement_frames (line 332) | def get_best_measurement_frames(self, n_requested_measurement_frames): method get_best_measurement_frames_for_0index (line 357) | def get_best_measurement_frames_for_0index(self, n_requested_measureme... FILE: tools/mesh_renderer.py class Renderer (line 26) | class Renderer(): method __init__ (line 32) | def __init__(self, height=480, width=640, flat_render=False): method render (line 37) | def render(self, height, width, method __call__ (line 66) | def __call__(self, height, width, intrinsics, pose, meshes): method fix_pose (line 69) | def fix_pose(self, pose): method mesh_opengl (line 81) | def mesh_opengl(self, mesh, mesh_material=None): method delete (line 87) | def delete(self): method render_mesh (line 90) | def render_mesh( method render_mesh_cull_composite (line 134) | def render_mesh_cull_composite(self, alpha, **kwargs): function render_colour (line 143) | def render_colour(renderer, meshes, world_T_cam, K, height=256, width=320): class SmoothBirdsEyeCamera (line 155) | class SmoothBirdsEyeCamera(): method __init__ (line 160) | def __init__( method get_bird_eye_trans (line 178) | def get_bird_eye_trans(self, function camera_marker (line 265) | def camera_marker( function get_image_box (line 428) | def get_image_box( function create_light_array (line 496) | def create_light_array(light_type, center_loc, function transform_trimesh (line 523) | def transform_trimesh(mesh, transform): FILE: tools/torch_point_cloud_fusion.py function process_depth (line 12) | def process_depth(ref_depth, ref_image, src_depths, src_images, ref_P, s... function process_scene (line 100) | def process_scene(depth_preds, images, poses, K, z_thresh, n_consistent_... FILE: tools/tsdf.py class TSDF (line 11) | class TSDF: method __init__ (line 19) | def __init__( method from_file (line 37) | def from_file(cls, tsdf_file): method from_mesh (line 52) | def from_mesh(cls, mesh: trimesh.Trimesh, voxel_size: float): method from_bounds (line 70) | def from_bounds(cls, bounds: dict, voxel_size: float): method generate_voxel_coords (line 98) | def generate_voxel_coords(cls, method cuda (line 111) | def cuda(self): method cpu (line 118) | def cpu(self): method to_mesh (line 125) | def to_mesh(self, scale_to_world=True, export_single_mesh=False): method save (line 159) | def save(self, savepath, filename, save_mesh=True): class TSDFFuser (line 171) | class TSDFFuser: method __init__ (line 175) | def __init__(self, tsdf, min_depth=0.5, max_depth=5.0, use_gpu=True): method voxel_coords (line 198) | def voxel_coords(self): method tsdf_values (line 202) | def tsdf_values(self): method tsdf_weights (line 206) | def tsdf_weights(self): method voxel_size (line 210) | def voxel_size(self): method shape (line 214) | def shape(self): method truncation (line 218) | def truncation(self): method project_to_camera (line 221) | def project_to_camera(self, cam_T_world_T_b44, K_b44): method integrate_depth (line 238) | def integrate_depth( FILE: train.py function main (line 34) | def main(opts): FILE: utils/dataset_utils.py function get_dataset (line 8) | def get_dataset(dataset_name, FILE: utils/generic_utils.py function copy_code_state (line 15) | def copy_code_state(path): function readlines (line 36) | def readlines(filepath): function normalize_depth_single (line 42) | def normalize_depth_single(depth_11hw, mask_11hw, robust=False): function normalize_depth (line 73) | def normalize_depth(depth_b1hw: torch.Tensor, function pyrdown (line 88) | def pyrdown(input_tensor: torch.Tensor, num_scales: int = 4): function upsample (line 96) | def upsample(x): function batched_trace (line 107) | def batched_trace(mat_bNN): function tensor_B_to_bM (line 110) | def tensor_B_to_bM(tensor_BS, batch_size, num_views): function tensor_bM_to_B (line 121) | def tensor_bM_to_B(tensor_bMS): function combine_dims (line 132) | def combine_dims(x, dim_begin, dim_end): function to_gpu (line 138) | def to_gpu(input_dict, key_ignores=[]): function imagenet_normalize (line 147) | def imagenet_normalize(image): function reverse_imagenet_normalize (line 153) | def reverse_imagenet_normalize(image): function read_image_file (line 162) | def read_image_file(filepath, function crop_image_to_target_ratio (line 210) | def crop_image_to_target_ratio(image, target_aspect_ratio=4.0/3.0): function cache_model_outputs (line 241) | def cache_model_outputs( FILE: 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 51) | def forward(self, depth_b1hw: Tensor, invK_b44: Tensor) -> Tensor: class Project3D (line 62) | class Project3D(jit.ScriptModule): method __init__ (line 66) | def __init__(self, eps: float = 1e-8): method forward (line 72) | def forward(self, points_b4N: Tensor, class NormalGenerator (line 92) | class NormalGenerator(jit.ScriptModule): method __init__ (line 93) | def __init__(self, height: int, width: int, method forward (line 108) | def forward(self, depth_b1hw: Tensor, invK_b44: Tensor) -> Tensor: function get_angle_dif (line 135) | def get_angle_dif(matA_b33, matB_b33): function get_camera_rays (line 143) | def get_camera_rays( function pose_distance (line 178) | def pose_distance(pose_b44): function qvec2rotmat (line 193) | def qvec2rotmat(qvec): function rotx (line 213) | def rotx(t): function roty (line 223) | def roty(t): function rotz (line 233) | def rotz(t): FILE: 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 126) | def __init__(self, exp_name, metrics_name): method update_results (line 141) | def update_results(self, elem_metrics): method print_sheets_friendly (line 163) | def print_sheets_friendly( method output_json (line 200) | def output_json(self, filepath, print_running_metrics=False): method pretty_print_results (line 236) | def pretty_print_results( method compute_final_average (line 263) | def compute_final_average(self, ignore_nans=False): FILE: utils/visualization_utils.py function colormap_image (line 12) | def colormap_image( function save_viz_video_frames (line 74) | def save_viz_video_frames(frame_list, path, fps=30): function quick_viz_export (line 84) | def quick_viz_export( FILE: visualization_scripts/generate_gt_min_max_cache.py function main (line 26) | def main(opts): FILE: visualization_scripts/visualize_scene_depth_output.py function main (line 34) | def main(opts): FILE: visualize_live_meshing.py function main (line 27) | def main(opts):