SYMBOL INDEX (146 symbols across 20 files)

FILE: ckpts/download.py
  function download_file_from_google_drive (line 6) | def download_file_from_google_drive(id, destination):

FILE: data/dataloader_7scenes.py
  function _read_ExtM_from_txt (line 16) | def _read_ExtM_from_txt(fpath_txt):
  class SevenScenesLoader (line 30) | class SevenScenesLoader(object):
    method __init__ (line 31) | def __init__(self, args, mode):
  class SevenScenesLoadPreprocess (line 36) | class SevenScenesLoadPreprocess(Dataset):
    method __init__ (line 37) | def __init__(self, args, mode):
    method __len__ (line 68) | def __len__(self):
    method get_ray_array (line 72) | def get_ray_array(self):
    method get_cam_intrinsics (line 83) | def get_cam_intrinsics(self):
    method __getitem__ (line 118) | def __getitem__(self, idx):

FILE: data/dataloader_7scenes_D.py
  class SevenScenesLoader (line 16) | class SevenScenesLoader(object):
    method __init__ (line 17) | def __init__(self, args, mode):
  class SevenScenesLoadPreprocess (line 22) | class SevenScenesLoadPreprocess(Dataset):
    method __init__ (line 23) | def __init__(self, args, mode):
    method __len__ (line 38) | def __len__(self):
    method __getitem__ (line 41) | def __getitem__(self, idx):

FILE: data/dataloader_kitti.py
  class KittiLoader (line 17) | class KittiLoader(object):
    method __init__ (line 18) | def __init__(self, args, mode):
  class KittiLoadPreprocess (line 38) | class KittiLoadPreprocess(Dataset):
    method __init__ (line 39) | def __init__(self, args, mode):
    method __len__ (line 79) | def __len__(self):
    method get_ray_array (line 83) | def get_ray_array(self):
    method get_cam_intrinsics (line 94) | def get_cam_intrinsics(self, p_data):
    method __getitem__ (line 129) | def __getitem__(self, idx):
    method augment_image (line 201) | def augment_image(self, image, gamma, brightness, colors):

FILE: data/dataloader_kitti_D.py
  class KittiLoader (line 16) | class KittiLoader(object):
    method __init__ (line 17) | def __init__(self, args, mode):
  class KittiLoadPreprocess (line 37) | class KittiLoadPreprocess(Dataset):
    method __init__ (line 38) | def __init__(self, args, mode):
    method __len__ (line 66) | def __len__(self):
    method __getitem__ (line 69) | def __getitem__(self, idx):
    method random_crop (line 134) | def random_crop(self, img, depth, height, width):
    method augment_image (line 145) | def augment_image(self, image):

FILE: data/dataloader_scannet.py
  function _read_ExtM_from_txt (line 16) | def _read_ExtM_from_txt(fpath_txt):
  function _read_IntM_from_txt (line 30) | def _read_IntM_from_txt(fpath_txt):
  class ScannetLoader (line 43) | class ScannetLoader(object):
    method __init__ (line 44) | def __init__(self, args, mode):
  class ScannetLoadPreprocess (line 64) | class ScannetLoadPreprocess(Dataset):
    method __init__ (line 65) | def __init__(self, args, mode):
    method __len__ (line 109) | def __len__(self):
    method get_ray_array (line 113) | def get_ray_array(self):
    method get_cam_intrinsics (line 124) | def get_cam_intrinsics(self, scene_dir, scene_name):
    method __getitem__ (line 155) | def __getitem__(self, idx):
    method augment_image (line 219) | def augment_image(self, image, gamma, brightness, colors):

FILE: data/dataloader_scannet_D.py
  class ScannetLoader (line 16) | class ScannetLoader(object):
    method __init__ (line 17) | def __init__(self, args, mode):
  class ScannetLoadPreprocess (line 37) | class ScannetLoadPreprocess(Dataset):
    method __init__ (line 38) | def __init__(self, args, mode):
    method __len__ (line 70) | def __len__(self):
    method __getitem__ (line 73) | def __getitem__(self, idx):
    method random_crop (line 129) | def random_crop(self, img, depth, height, width):
    method augment_image (line 140) | def augment_image(self, image):

