SYMBOL INDEX (96 symbols across 11 files)

FILE: CAPS/caps_model.py
  class CAPSModel (line 13) | class CAPSModel():
    method name (line 15) | def name(self):
    method __init__ (line 18) | def __init__(self, args):
    method set_input (line 34) | def set_input(self, data):
    method forward (line 49) | def forward(self):
    method backward_net (line 52) | def backward_net(self):
    method optimize_parameters (line 57) | def optimize_parameters(self):
    method test (line 64) | def test(self):
    method extract_features (line 70) | def extract_features(self, im, coord):
    method write_summary (line 76) | def write_summary(self, writer, n_iter):
    method load_model (line 107) | def load_model(self, filename):
    method load_from_ckpt (line 116) | def load_from_ckpt(self):
    method save_model (line 146) | def save_model(self, step):

FILE: CAPS/criterion.py
  class CtoFCriterion (line 5) | class CtoFCriterion(nn.Module):
    method __init__ (line 6) | def __init__(self, args):
    method homogenize (line 15) | def homogenize(self, coord):
    method set_weight (line 19) | def set_weight(self, std, mask=None, regularizer=0.0):
    method epipolar_cost (line 32) | def epipolar_cost(self, coord1, coord2, fmatrix):
    method epipolar_loss (line 40) | def epipolar_loss(self, coord1, coord2, fmatrix, weight):
    method cycle_consistency_loss (line 45) | def cycle_consistency_loss(self, coord1, coord1_loop, weight, th=40):
    method forward (line 60) | def forward(self, coord1, data, fmatrix, pose, im_size):

FILE: CAPS/network.py
  class CAPSNet (line 7) | class CAPSNet(nn.Module):
    method __init__ (line 8) | def __init__(self, args, device):
    method normalize (line 18) | def normalize(coord, h, w):
    method denormalize (line 31) | def denormalize(coord_norm, h, w):
    method ind2coord (line 43) | def ind2coord(self, ind, width):
    method gen_grid (line 50) | def gen_grid(self, h_min, h_max, w_min, w_max, len_h, len_w):
    method sample_feat_by_coord (line 55) | def sample_feat_by_coord(self, x, coord_n, norm=False):
    method compute_prob (line 69) | def compute_prob(self, feat1, feat2):
    method get_1nn_coord (line 87) | def get_1nn_coord(self, feat1, featmap2):
    method get_expected_correspondence_locs (line 111) | def get_expected_correspondence_locs(self, feat1, featmap2, with_std=F...
    method get_expected_correspondence_within_window (line 135) | def get_expected_correspondence_within_window(self, feat1, featmap2, c...
    method forward (line 167) | def forward(self, im1, im2, coord1):
    method extract_features (line 203) | def extract_features(self, im, coord):
    method test (line 217) | def test(self, im1, im2, coord1):
  function class_for_name (line 250) | def class_for_name(module_name, class_name):
  class conv (line 256) | class conv(nn.Module):
    method __init__ (line 257) | def __init__(self, num_in_layers, num_out_layers, kernel_size, stride):
    method forward (line 267) | def forward(self, x):
  class upconv (line 271) | class upconv(nn.Module):
    method __init__ (line 272) | def __init__(self, num_in_layers, num_out_layers, kernel_size, scale):
    method forward (line 277) | def forward(self, x):
  class ResUNet (line 282) | class ResUNet(nn.Module):
    method __init__ (line 283) | def __init__(self,
    method skipconnect (line 320) | def skipconnect(self, x1, x2):
    method forward (line 334) | def forward(self, x):

FILE: config.py
  function get_args (line 4) | def get_args():

FILE: dataloader/data_utils.py
  function skew (line 5) | def skew(x):
  function rotateImage (line 11) | def rotateImage(image, angle):
  function perspective_transform (line 28) | def perspective_transform(img, param=0.001):
  function generate_query_kpts (line 45) | def generate_query_kpts(img, mode, num_pts, h, w):
  function prune_kpts (line 78) | def prune_kpts(coord1, F_gt, im2_size, intrinsic1, intrinsic2, pose, d_m...

FILE: dataloader/megadepth.py
  class MegaDepthLoader (line 17) | class MegaDepthLoader():
    method __init__ (line 18) | def __init__(self, args):
    method my_collate (line 24) | def my_collate(self, batch):
    method load_data (line 29) | def load_data(self):
    method name (line 32) | def name(self):
    method __len__ (line 35) | def __len__(self):
  class MegaDepth (line 39) | class MegaDepth(Dataset):
    method __init__ (line 40) | def __init__(self, args):
    method read_img_cam (line 67) | def read_img_cam(self):
    method read_pairs (line 97) | def read_pairs(self):
    method get_intrinsics (line 129) | def get_intrinsics(im_meta):
    method get_extrinsics (line 135) | def get_extrinsics(im_meta):
    method __getitem__ (line 143) | def __getitem__(self, item):
    method __len__ (line 209) | def __len__(self):

FILE: extract_features.py
  class HPatchDataset (line 13) | class HPatchDataset(Dataset):
    method __init__ (line 14) | def __init__(self, imdir):
    method __getitem__ (line 24) | def __getitem__(self, item):
    method __len__ (line 37) | def __len__(self):

FILE: jupyter/functions.py
  class Visualization (line 13) | class Visualization(object):
    method __init__ (line 14) | def __init__(self, args):
    method random_sample (line 20) | def random_sample(self):
    method plot_img_pair (line 23) | def plot_img_pair(self, with_std=False, with_epipline=False):
    method onclick (line 42) | def onclick(self, event):
    method find_correspondence (line 53) | def find_correspondence(self):
    method plot_correspondence (line 62) | def plot_correspondence(self):

FILE: test/eval_pose_megadepth.py
  class MegaDepthPose (line 16) | class MegaDepthPose(object):
    method __init__ (line 17) | def __init__(self, pose_args, mode):
    method read_img_cam (line 27) | def read_img_cam(self):
    method read_pairs (line 54) | def read_pairs(self):
    method load_kp_desc (line 66) | def load_kp_desc(self, imf):
    method compose_intrinsic_extrinsic (line 74) | def compose_intrinsic_extrinsic(self, imf):
    method get_pose_error (line 86) | def get_pose_error(self, kp1, kp2, desc1, desc2, intrinsic1, intrinsic...
    method visMatch (line 199) | def visMatch(self, imf1, imf2, pt1, pt2, mask, text, R_error, T_error,...
    method pose_parallel (line 225) | def pose_parallel(self, imf1, imf2, vis=False):
    method run (line 243) | def run(self):

FILE: train.py
  function train_megadepth (line 9) | def train_megadepth(args):

FILE: utils.py
  function cycle (line 6) | def cycle(iterable):
  function evaluate_pose (line 12) | def evaluate_pose(E, P):
  function average_precision (line 25) | def average_precision(labels, logits):
  function homogenize (line 45) | def homogenize(kp):
  function random_choice (line 55) | def random_choice(array, size):
  function drawlines (line 65) | def drawlines(img1, img2, lines, pts1, pts2, color=None, thickness=-1):
  function to_jet (line 87) | def to_jet(input, type='tensor', mode='HW1'):
  function drawlinesMatch (line 110) | def drawlinesMatch(img1, img2, pts1, pts2, concat_row=True):