SYMBOL INDEX (166 symbols across 17 files)

FILE: datasets/blendedmvs.py
  function motion_blur (line 19) | def motion_blur(img: np.ndarray, max_kernel_size=3):
  class BlendedMVSDataset (line 42) | class BlendedMVSDataset(Dataset):
    method __init__ (line 43) | def __init__(self, root_dir, list_file, split, n_views, **kwargs):
    method build_metas (line 69) | def build_metas(self):
    method read_cam_file (line 84) | def read_cam_file(self, scan, filename):
    method read_depth_mask (line 104) | def read_depth_mask(self, scan, filename, depth_min, depth_max, scale):
    method read_img (line 127) | def read_img(self, filename):
    method __len__ (line 138) | def __len__(self):
    method __getitem__ (line 142) | def __getitem__(self, idx):

FILE: datasets/data_io.py
  function read_pfm (line 11) | def read_pfm(filename):
  function save_pfm (line 49) | def save_pfm(filename, image, scale=1):
  function write_cam (line 79) | def write_cam(file, cam):

FILE: datasets/dtu.py
  class DTUDataset (line 19) | class DTUDataset(Dataset):
    method __init__ (line 20) | def __init__(self, root_dir, list_file, mode, n_views, **kwargs):
    method build_metas (line 43) | def build_metas(self):
    method __len__ (line 73) | def __len__(self):
    method read_cam_file (line 77) | def read_cam_file(self, filename):
    method read_img (line 103) | def read_img(self, filename):
    method crop_img (line 112) | def crop_img(self, img):
    method prepare_img (line 119) | def prepare_img(self, hr_img):
    method scale_mvs_input (line 132) | def scale_mvs_input(self, img, intrinsics, max_w, max_h, base=64):
    method read_mask_hr (line 152) | def read_mask_hr(self, filename):
    method read_depth_hr (line 168) | def read_depth_hr(self, filename, scale):
    method __getitem__ (line 182) | def __getitem__(self, idx):

FILE: datasets/tnt.py
  class TNTDataset (line 17) | class TNTDataset(Dataset):
    method __init__ (line 18) | def __init__(self, root_dir, list_file, split, n_views, **kwargs):
    method build_metas (line 42) | def build_metas(self):
    method read_cam_file (line 59) | def read_cam_file(self, filename):
    method read_img (line 75) | def read_img(self, filename):
    method scale_tnt_input (line 81) | def scale_tnt_input(self, intrinsics, img):
    method __len__ (line 102) | def __len__(self):
    method __getitem__ (line 106) | def __getitem__(self, idx):

FILE: fusions/dtu/_open3d.py
  function homo_warping (line 38) | def homo_warping(src_fea, src_proj, ref_proj, depth_values):
  function generate_points_from_depth (line 76) | def generate_points_from_depth(depth, proj):
  function mkdir_p (line 102) | def mkdir_p(path):
  function read_pfm (line 112) | def read_pfm(filename):
  function write_pfm (line 150) | def write_pfm(file, image, scale=1):
  function write_ply (line 180) | def write_ply(file, points):
  function filter_depth (line 187) | def filter_depth(ref_depth, src_depths, ref_proj, src_projs):
  function parse_cameras (line 209) | def parse_cameras(path):
  function load_data (line 222) | def load_data(root_path, depth_path, scene_name, thresh):
  function extract_points (line 260) | def extract_points(pc, mask, rgb):
  function open3d_filter (line 276) | def open3d_filter():