FILE: models/DNET.py
  class DNET (line 7) | class DNET(nn.Module):
    method __init__ (line 8) | def __init__(self, args, dnet=True):
    method forward (line 50) | def forward(self, img, **kwargs):
    method activation_none (line 53) | def activation_none(self, out):
    method activation_G (line 56) | def activation_G(self, out):
    method activation_G_magnet (line 62) | def activation_G_magnet(self, outs):

FILE: models/FNET.py
  class FNET (line 7) | class FNET(nn.Module):
    method __init__ (line 8) | def __init__(self, args):
    method forward (line 19) | def forward(self, img):

FILE: models/MAGNET.py
  function upsample_depth_via_mask (line 15) | def upsample_depth_via_mask(depth, up_mask, k):
  function load_checkpoint (line 31) | def load_checkpoint(fpath, model):
  class GNET (line 47) | class GNET(nn.Module):
    method __init__ (line 48) | def __init__(self, ch_in, ch_out=2):
    method forward (line 58) | def forward(self, cost_volume, ref_gmm):
  class MAGNET (line 73) | class MAGNET(nn.Module):
    method __init__ (line 74) | def __init__(self, args):
    method depth_sampling (line 120) | def depth_sampling(self):
    method forward (line 130) | def forward(self, ref_img, nghbr_imgs, nghbr_poses, is_valid, cam_intr...
  class MAGNET_F (line 179) | class MAGNET_F(nn.Module):
    method __init__ (line 180) | def __init__(self, args):
    method forward (line 184) | def forward(self, ref_img, nghbr_imgs, nghbr_poses, is_valid, cam_intr...

FILE: models/submodules/D_dense_depth.py
  class Encoder (line 7) | class Encoder(nn.Module):
    method __init__ (line 8) | def __init__(self):
    method forward (line 17) | def forward(self, x):
  class UpSampleBN (line 29) | class UpSampleBN(nn.Module):
    method __init__ (line 30) | def __init__(self, skip_input, output_features):
    method forward (line 39) | def forward(self, x, concat_with):
  class UpSampleGN (line 46) | class UpSampleGN(nn.Module):
    method __init__ (line 47) | def __init__(self, skip_input, output_features):
    method forward (line 56) | def forward(self, x, concat_with):
  class Conv2d (line 63) | class Conv2d(nn.Conv2d):
    method __init__ (line 64) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 69) | def forward(self, x):
  function upsample_depth_via_bilinear (line 81) | def upsample_depth_via_bilinear(depth, up_mask, downsample_ratio):
  function upsample_depth_via_mask (line 86) | def upsample_depth_via_mask(depth, up_mask, downsample_ratio):
  class Decoder (line 104) | class Decoder(nn.Module):
    method __init__ (line 105) | def __init__(self, num_classes, downsample_ratio, learned_upsampling, ...
    method forward (line 166) | def forward(self, features):
  class DenseDepth (line 199) | class DenseDepth(nn.Module):
    method __init__ (line 200) | def __init__(self, n_bins, downsample_ratio, learned_upsampling, BN=Tr...
    method forward (line 205) | def forward(self, x):
    method get_1x_lr_params (line 208) | def get_1x_lr_params(self):  # lr/10 learning rate
    method get_10x_lr_params (line 211) | def get_10x_lr_params(self):  # lr learning rate

FILE: models/submodules/F_psmnet.py
  function convbn (line 10) | def convbn(in_planes, out_planes, kernel_size, stride, pad, dilation):
  class BasicBlock (line 18) | class BasicBlock(nn.Module):
    method __init__ (line 20) | def __init__(self, inplanes, planes, stride, downsample, pad, dilation):
    method forward (line 28) | def forward(self, x):
  class PSMNet (line 37) | class PSMNet(nn.Module):
    method __init__ (line 38) | def __init__(self, feature_dim=32):
    method _make_layer (line 87) | def _make_layer(self, block, planes, blocks, stride, pad, dilation):
    method forward (line 103) | def forward(self, x):

FILE: models/submodules/homography.py
  function est_costvolume_F (line 10) | def est_costvolume_F(d_center, ref_feat, nghbr_feat, R, t, is_valid, cam...
  function _compute_cost_F (line 50) | def _compute_cost_F(ref_feat_, nghbr_feat_, d_center, term1_pix, term2_p...
  function est_costvolume_CW (line 79) | def est_costvolume_CW(d_volume, ref_feat, nghbr_feat, ref_gmms, nghbr_gmms,
  function _compute_cost_CW (line 124) | def _compute_cost_CW(ref_feat_, nghbr_feat_, nghbr_mu_, nghbr_sigma_, d_...

FILE: test_DNet.py
  function validate (line 22) | def validate(model, args, test_loader, device='cpu', vis_dir=None):

FILE: test_MaGNet.py
  function validate (line 27) | def validate(model, args, test_loader, device, vis_dir=None):

FILE: train_DNet.py
  function train (line 19) | def train(model, args, device):
  function validate (line 130) | def validate(model, args, test_loader, device='cpu'):
  function main_worker (line 180) | def main_worker(gpu, ngpus_per_node, args):

FILE: train_FNet.py
  function train (line 19) | def train(model, args, device):
  function validate (line 148) | def validate(model, args, test_loader, device, d_center):
  function main_worker (line 199) | def main_worker(gpu, ngpus_per_node, args):

FILE: train_MaGNet.py
  function train (line 19) | def train(model, args, device):
  function validate (line 132) | def validate(model, args, test_loader, device='cpu'):
  function main_worker (line 186) | def main_worker(gpu, ngpus_per_node, args):

FILE: utils/losses.py
  class DnetLoss (line 8) | class DnetLoss(nn.Module):
    method __init__ (line 9) | def __init__(self, args):
    method forward (line 13) | def forward(self, pred, gt_depth, gt_depth_mask):
  class MagnetLoss (line 28) | class MagnetLoss(nn.Module):
    method __init__ (line 29) | def __init__(self, args):
    method forward (line 34) | def forward(self, pred_list, gt_depth, gt_depth_mask):

FILE: utils/utils.py
  function convert_arg_line_to_args (line 18) | def convert_arg_line_to_args(arg_line):
  function save_args (line 25) | def save_args(args, filename):
  function write_to_log (line 31) | def write_to_log(txt_filename, msg):
  function makedir (line 36) | def makedir(dirpath):
  function make_dir_from_list (line 41) | def make_dir_from_list(dirpath_list):
  function load_checkpoint (line 46) | def load_checkpoint(fpath, model):
  function split_data_array (line 64) | def split_data_array(data_array):
  function data_preprocess (line 72) | def data_preprocess(data_array, cur_batch_size):
  function compute_depth_errors (line 106) | def compute_depth_errors(gt, pred, var=None):
  class RunningAverage (line 147) | class RunningAverage:
    method __init__ (line 148) | def __init__(self):
    method append (line 152) | def append(self, value):
    method get_value (line 156) | def get_value(self):
  class RunningAverageDict (line 160) | class RunningAverageDict:
    method __init__ (line 161) | def __init__(self):
    method update (line 164) | def update(self, new_dict):
    method get_value (line 173) | def get_value(self):
  function log_metrics (line 177) | def log_metrics(txt_path, metrics, first_line):
  function unnormalize (line 206) | def unnormalize(img_in):
  function visualize_D (line 216) | def visualize_D(args, img, gt_dmap, gt_dmap_mask, out, total_iter):
  function visualize_F (line 257) | def visualize_F(args, img, gt_dmap, gt_dmap_mask, pred_dmap, total_iter):
  function visualize_MaG (line 294) | def visualize_MaG(args, img, gt_dmap, gt_dmap_mask, pred_list, total_iter):