SYMBOL INDEX (265 symbols across 13 files)

FILE: datasets/__init__.py
  function find_dataset_def (line 5) | def find_dataset_def(dataset_name):

FILE: datasets/blendedmvs.py
  function check_invalid_input (line 11) | def check_invalid_input(imgs, depths, masks, depth_mins, depth_maxs):
  class MVSDataset (line 26) | class MVSDataset(Dataset):
    method __init__ (line 27) | def __init__(self, datapath, listfile, split, nviews, img_wh=(768, 576...
    method build_metas (line 49) | def build_metas(self):
    method read_cam_file (line 62) | def read_cam_file(self, scan, filename):
    method read_depth_mask (line 81) | def read_depth_mask(self, scan, filename, depth_min, depth_max, scale):
    method read_img (line 108) | def read_img(self, filename):
    method __len__ (line 115) | def __len__(self):
    method __getitem__ (line 118) | def __getitem__(self, idx):

FILE: datasets/data_io.py
  function read_pfm (line 6) | def read_pfm(filename):
  function save_pfm (line 44) | def save_pfm(filename, image, scale=1):
  class RandomCrop (line 75) | class RandomCrop(object):
    method __init__ (line 76) | def __init__(self, CropSize=0.1):
    method __call__ (line 79) | def __call__(self, image, normal):

FILE: datasets/dtu_yao4.py
  class MVSDataset (line 9) | class MVSDataset(Dataset):
    method __init__ (line 10) | def __init__(self, datapath, listfile, mode, nviews, interval_scale=1....
    method build_list (line 26) | def build_list(self):
    method __len__ (line 48) | def __len__(self):
    method read_cam_file (line 51) | def read_cam_file(self, filename):
    method read_img (line 64) | def read_img(self, filename):
    method crop_img (line 72) | def crop_img(self, img):
    method prepare_img (line 78) | def prepare_img(self, hr_img):
    method read_mask_hr (line 92) | def read_mask_hr(self, filename):
    method read_depth_hr (line 108) | def read_depth_hr(self, filename, scale):
    method __getitem__ (line 123) | def __getitem__(self, idx):

FILE: datasets/eth3d.py
  class MVSDataset (line 8) | class MVSDataset(Dataset):
    method __init__ (line 9) | def __init__(self, datapath, split='test', n_views=7, img_wh=(1920,128...
    method build_metas (line 17) | def build_metas(self):
    method read_cam_file (line 40) | def read_cam_file(self, filename):
    method read_img (line 57) | def read_img(self, filename):
    method __len__ (line 64) | def __len__(self):
    method __getitem__ (line 67) | def __getitem__(self, idx):

FILE: datasets/general_eval4.py
  class MVSDataset (line 8) | class MVSDataset(Dataset):
    method __init__ (line 9) | def __init__(self, datapath, listfile, mode, nviews, interval_scale=1....
    method build_list (line 24) | def build_list(self):
    method __len__ (line 56) | def __len__(self):
    method read_cam_file (line 59) | def read_cam_file(self, filename, interval_scale):
    method read_img (line 81) | def read_img(self, filename):
    method read_depth (line 88) | def read_depth(self, filename):
    method scale_mvs_input (line 92) | def scale_mvs_input(self, img, intrinsics, max_w, max_h, base=64):
    method __getitem__ (line 111) | def __getitem__(self, idx):

FILE: datasets/tanks.py
  class MVSDataset (line 8) | class MVSDataset(Dataset):
    method __init__ (line 9) | def __init__(self, datapath, n_views=7, split='intermediate'):
    method build_metas (line 16) | def build_metas(self):
    method read_cam_file (line 33) | def read_cam_file(self, filename):
    method read_img (line 48) | def read_img(self, filename):
    method scale_input (line 53) | def scale_input(self, intrinsics, img):
    method __len__ (line 62) | def __len__(self):
    method __getitem__ (line 65) | def __getitem__(self, idx):

FILE: models/MVS4Net.py
  class MVS4net (line 9) | class MVS4net(nn.Module):
    method __init__ (line 10) | def __init__(self, arch_mode="fpn", reg_net='reg2d', num_stage=4, fpn_...
    method forward (line 60) | def forward(self, imgs, proj_matrices, depth_values, filename=None):
  function MVS4net_loss (line 113) | def MVS4net_loss(inputs, depth_gt_ms, mask_ms, **kwargs):
  function Blend_loss (line 158) | def Blend_loss(inputs, depth_gt_ms, mask_ms, **kwargs):

FILE: models/module.py
  function init_bn (line 14) | def init_bn(module):
  function init_uniform (line 22) | def init_uniform(module, init_method):
  class Conv2d (line 30) | class Conv2d(nn.Module):
    method __init__ (line 44) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 58) | def forward(self, x):
    method init_weights (line 66) | def init_weights(self, init_method):
  class DCNConv2d (line 72) | class DCNConv2d(nn.Module):
    method __init__ (line 74) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 87) | def forward(self, x):
    method init_weights (line 95) | def init_weights(self, init_method):
  class Deconv2d (line 101) | class Deconv2d(nn.Module):
    method __init__ (line 115) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 130) | def forward(self, x):
    method init_weights (line 141) | def init_weights(self, init_method):
  class Conv3d (line 147) | class Conv3d(nn.Module):
    method __init__ (line 161) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,
    method forward (line 177) | def forward(self, x):
    method init_weights (line 185) | def init_weights(self, init_method):
  class PConv3d (line 191) | class PConv3d(nn.Module):
    method __init__ (line 193) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,
    method forward (line 213) | def forward(self, x):
    method init_weights (line 222) | def init_weights(self, init_method):
  class Deconv3d (line 230) | class Deconv3d(nn.Module):
    method __init__ (line 244) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,
    method forward (line 259) | def forward(self, x):
    method init_weights (line 267) | def init_weights(self, init_method):
  class PDeconv3d (line 274) | class PDeconv3d(nn.Module):
    method __init__ (line 276) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 298) | def forward(self, x):
    method init_weights (line 307) | def init_weights(self, init_method):
  class ConvBnReLU (line 314) | class ConvBnReLU(nn.Module):
    method __init__ (line 315) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 320) | def forward(self, x):
  class ConvBn (line 323) | class ConvBn(nn.Module):
    method __init__ (line 324) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 329) | def forward(self, x):
  class ConvBnReLU3D (line 332) | class ConvBnReLU3D(nn.Module):
    method __init__ (line 333) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 338) | def forward(self, x):
  class ConvBn3D (line 342) | class ConvBn3D(nn.Module):
    method __init__ (line 343) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 348) | def forward(self, x):
  class BasicBlock (line 352) | class BasicBlock(nn.Module):
    method __init__ (line 353) | def __init__(self, in_channels, out_channels, stride, downsample=None):
    method forward (line 362) | def forward(self, x):
  class Hourglass3d (line 371) | class Hourglass3d(nn.Module):
    method __init__ (line 372) | def __init__(self, channels):
    method forward (line 394) | def forward(self, x):
  function homo_warping (line 402) | def homo_warping(src_fea, src_proj, ref_proj, depth_values, align_corner...
  class DeConv2dFuse (line 437) | class DeConv2dFuse(nn.Module):
    method __init__ (line 438) | def __init__(self, in_channels, out_channels, kernel_size, relu=True, ...
    method forward (line 451) | def forward(self, x_pre, x):
  class FeatureNet (line 458) | class FeatureNet(nn.Module):
    method __init__ (line 459) | def __init__(self, base_channels, num_stage=3, stride=4, arch_mode="un...
    method forward (line 520) | def forward(self, x):
  class FPNDCNpath (line 561) | class FPNDCNpath(nn.Module):
    method __init__ (line 564) | def __init__(self, base_channels, stride=4):
    method forward (line 612) | def forward(self, x):
  class FPNDCN (line 635) | class FPNDCN(nn.Module):
    method __init__ (line 638) | def __init__(self, base_channels, stride=4):
    method forward (line 684) | def forward(self, x):
  class FPNA (line 705) | class FPNA(nn.Module):
    method __init__ (line 708) | def __init__(self, base_channels, stride=4):
    method forward (line 743) | def forward(self, x):
  class FPNA4 (line 764) | class FPNA4(nn.Module):
    method __init__ (line 767) | def __init__(self, base_channels):
    method forward (line 810) | def forward(self, x):
  class CostRegNet (line 836) | class CostRegNet(nn.Module):
    method __init__ (line 837) | def __init__(self, in_channels, base_channels, down_size=3):
    method forward (line 860) | def forward(self, x):
  class P3DConv (line 884) | class P3DConv(nn.Module):
    method __init__ (line 888) | def __init__(self, in_channels, base_channels):
    method forward (line 909) | def forward(self, x):
  class RefineNet (line 920) | class RefineNet(nn.Module):
    method __init__ (line 921) | def __init__(self):
    method forward (line 928) | def forward(self, img, depth_init):
  function depth_regression (line 935) | def depth_regression(p, depth_values):
  function cas_mvsnet_loss (line 943) | def cas_mvsnet_loss(inputs, depth_gt_ms, mask_ms, **kwargs):
  function cas_mvsnet_T_loss (line 964) | def cas_mvsnet_T_loss(inputs, depth_gt_ms, mask_ms, **kwargs):
  function get_cur_depth_range_samples (line 1138) | def get_cur_depth_range_samples(cur_depth, ndepth, depth_inteval_pixel, ...
  function get_depth_range_samples (line 1157) | def get_depth_range_samples(cur_depth, ndepth, depth_inteval_pixel, devi...

FILE: models/mvs4net_utils.py
  function homo_warping (line 13) | def homo_warping(src_fea, src_proj, ref_proj, depth_values, vis_ETA=Fals...
  function init_range (line 61) | def init_range(cur_depth, ndepths, device, dtype, H, W):
  function init_inverse_range (line 71) | def init_inverse_range(cur_depth, ndepths, device, dtype, H, W):
  function schedule_inverse_range (line 79) | def schedule_inverse_range(inverse_min_depth, inverse_max_depth, ndepths...
  function schedule_range (line 88) | def schedule_range(cur_depth, ndepth, depth_inteval_pixel, H, W):
  function init_bn (line 101) | def init_bn(module):
  function init_uniform (line 108) | def init_uniform(module, init_method):
  class ConvBnReLU3D (line 116) | class ConvBnReLU3D(nn.Module):
    method __init__ (line 117) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 122) | def forward(self, x):
  class ConvBnReLU3D_CAM (line 125) | class ConvBnReLU3D_CAM(nn.Module):
    method __init__ (line 126) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 136) | def forward(self, input):
  class ConvBnReLU3D_DCAM (line 145) | class ConvBnReLU3D_DCAM(nn.Module):
    method __init__ (line 146) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 156) | def forward(self, input):
  class ConvBnReLU3D_PAM (line 165) | class ConvBnReLU3D_PAM(nn.Module):
    method __init__ (line 166) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 172) | def forward(self, input):
  class ConvBnReLU3D_PDAM (line 181) | class ConvBnReLU3D_PDAM(nn.Module):
    method __init__ (line 182) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,...
    method forward (line 188) | def forward(self, input):
  class Deconv3d (line 197) | class Deconv3d(nn.Module):
    method __init__ (line 199) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,
    method forward (line 211) | def forward(self, x):
    method init_weights (line 219) | def init_weights(self, init_method):
  class Conv2d (line 224) | class Conv2d(nn.Module):
    method __init__ (line 226) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 238) | def forward(self, x):
    method init_weights (line 248) | def init_weights(self, init_method):
  class Deconv2d (line 253) | class Deconv2d(nn.Module):
    method __init__ (line 255) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,
  class DeformConv2d (line 267) | class DeformConv2d(nn.Module):
    method __init__ (line 268) | def __init__(self, inc, outc, kernel_size=3, padding=1, stride=1, bias...
    method _set_lr (line 287) | def _set_lr(module, grad_input, grad_output):
    method forward (line 291) | def forward(self, x):
    method _get_p_n (line 349) | def _get_p_n(self, N, dtype):
    method _get_p_0 (line 359) | def _get_p_0(self, h, w, N, dtype):
    method _get_p (line 369) | def _get_p(self, offset, dtype):
    method _get_x_q (line 379) | def _get_x_q(self, x, q, N):
    method _reshape_x_offset (line 396) | def _reshape_x_offset(x_offset, ks):
  function NA_DCN (line 403) | def NA_DCN(in_channels, kernel_size=3, stride=1, dilation=1, bias=True, ...
  class FPN4 (line 419) | class FPN4(nn.Module):
    method __init__ (line 422) | def __init__(self, base_channels, gn=False, dcn=False):
    method forward (line 472) | def forward(self, x):
  class LayerNorm (line 504) | class LayerNorm(nn.Module):
    method __init__ (line 506) | def __init__(self, normalized_shape, eps=1e-6, data_format="channels_l...
    method forward (line 516) | def forward(self, x):
  class convnext_block (line 526) | class convnext_block(nn.Module):
    method __init__ (line 528) | def __init__(self, dim, layer_scale_init_value=1e-6):
    method forward (line 538) | def forward(self, x):
  class convnext4_block (line 553) | class convnext4_block(nn.Module):
    method __init__ (line 555) | def __init__(self, dim, layer_scale_init_value=1e-6):
    method forward (line 566) | def forward(self, x):
  class FPN4_convnext (line 581) | class FPN4_convnext(nn.Module):
    method __init__ (line 584) | def __init__(self, base_channels, gn=False, dcn=False):
    method forward (line 620) | def forward(self, x):
  class FPN4_convnext4 (line 652) | class FPN4_convnext4(nn.Module):
    method __init__ (line 655) | def __init__(self, base_channels, gn=False, dcn=False):
    method forward (line 691) | def forward(self, x):
  class ASFF (line 723) | class ASFF(nn.Module):
    method __init__ (line 724) | def __init__(self, level):
    method forward (line 758) | def forward(self, x_level_0, x_level_1, x_level_2, x_level_3):
  class FullImageEncoder (line 807) | class FullImageEncoder(nn.Module):
    method __init__ (line 808) | def __init__(self, h, w, kernel_size):
    method forward (line 819) | def forward(self, x):
  class mono_depth_decoder (line 833) | class mono_depth_decoder(nn.Module):
    method __init__ (line 835) | def __init__(self):
    method forward (line 849) | def forward(self, outputs, d_min, d_max):
  class reg2d (line 870) | class reg2d(nn.Module):
    method __init__ (line 871) | def __init__(self, input_channel=128, base_channel=32, conv_name='Conv...
    method forward (line 902) | def forward(self, x):
  class reg3d (line 914) | class reg3d(nn.Module):
    method __init__ (line 915) | def __init__(self, in_channels, base_channels, down_size=3):
    method forward (line 943) | def forward(self, x):
  class PosEncSine (line 967) | class PosEncSine(nn.Module):
    method __init__ (line 969) | def __init__(self, temperature=1000):
    method forward (line 973) | def forward(self, x, depth):
  class PosEncLearned (line 983) | class PosEncLearned(nn.Module):
    method __init__ (line 987) | def __init__(self, D, C):
    method reset_parameters (line 994) | def reset_parameters(self):
    method forward (line 997) | def forward(self, x, **kwargs):
  class stagenet (line 1003) | class stagenet(nn.Module):
    method __init__ (line 1004) | def __init__(self, inverse_depth=False, mono=False, attn_fuse_d=True, ...
    method forward (line 1012) | def forward(self, features, proj_matrices, depth_hypo, regnet, stage_i...
  function sinkhorn (line 1096) | def sinkhorn(gt_depth, hypo_depth, attn_weight, mask, iters, eps=1, cont...

FILE: test_mvs4.py
  function read_camera_parameters (line 94) | def read_camera_parameters(filename):
  function read_img (line 106) | def read_img(filename):
  function read_mask (line 114) | def read_mask(filename):
  function save_mask (line 119) | def save_mask(filename, mask):
  function read_pair_file (line 126) | def read_pair_file(filename):
  function write_cam (line 138) | def write_cam(file, cam):
  function save_depth (line 157) | def save_depth(testlist):
  function save_scene_depth (line 170) | def save_scene_depth(testlist):
  function reproject_with_depth (line 273) | def reproject_with_depth(depth_ref, intrinsics_ref, extrinsics_ref, dept...
  function check_geometric_consistency (line 313) | def check_geometric_consistency(depth_ref, intrinsics_ref, extrinsics_re...
  function filter_depth (line 331) | def filter_depth(pair_folder, scan_folder, out_folder, plyfilename):
  function init_worker (line 424) | def init_worker():
  function pcd_filter_worker (line 431) | def pcd_filter_worker(scan):
  function pcd_filter (line 443) | def pcd_filter(testlist, number_worker):
  function mrun_rst (line 457) | def mrun_rst(eval_dir, plyPath):

FILE: train_mvs4.py
  function train (line 83) | def train(model, model_loss, optimizer, TrainImgLoader, TestImgLoader, s...
  function test (line 179) | def test(model, model_loss, TestImgLoader, args):
  function train_sample (line 195) | def train_sample(model, model_loss, optimizer, sample, args):
  function test_sample_depth (line 253) | def test_sample_depth(model, model_loss, sample, args):

FILE: utils.py
  function print_args (line 8) | def print_args(args):
  function make_nograd_func (line 16) | def make_nograd_func(func):
  function make_recursive_func (line 26) | def make_recursive_func(func):
  function tensor2float (line 41) | def tensor2float(vars):
  function tensor2numpy (line 51) | def tensor2numpy(vars):
  function tocuda (line 61) | def tocuda(vars):
  function save_scalars (line 70) | def save_scalars(logger, mode, scalar_dict, global_step):
  function save_images (line 82) | def save_images(logger, mode, images_dict, global_step):
  class DictAverageMeter (line 103) | class DictAverageMeter(object):
    method __init__ (line 104) | def __init__(self):
    method update (line 108) | def update(self, new_input):
    method mean (line 121) | def mean(self):
  function compute_metrics_for_each_image (line 126) | def compute_metrics_for_each_image(metric_func):
  function Thres_metrics (line 141) | def Thres_metrics(depth_est, depth_gt, mask, thres):
  function AbsDepthError_metrics (line 152) | def AbsDepthError_metrics(depth_est, depth_gt, mask, thres=None):
  function synchronize (line 162) | def synchronize():
  function get_world_size (line 176) | def get_world_size():
  function reduce_scalar_outputs (line 183) | 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):
  function local_pcd (line 260) | def local_pcd(depth, intr):
  function generate_pointcloud (line 274) | def generate_pointcloud(rgb, depth, ply_file, intr, scale=1.0):