FILE: fusions/dtu/gipuma.py
  function load_cam (line 37) | def load_cam(file, interval_scale=1):
  function load_pfm (line 77) | def load_pfm(file):
  function write_pfm (line 110) | def write_pfm(file, image, scale=1):
  function read_gipuma_dmb (line 144) | def read_gipuma_dmb(path):
  function write_gipuma_dmb (line 159) | def write_gipuma_dmb(path, image):
  function mvsnet_to_gipuma_dmb (line 183) | def mvsnet_to_gipuma_dmb(in_path, out_path):
  function mvsnet_to_gipuma_cam (line 192) | def mvsnet_to_gipuma_cam(in_path, out_path):
  function fake_gipuma_normal (line 220) | def fake_gipuma_normal(in_depth_path, out_normal_path):
  function mvsnet_to_gipuma (line 242) | def mvsnet_to_gipuma(scan_folder, scan, root_dir, gipuma_point_folder):
  function probability_filter (line 287) | def probability_filter(scan_folder, prob_threshold):
  function depth_map_fusion (line 303) | def depth_map_fusion(point_folder, fusibile_exe_path, disp_thresh, num_c...
  function collectPly (line 326) | def collectPly(point_folder, scan_id):

FILE: fusions/dtu/pcd.py
  function read_pfm (line 36) | def read_pfm(filename):
  function read_camera_parameters (line 74) | def read_camera_parameters(filename):
  function read_img (line 85) | def read_img(filename):
  function read_mask (line 92) | def read_mask(filename):
  function save_mask (line 96) | def save_mask(filename, mask):
  function read_pair_file (line 102) | def read_pair_file(filename):
  function reproject_with_depth (line 116) | def reproject_with_depth(depth_ref, intrinsics_ref, extrinsics_ref, dept...
  function check_geometric_consistency (line 156) | def check_geometric_consistency(depth_ref, intrinsics_ref, extrinsics_re...
  function filter_depth (line 174) | def filter_depth(pair_folder, scan_folder, out_folder, plyfilename):
  function init_worker (line 266) | def init_worker():
  function pcd_filter_worker (line 273) | def pcd_filter_worker(scan):
  function pcd_filter (line 283) | def pcd_filter(testlist, number_worker):
  function initLogger (line 298) | def initLogger():

FILE: fusions/tnt/dypcd.py
  function read_pfm (line 72) | def read_pfm(filename):
  function save_mask (line 111) | def save_mask(filename, mask):
  function read_img (line 118) | def read_img(filename):
  function read_camera_parameters (line 126) | def read_camera_parameters(filename):
  function read_pair_file (line 140) | def read_pair_file(filename):
  function reproject_with_depth (line 154) | def reproject_with_depth(depth_ref, intrinsics_ref, extrinsics_ref, dept...
  function check_geometric_consistency (line 195) | def check_geometric_consistency(depth_ref, intrinsics_ref, extrinsics_re...
  function scale_input (line 218) | def scale_input(intrinsics, img):
  function filter_depth (line 231) | def filter_depth(scene, root_dir, split, out_dir, plyfilename, fusion_exp):
  function dypcd_filter_worker (line 359) | def dypcd_filter_worker(scene):
  function init_worker (line 365) | def init_worker():

FILE: models/filter.py
  function frequency_domain_filter (line 12) | def frequency_domain_filter(depth, rho_ratio):
  function visual_fft_fig (line 34) | def visual_fft_fig(fshift):

FILE: models/geometry.py
  class GeoFeatureFusion (line 17) | class GeoFeatureFusion(nn.Module):
    method __init__ (line 18) | def __init__(self, convolutional_layer_encoding="z", mask_type="basic"...
    method forward (line 84) | def forward(self, rgb, depth, confidence, depth_values, stage_idx, ori...
  class GeoRegNet2d (line 242) | class GeoRegNet2d(nn.Module):
    method __init__ (line 243) | def __init__(self, input_channel=128, base_channel=32, convolutional_l...
    method forward (line 275) | def forward(self, x, stage_idx, geo_reg_data=None):
  class BasicBlockGeo (line 329) | class BasicBlockGeo(nn.Module):
    method __init__ (line 333) | def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
    method forward (line 358) | def forward(self, x, g1=None, g2=None):
  class GeometryFeature (line 380) | class GeometryFeature(nn.Module):
    method __init__ (line 381) | def __init__(self):
    method forward (line 384) | def forward(self, z, vnorm, unorm, h, w, ch, cw, fh, fw):
  class SparseDownSampleClose (line 390) | class SparseDownSampleClose(nn.Module):
    method __init__ (line 391) | def __init__(self, stride):
    method forward (line 395) | def forward(self, d, mask):
  function convbnrelu (line 405) | def convbnrelu(in_channels, out_channels, kernel_size=3,stride=1, paddin...
  function deconvbnrelu (line 413) | def deconvbnrelu(in_channels, out_channels, kernel_size=5, stride=2, pad...
  function weights_init (line 421) | def weights_init(m):
  function conv3x3 (line 438) | def conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1, bias=...
  function conv1x1 (line 446) | def conv1x1(in_planes, out_planes, stride=1, groups=1, bias=False):
  class AddCoordsNp (line 451) | class AddCoordsNp():
    method __init__ (line 453) | def __init__(self, x_dim=64, y_dim=64, with_r=False):
    method call (line 458) | def call(self):
  class Reg_BasicBlockGeo (line 496) | class Reg_BasicBlockGeo(nn.Module):
    method __init__ (line 498) | def __init__(self, inplanes, planes, kernel_size, stride, padding, dow...
    method forward (line 515) | def forward(self, x, g1=None, g2=None):
  function regconv3D (line 537) | def regconv3D(in_planes, out_planes, kernel_size, stride, padding, group...
  function regconv1x1 (line 542) | def regconv1x1(in_planes, out_planes, kernel_size, stride, padding, grou...
  function reg_deconvbnrelu (line 546) | def reg_deconvbnrelu(in_channels, out_channels, kernel_size, stride, pad...

FILE: models/geomvsnet.py
  class GeoMVSNet (line 17) | class GeoMVSNet(nn.Module):
    method __init__ (line 18) | def __init__(self, levels, hypo_plane_num_stages, depth_interal_ratio_...
    method forward (line 52) | def forward(self, imgs, proj_matrices, intrinsics_matrices, depth_valu...
  class StageNet (line 131) | class StageNet(nn.Module):
    method __init__ (line 132) | def __init__(self, attn_temp=2):
    method forward (line 136) | def forward(self, stage_idx, features, proj_matrices, depth_hypo, regnet,

FILE: models/loss.py
  function geomvsnet_loss (line 10) | def geomvsnet_loss(inputs, depth_gt_ms, mask_ms, **kwargs):
  function pixel_wise_loss (line 50) | def pixel_wise_loss(prob_volume, depth_gt, mask, depth_value):
  function depth_distribution_similarity_loss (line 75) | def depth_distribution_similarity_loss(depth, depth_gt, mask, depth_min,...
  function cal_metrics (line 100) | def cal_metrics(depth_pred, depth_gt, mask, depth_min, depth_max):

FILE: models/submodules.py
  class FPN (line 12) | class FPN(nn.Module):
    method __init__ (line 14) | def __init__(self, base_channels, gn=False):
    method forward (line 57) | def forward(self, x):
  class Reg2d (line 84) | class Reg2d(nn.Module):
    method __init__ (line 85) | def __init__(self, input_channel=128, base_channel=32):
    method forward (line 115) | def forward(self, x):
  function homo_warping (line 128) | def homo_warping(src_fea, src_proj, ref_proj, depth_values):
  function init_inverse_range (line 174) | def init_inverse_range(cur_depth, ndepths, device, dtype, H, W):
  function schedule_inverse_range (line 183) | def schedule_inverse_range(inverse_min_depth, inverse_max_depth, ndepths...
  function init_bn (line 196) | def init_bn(module):
  function init_uniform (line 204) | def init_uniform(module, init_method):
  class ConvBnReLU3D (line 213) | class ConvBnReLU3D(nn.Module):
    method __init__ (line 214) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 219) | def forward(self, x):
  class Conv2d (line 223) | class Conv2d(nn.Module):
    method __init__ (line 225) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 237) | def forward(self, x):
    method init_weights (line 247) | def init_weights(self, init_method):

FILE: models/utils/opts.py
  function get_opts (line 9) | def get_opts():

FILE: models/utils/utils.py
  function make_nograd_func (line 15) | def make_nograd_func(func):
  function make_recursive_func (line 25) | def make_recursive_func(func):
  function tensor2float (line 40) | def tensor2float(vars):
  function tensor2numpy (line 50) | def tensor2numpy(vars):
  function tocuda (line 60) | def tocuda(vars):
  function tb_save_scalars (line 69) | def tb_save_scalars(logger, mode, scalar_dict, global_step):
  function tb_save_images (line 81) | def tb_save_images(logger, mode, images_dict, global_step):
  class DictAverageMeter (line 102) | class DictAverageMeter(object):
    method __init__ (line 103) | def __init__(self):
    method update (line 107) | def update(self, new_input):
    method mean (line 120) | def mean(self):
  function compute_metrics_for_each_image (line 125) | def compute_metrics_for_each_image(metric_func):
  function Thres_metrics (line 140) | def Thres_metrics(depth_est, depth_gt, mask, thres):
  function AbsDepthError_metrics (line 151) | def AbsDepthError_metrics(depth_est, depth_gt, mask, thres=None):
  function synchronize (line 162) | def synchronize():
  function get_world_size (line 177) | def get_world_size():
  function reduce_scalar_outputs (line 185) | def reduce_scalar_outputs(scalar_outputs):
  class WarmupMultiStepLR (line 208) | class WarmupMultiStepLR(torch.optim.lr_scheduler._LRScheduler):
    method __init__ (line 209) | def __init__(
    method get_lr (line 237) | def get_lr(self):
  function set_random_seed (line 253) | def set_random_seed(seed):

FILE: test.py
  function test (line 29) | def test():
  function initLogger (line 97) | def initLogger():

FILE: train.py
  function train (line 33) | def train(model, model_loss, optimizer, TrainImgLoader, TestImgLoader, s...
  function train_sample (line 116) | def train_sample(model, model_loss, optimizer, sample, args):
  function test_sample_depth (line 174) | def test_sample_depth(model, model_loss, sample, args):
  function initLogger (line 230) | def initLogger():