SYMBOL INDEX (198 symbols across 16 files)

FILE: data/common.py
  function _assert_valid_param (line 6) | def _assert_valid_param(param):
  function assert_valid_param (line 12) | def assert_valid_param(param):
  function _fit_line (line 17) | def _fit_line(pts):
  function fit_line (line 28) | def fit_line(pts):
  function dist_pts_to_line (line 32) | def dist_pts_to_line(pts, param):
  function assert_pts_in_line (line 41) | def assert_pts_in_line(pts, param, atol=1.):
  function _find_pt_in_line (line 47) | def _find_pt_in_line(param):
  function find_pt_in_line (line 57) | def find_pt_in_line(param):
  function project_pts_on_line (line 61) | def project_pts_on_line(pts, param):
  function find_lines_intersect (line 74) | def find_lines_intersect(params):
  function is_pt_in_line_seg (line 90) | def is_pt_in_line_seg(eps, pt, pt1, pt2):

FILE: data/line_graph.py
  class LineGraph (line 11) | class LineGraph(object):
    method __init__ (line 12) | def __init__(
    method load (line 27) | def load(self, filename):
    method save (line 39) | def save(self, filename):
    method _is_pt_in_line_seg (line 52) | def _is_pt_in_line_seg(self, pt, pt1, pt2):
    method freeze_junction (line 60) | def freeze_junction(self, status=True):
    method _can_be_extented_by (line 80) | def _can_be_extented_by(self, line_seg1, line_seg2):
    method freeze_line_seg (line 93) | def freeze_line_seg(self, status=True):
    method add_junction (line 377) | def add_junction(self, junction):
    method add_line_seg (line 380) | def add_line_seg(self, junction1, junction2):
    method junctions (line 401) | def junctions(self):
    method line_segs (line 406) | def line_segs(self):
    method longest_line_segs (line 412) | def longest_line_segs(self):
    method adj_mtx (line 417) | def adj_mtx(self):
    method line_map (line 426) | def line_map(self, size, scale_x=1., scale_y=1., line_width=2.):
    method num_junctions (line 443) | def num_junctions(self):
    method num_line_segs (line 447) | def num_line_segs(self):

FILE: data/sist_line.py
  class SISTLine (line 11) | class SISTLine(data.Dataset):
    method __init__ (line 12) | def __init__(self, data_root, transforms, phase="train", sigma_junctio...
    method __getitem__ (line 20) | def __getitem__(self, item):
    method __call__ (line 73) | def __call__(self, item):
    method __len__ (line 76) | def __len__(self):
  function readnsave (line 88) | def readnsave(i):
  function juncsave (line 93) | def juncsave(i):

FILE: data/transforms.py
  class Compose (line 8) | class Compose(object):
    method __init__ (line 9) | def __init__(self, *transforms):
    method __call__ (line 12) | def __call__(self, img, pt):
  class RandomCompose (line 19) | class RandomCompose(object):
    method __init__ (line 20) | def __init__(self, *transforms):
    method __call__ (line 23) | def __call__(self, img, pt):
  class Resize (line 31) | class Resize(object):
    method __init__ (line 32) | def __init__(self, size, interpolation=Image.BILINEAR):
    method __call__ (line 35) | def __call__(self, img, pt):
  class RandomHorizontalFlip (line 49) | class RandomHorizontalFlip(object):
    method __init__ (line 50) | def __init__(self, p=0.5):
    method __call__ (line 53) | def __call__(self, img, pt):
  class RandomColorAug (line 64) | class RandomColorAug(object):
    method __init__ (line 65) | def __init__(self, factor=0.2):
    method __call__ (line 68) | def __call__(self, img, pt):

FILE: data/utils.py
  function apply_gaussian (line 6) | def apply_gaussian(accumulate_confid_map, centers, xx, yy, sigma):
  function gen_gaussian_map (line 17) | def gen_gaussian_map(centers, shape, sigma):

FILE: data/york_urban.py
  class YorkUrban (line 11) | class YorkUrban(data.Dataset):
    method __init__ (line 12) | def __init__(self, data_root, transforms, phase="test", sigma_junction...
    method __getitem__ (line 22) | def __getitem__(self, item):
    method __call__ (line 70) | def __call__(self, item):
    method __len__ (line 73) | def __len__(self):

FILE: main.py
  function weight_fn (line 25) | def weight_fn(dist_map, max_dist, mid=0.1, scale=10):
  class LSDTrainer (line 33) | class LSDTrainer(object):
    method __init__ (line 34) | def __init__(
    method _group_weight (line 151) | def _group_weight(module, lr):
    method end (line 177) | def end(self):
    method _train_epoch (line 181) | def _train_epoch(self):
    method _vis_train (line 286) | def _vis_train(self, epoch, i, len_loader, img, heatmap, adj_mtx, junc...
    method _checkpoint (line 317) | def _checkpoint(self):
    method train (line 336) | def train(
    method _vis_eval (line 365) | def _vis_eval(self, epoch, i, len_loader, img, heatmap, adj_mtx, junct...
    method eval (line 404) | def eval(self,

FILE: models/backbone.py
  class Bottleneck (line 23) | class Bottleneck(nn.Module):
    method __init__ (line 26) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 39) | def forward(self, x):
  class ResNet (line 62) | class ResNet(nn.Module):
    method __init__ (line 64) | def __init__(self, block, layers, num_classes=1000):
    method _make_layer (line 93) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 110) | def forward(self, x):
  class ResnetDilated (line 128) | class ResnetDilated(nn.Module):
    method __init__ (line 129) | def __init__(self, orig_resnet, dilate_scale=8):
    method _nostride_dilate (line 159) | def _nostride_dilate(m, dilate):
    method forward (line 174) | def forward(self, x):
  function resnet50 (line 194) | def resnet50(pretrained=False, **kwargs):
  function load_url (line 206) | def load_url(url, model_dir='./pretrained', map_location=None):
  class UPerNet (line 217) | class UPerNet(nn.Module):
    method __init__ (line 218) | def __init__(self, num_class=150, fc_dim=4096, pool_scales=(1, 2, 3, 6),
    method forward (line 259) | def forward(self, conv_out, segSize=None):
  class ResNetU50Backbone (line 300) | class ResNetU50Backbone(nn.Module):
    method __init__ (line 301) | def __init__(self, dim_embedding=256, encoder_weights="", decoder_weig...
    method forward (line 306) | def forward(self, img):
    method build_encoder (line 313) | def build_encoder(weights=''):
    method build_decoder (line 327) | def build_decoder(weights='', dim_embedding=64):
  class UNetBackbone (line 346) | class UNetBackbone(nn.Module):
    method __init__ (line 347) | def __init__(self, dim_embedding=256, n_downs=5, n_ups=3, weights=""):
    method forward (line 379) | def forward(self, x):

FILE: models/common.py
  class LMFPeakFinder (line 13) | class LMFPeakFinder(object):
    method __init__ (line 21) | def __init__(self, min_dist=5., min_th=0.3):
    method detect (line 25) | def detect(self, image):
  function conv3x3 (line 75) | def conv3x3(in_planes, out_planes, stride=1):
  function conv3x3_bn_relu (line 81) | def conv3x3_bn_relu(in_planes, out_planes, stride=1):
  class double_conv (line 89) | class double_conv(nn.Module):
    method __init__ (line 92) | def __init__(self, in_ch, out_ch):
    method forward (line 103) | def forward(self, x):
  class inconv (line 108) | class inconv(nn.Module):
    method __init__ (line 109) | def __init__(self, in_ch, out_ch):
    method forward (line 113) | def forward(self, x):
  class down (line 118) | class down(nn.Module):
    method __init__ (line 119) | def __init__(self, in_ch, out_ch):
    method forward (line 126) | def forward(self, x):
  class up (line 131) | class up(nn.Module):
    method __init__ (line 132) | def __init__(self, in_ch, out_ch, bilinear=True):
    method forward (line 144) | def forward(self, x1, x2):
  class outconv (line 155) | class outconv(nn.Module):
    method __init__ (line 156) | def __init__(self, in_ch, out_ch):
    method forward (line 160) | def forward(self, x):
  function weights_init (line 165) | def weights_init(m):
  function roi_pooling (line 174) | def roi_pooling(input, rois, size=(7, 7), spatial_scale=1.0):
  class GradAccumulatorFunction (line 204) | class GradAccumulatorFunction(Function):
    method forward (line 206) | def forward(ctx, input, accumulated_grad=None, mode="release"):
    method backward (line 213) | def backward(ctx, grad_output):
  class GradAccumulator (line 230) | class GradAccumulator(nn.Module):
    method __init__ (line 238) | def __init__(self, criterion_fns, submodules, collect_fn=None, reduce_...
    method forward (line 260) | def forward(self, tensor):

FILE: models/graph.py
  class JunctionInference (line 7) | class JunctionInference(nn.Module):
    method __init__ (line 8) | def __init__(self, dim_embedding, pooling_threshold=0.2, max_junctions...
    method forward (line 23) | def forward(self, feat):
  class LinePooling (line 60) | class LinePooling(nn.Module):
    method __init__ (line 61) | def __init__(self, align_size=256, spatial_scale=0.25):
    method forward (line 67) | def forward(self, feat, coord_st, coord_ed):
  class AdjacencyMatrixInference (line 93) | class AdjacencyMatrixInference(nn.Module):
    method __init__ (line 94) | def __init__(self, dim_embedding=256, align_size=256):
    method forward (line 114) | def forward(self, line_feat):

FILE: models/lsd.py
  class LSDDataLayer (line 10) | class LSDDataLayer(nn.Module):
    method __init__ (line 11) | def __init__(self, mean=None, std=None):
    method forward (line 16) | def forward(self, img):
  class BinaryFocalLoss (line 25) | class BinaryFocalLoss(nn.Module):
    method __init__ (line 26) | def __init__(self, gamma=2., alpha=0.25, size_average=True):
    method forward (line 32) | def forward(self, input, target, weight=None):
  class BlockAdjacencyMatrixInference (line 51) | class BlockAdjacencyMatrixInference(nn.Module):
    method __init__ (line 52) | def __init__(self,
    method forward (line 66) | def forward(self, feat):
  class BlockAdjacencyMatrixInferenceCriterion (line 76) | class BlockAdjacencyMatrixInferenceCriterion(nn.Module):
    method __init__ (line 77) | def __init__(self, adj_matrix_crit, adj_matrix_gt, adj_matrix_loss_lam...
    method forward (line 94) | def forward(self, block_adj_matrix):
  class LSDModule (line 109) | class LSDModule(nn.Module):
    method __init__ (line 110) | def __init__(
    method forward (line 195) | def forward(self, img, junc_map_gt, adj_matrix_gt, junc_loss_lambda=1....

FILE: models/lsd_test.py
  class LSDDataLayer (line 11) | class LSDDataLayer(nn.Module):
    method __init__ (line 12) | def __init__(self, mean=None, std=None):
    method forward (line 17) | def forward(self, img):
  class BinaryFocalLoss (line 26) | class BinaryFocalLoss(nn.Module):
    method __init__ (line 27) | def __init__(self, gamma=2., alpha=0.25, size_average=True):
    method forward (line 33) | def forward(self, input, target, weight=None):
  class BlockAdjacencyMatrixInference (line 52) | class BlockAdjacencyMatrixInference(nn.Module):
    method __init__ (line 53) | def __init__(self,
    method forward (line 67) | def forward(self, feat):
  class BlockAdjacencyMatrixInferenceCriterion (line 77) | class BlockAdjacencyMatrixInferenceCriterion(nn.Module):
    method __init__ (line 78) | def __init__(self, adj_matrix_crit, adj_matrix_gt, adj_matrix_loss_lam...
    method forward (line 95) | def forward(self, block_adj_matrix):
  class LSDTestModule (line 110) | class LSDTestModule(LSDModule):
    method __init__ (line 111) | def __init__(
    method forward (line 146) | def forward(self, img):

FILE: models/test_common.py
  class TestRoiPooling (line 10) | class TestRoiPooling(TestCase):
    method setUp (line 11) | def setUp(self):
    method test_output_size_7x7 (line 30) | def test_output_size_7x7(self):
    method test_output_size_5x5 (line 43) | def test_output_size_5x5(self):
    method test_output_size_1x1 (line 54) | def test_output_size_1x1(self):
    method test_output_value (line 64) | def test_output_value(self):
  class TestGradAccumulator (line 96) | class TestGradAccumulator(TestCase):
    method setUp (line 97) | def setUp(self):
    method test_forward_pass (line 118) | def test_forward_pass(self):
    method test_backward_pass (line 169) | def test_backward_pass(self):

FILE: models/test_graph.py
  class TestJunctionInference (line 7) | class TestJunctionInference(TestCase):
    method test_junction_inference_forward (line 8) | def test_junction_inference_forward(self):
  class TestJunctionPooling (line 19) | class TestJunctionPooling(TestCase):
    method test_junc_pooling_forward (line 20) | def test_junc_pooling_forward(self):
  class TestDirectionalAttention (line 32) | class TestDirectionalAttention(TestCase):
    method test_attention_forward (line 33) | def test_attention_forward(self):
  class TestAdjacencyMatrixInference (line 62) | class TestAdjacencyMatrixInference(TestCase):
    method test_adjacency_matrix_inference_forward (line 63) | def test_adjacency_matrix_inference_forward(self):

FILE: test.py
  class LSD (line 26) | class LSD(object):
    method __init__ (line 27) | def __init__(
    method end (line 87) | def end(self):
    method test (line 91) | def test(self, path_to_image):

FILE: utils.py
  function graph2line (line 9) | def graph2line(junctions, adj_mtx, threshold=0.5):
  function draw_lines (line 32) | def draw_lines(imgs, lines, scores=None, width=2):
  function draw_jucntions (line 57) | def draw_jucntions(hms, junctions):
  class AverageMeter (line 86) | class AverageMeter(object):
    method __init__ (line 89) | def __init__(self):
    method initialize (line 96) | def initialize(self, val, weight):
    method update (line 103) | def update(self, val, weight=1):
    method add (line 109) | def add(self, val, weight):
    method value (line 115) | def value(self):
    method average (line 118) | def average(self):
  function unique (line 122) | def unique(ar, return_index=False, return_inverse=False, return_counts=F...
  function colorEncode (line 165) | def colorEncode(labelmap, colors, mode='BGR'):
  function accuracy (line 182) | def accuracy(preds, label):
  function intersectionAndUnion (line 190) | def intersectionAndUnion(imPred, imLab, numClass):
  class NotSupportedCliException (line 213) | class NotSupportedCliException(Exception):
  function process_range (line 217) | def process_range(xpu, inp):
  function parse_devices (line 234) | def parse_devices(input_devices):