SYMBOL INDEX (414 symbols across 37 files)

FILE: evaluation/eval-aph-post-wireframe.py
  function c (line 43) | def c(x):
  function imshow (line 47) | def imshow(im):
  function main (line 63) | def main():

FILE: evaluation/eval-aph-post-york.py
  function c (line 43) | def c(x):
  function imshow (line 47) | def imshow(im):
  function main (line 63) | def main():

FILE: evaluation/eval-aph-score-wireframe.py
  function main (line 40) | def main():

FILE: evaluation/eval-aph-score-york.py
  function main (line 40) | def main():

FILE: evaluation/eval-fscore-wireframe.py
  function f_score (line 35) | def f_score(tp, fp):
  function line_score (line 45) | def line_score(path, threshold=5):
  function work (line 87) | def work(path):

FILE: evaluation/eval-fscore-york.py
  function f_score (line 33) | def f_score(tp, fp):
  function line_score (line 43) | def line_score(path, threshold=5):
  function work (line 86) | def work(path):

FILE: evaluation/eval-sAP-wireframe.py
  function line_score (line 34) | def line_score(path, threshold=5):
  function work (line 79) | def work(path):

FILE: evaluation/eval-sAP-york.py
  function line_score (line 34) | def line_score(path, threshold=5):
  function work (line 80) | def work(path):

FILE: evaluation/lcnn/box.py
  class BoxError (line 50) | class BoxError(Exception):
  class BoxKeyError (line 54) | class BoxKeyError(BoxError, KeyError, AttributeError):
  function _to_json (line 61) | def _to_json(obj, filename=None,
  function _from_json (line 73) | def _from_json(json_string=None, filename=None,
  function _to_yaml (line 89) | def _to_yaml(obj, filename=None, default_flow_style=False,
  function _from_yaml (line 104) | def _from_yaml(yaml_string=None, filename=None,
  function _safe_key (line 121) | def _safe_key(key):
  function _safe_attr (line 128) | def _safe_attr(attr, camel_killer=False, replacement_char='x'):
  function _camel_killer (line 157) | def _camel_killer(attr):
  function _recursive_tuples (line 174) | def _recursive_tuples(iterable, box_class, recreate_tuples=False, **kwar...
  function _conversion_checks (line 187) | def _conversion_checks(item, keys, box_config, check_only=False,
  function _get_box_config (line 230) | def _get_box_config(cls, kwargs):
  class Box (line 250) | class Box(dict):
    method __new__ (line 271) | def __new__(cls, *args, **kwargs):
    method __init__ (line 280) | def __init__(self, *args, **kwargs):
    method __add_ordered (line 320) | def __add_ordered(self, key):
    method box_it_up (line 325) | def box_it_up(self):
    method __hash__ (line 337) | def __hash__(self):
    method __dir__ (line 345) | def __dir__(self):
    method get (line 382) | def get(self, key, default=None):
    method copy (line 392) | def copy(self):
    method __copy__ (line 395) | def __copy__(self):
    method __deepcopy__ (line 398) | def __deepcopy__(self, memodict=None):
    method __setstate__ (line 406) | def __setstate__(self, state):
    method __getitem__ (line 410) | def __getitem__(self, item, _ignore_default=False):
    method keys (line 423) | def keys(self):
    method values (line 428) | def values(self):
    method items (line 431) | def items(self):
    method __get_default (line 434) | def __get_default(self, item):
    method __box_config (line 445) | def __box_config(self):
    method __convert_and_store (line 452) | def __convert_and_store(self, item, value):
    method __create_lineage (line 481) | def __create_lineage(self):
    method __getattr__ (line 489) | def __getattr__(self, item):
    method __setitem__ (line 517) | def __setitem__(self, key, value):
    method __setattr__ (line 528) | def __setattr__(self, key, value):
    method __delitem__ (line 558) | def __delitem__(self, key):
    method __delattr__ (line 566) | def __delattr__(self, item):
    method pop (line 583) | def pop(self, key, *args):
    method clear (line 603) | def clear(self):
    method popitem (line 607) | def popitem(self):
    method __repr__ (line 614) | def __repr__(self):
    method __str__ (line 617) | def __str__(self):
    method __iter__ (line 620) | def __iter__(self):
    method __reversed__ (line 624) | def __reversed__(self):
    method to_dict (line 628) | def to_dict(self):
    method update (line 645) | def update(self, item=None, **kwargs):
    method setdefault (line 664) | def setdefault(self, item, default=None):
    method to_json (line 675) | def to_json(self, filename=None,
    method from_json (line 690) | def from_json(cls, json_string=None, filename=None,
    method to_yaml (line 717) | def to_yaml(self, filename=None, default_flow_style=False,
    method from_yaml (line 735) | def from_yaml(cls, yaml_string=None, filename=None,
  class BoxList (line 763) | class BoxList(list):
    method __init__ (line 769) | def __init__(self, iterable=None, box_class=Box, **box_options):
    method __delitem__ (line 784) | def __delitem__(self, key):
    method __setitem__ (line 789) | def __setitem__(self, key, value):
    method append (line 794) | def append(self, p_object):
    method extend (line 810) | def extend(self, iterable):
    method insert (line 814) | def insert(self, index, p_object):
    method __repr__ (line 822) | def __repr__(self):
    method __str__ (line 825) | def __str__(self):
    method __copy__ (line 828) | def __copy__(self):
    method __deepcopy__ (line 833) | def __deepcopy__(self, memodict=None):
    method __hash__ (line 841) | def __hash__(self):
    method to_list (line 848) | def to_list(self):
    method to_json (line 861) | def to_json(self, filename=None,
    method from_json (line 885) | def from_json(cls, json_string=None, filename=None, encoding="utf-8",
    method to_yaml (line 913) | def to_yaml(self, filename=None, default_flow_style=False,
    method from_yaml (line 931) | def from_yaml(cls, yaml_string=None, filename=None,
    method box_it_up (line 959) | def box_it_up(self):
  class ConfigBox (line 965) | class ConfigBox(Box):
    method __getattr__ (line 983) | def __getattr__(self, item):
    method __dir__ (line 991) | def __dir__(self):
    method bool (line 996) | def bool(self, item, default=None):
    method int (line 1019) | def int(self, item, default=None):
    method float (line 1034) | def float(self, item, default=None):
    method list (line 1049) | def list(self, item, default=None, spliter=",", strip=True, mod=None):
    method getboolean (line 1074) | def getboolean(self, item, default=None):
    method getint (line 1077) | def getint(self, item, default=None):
    method getfloat (line 1080) | def getfloat(self, item, default=None):
    method __repr__ (line 1083) | def __repr__(self):
  class SBox (line 1087) | class SBox(Box):
    method dict (line 1097) | def dict(self):
    method json (line 1101) | def json(self):
    method yaml (line 1106) | def yaml(self):
    method __repr__ (line 1109) | def __repr__(self):

FILE: evaluation/lcnn/datasets.py
  class WireframeDataset (line 17) | class WireframeDataset(Dataset):
    method __init__ (line 18) | def __init__(self, rootdir, split):
    method __len__ (line 27) | def __len__(self):
    method __getitem__ (line 30) | def __getitem__(self, idx):
    method adjacency_matrix (line 81) | def adjacency_matrix(self, n, link):
  function collate (line 90) | def collate(batch):

FILE: evaluation/lcnn/metric.py
  function ap (line 11) | def ap(tp, fp):
  function fscore (line 23) | def fscore(tp, fp):
  function APJ (line 32) | def APJ(vert_pred, vert_gt, max_distance, im_ids):
  function nms_j (line 68) | def nms_j(heatmap, delta=1):
  function mAPJ (line 82) | def mAPJ(pred, truth, distances, im_ids):
  function post_jheatmap (line 86) | def post_jheatmap(heatmap, offset=None, delta=1):
  function vectorized_wireframe_2d_metric (line 103) | def vectorized_wireframe_2d_metric(
  function vectorized_wireframe_3d_metric (line 155) | def vectorized_wireframe_3d_metric(
  function msTPFP (line 194) | def msTPFP(line_pred, line_gt, threshold):
  function msAP (line 213) | def msAP(line_pred, line_gt, threshold):

FILE: evaluation/lcnn/models/hourglass_pose.py
  class Bottleneck2D (line 14) | class Bottleneck2D(nn.Module):
    method __init__ (line 17) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 30) | def forward(self, x):
  class Hourglass (line 53) | class Hourglass(nn.Module):
    method __init__ (line 54) | def __init__(self, block, num_blocks, planes, depth):
    method _make_residual (line 60) | def _make_residual(self, block, num_blocks, planes):
    method _make_hour_glass (line 66) | def _make_hour_glass(self, block, num_blocks, planes, depth):
    method _hour_glass_forward (line 77) | def _hour_glass_forward(self, n, x):
    method forward (line 91) | def forward(self, x):
  class HourglassNet (line 95) | class HourglassNet(nn.Module):
    method __init__ (line 98) | def __init__(self, block, head, depth, num_stacks, num_blocks, num_cla...
    method _make_residual (line 137) | def _make_residual(self, block, planes, blocks, stride=1):
    method _make_fc (line 157) | def _make_fc(self, inplanes, outplanes):
    method forward (line 162) | def forward(self, x):
  function hg (line 192) | def hg(**kwargs):

FILE: evaluation/lcnn/models/line_vectorizer.py
  class LineVectorizer (line 15) | class LineVectorizer(nn.Module):
    method __init__ (line 16) | def __init__(self, backbone):
    method forward (line 45) | def forward(self, input_dict):
    method sample_lines (line 152) | def sample_lines(self, meta, jmap, joff, mode):
  function non_maximum_suppression (line 248) | def non_maximum_suppression(a):
  class Bottleneck1D (line 254) | class Bottleneck1D(nn.Module):
    method __init__ (line 255) | def __init__(self, inplanes, outplanes):
    method forward (line 271) | def forward(self, x):

FILE: evaluation/lcnn/models/multitask_learner.py
  class MultitaskHead (line 11) | class MultitaskHead(nn.Module):
    method __init__ (line 12) | def __init__(self, input_channels, num_class):
    method forward (line 28) | def forward(self, x):
  class MultitaskLearner (line 32) | class MultitaskLearner(nn.Module):
    method __init__ (line 33) | def __init__(self, backbone):
    method forward (line 40) | def forward(self, input_dict):
  function l2loss (line 93) | def l2loss(input, target):
  function cross_entropy_loss (line 97) | def cross_entropy_loss(logits, positive):
  function sigmoid_l1_loss (line 102) | def sigmoid_l1_loss(logits, target, offset=0.0, mask=None):

FILE: evaluation/lcnn/postprocess.py
  function pline (line 4) | def pline(x1, y1, x2, y2, x, y):
  function psegment (line 14) | def psegment(x1, y1, x2, y2, x, y):
  function plambda (line 24) | def plambda(x1, y1, x2, y2, x, y):
  function postprocess (line 31) | def postprocess(lines, scores, threshold=0.01, tol=1e9, do_clip=False):

FILE: evaluation/lcnn/trainer.py
  class Trainer (line 23) | class Trainer(object):
    method __init__ (line 24) | def __init__(self, device, model, optimizer, train_loader, val_loader,...
    method run_tensorboard (line 54) | def run_tensorboard(self):
    method _loss (line 69) | def _loss(self, result):
    method validate (line 101) | def validate(self):
    method train_epoch (line 163) | def train_epoch(self):
    method _write_metrics (line 205) | def _write_metrics(self, size, total_loss, prefix, do_print=False):
    method _plot_samples (line 228) | def _plot_samples(self, i, index, result, meta, target, prefix):
    method train (line 281) | def train(self):
  function c (line 299) | def c(x):
  function imshow (line 303) | def imshow(im):
  function tprint (line 312) | def tprint(*args):
  function pprint (line 318) | def pprint(*args):
  function _launch_tensorboard (line 324) | def _launch_tensorboard(board_out, port, out):

FILE: evaluation/lcnn/utils.py
  class benchmark (line 11) | class benchmark(object):
    method __init__ (line 12) | def __init__(self, msg, enable=True, fmt="%0.3g"):
    method __enter__ (line 17) | def __enter__(self):
    method __exit__ (line 22) | def __exit__(self, *args):
  function quiver (line 29) | def quiver(x, y, ax):
  function recursive_to (line 45) | def recursive_to(input, device):
  function np_softmax (line 60) | def np_softmax(x, axis=0):
  function argsort2d (line 66) | def argsort2d(arr):
  function __parallel_handle (line 70) | def __parallel_handle(f, q_in, q_out):
  function parmap (line 78) | def parmap(f, X, nprocs=multiprocessing.cpu_count(), progress_bar=lambda...

FILE: evaluation/process.py
  function main (line 44) | def main():
  function c (line 129) | def c(x):

FILE: helper/wireframe.py
  function main (line 25) | def main():
  function save_and_process (line 77) | def save_and_process(image_path, image_name, image, lines):

FILE: helper/wireframe_eval.py
  function inrange (line 35) | def inrange(v, shape):
  function to_int (line 39) | def to_int(x):
  function save_heatmap (line 43) | def save_heatmap(prefix, image, lines):
  function main (line 195) | def main():

FILE: helper/york.py
  function main (line 30) | def main():
  function save_and_process (line 87) | def save_and_process(image_path, image_name, image, lines):

FILE: helper/york_eval.py
  function inrange (line 38) | def inrange(v, shape):
  function to_int (line 42) | def to_int(x):
  function save_heatmap (line 46) | def save_heatmap(prefix, image, lines):
  function main (line 190) | def main():

FILE: src/args.py
  function get_args_parser (line 4) | def get_args_parser():

FILE: src/datasets/__init__.py
  function get_coco_api_from_dataset (line 8) | def get_coco_api_from_dataset(dataset):
  function build_dataset (line 18) | def build_dataset(image_set, args):

FILE: src/datasets/coco.py
  class CocoDetection (line 14) | class CocoDetection(torchvision.datasets.CocoDetection):
    method __init__ (line 15) | def __init__(self, img_folder, ann_file, transforms, args):
    method __getitem__ (line 21) | def __getitem__(self, idx):
  class ConvertCocoPolysToMask (line 30) | class ConvertCocoPolysToMask(object):
    method __call__ (line 32) | def __call__(self, image, target, args):
  function make_coco_transforms (line 73) | def make_coco_transforms(image_set, args):
  function build (line 116) | def build(image_set, args):

FILE: src/datasets/transforms.py
  function crop (line 21) | def crop(image, target, region):
  function hflip (line 79) | def hflip(image, target):
  function vflip (line 95) | def vflip(image, target):
  function ccw_rotation (line 114) | def ccw_rotation(image, target):
  function cw_rotation (line 138) | def cw_rotation(image, target):
  function resize (line 165) | def resize(image, target, size, max_size=None):
  function pad (line 215) | def pad(image, target, padding):
  class RandomCrop (line 229) | class RandomCrop(object):
    method __init__ (line 230) | def __init__(self, size):
    method __call__ (line 233) | def __call__(self, img, target):
  class RandomSizeCrop (line 238) | class RandomSizeCrop(object):
    method __init__ (line 239) | def __init__(self, min_size: int, max_size: int):
    method __call__ (line 243) | def __call__(self, img: PIL.Image.Image, target: dict):
  class CenterCrop (line 250) | class CenterCrop(object):
    method __init__ (line 251) | def __init__(self, size):
    method __call__ (line 254) | def __call__(self, img, target):
  class RandomHorizontalFlip (line 262) | class RandomHorizontalFlip(object):
    method __init__ (line 263) | def __init__(self, p=0.5):
    method __call__ (line 266) | def __call__(self, img, target):
  class RandomVerticalFlip (line 271) | class RandomVerticalFlip(object):
    method __init__ (line 272) | def __init__(self, p=0.5):
    method __call__ (line 275) | def __call__(self, img, target):
  class RandomCounterClockwiseRotation (line 280) | class RandomCounterClockwiseRotation(object):
    method __init__ (line 281) | def __init__(self, p=0.5):
    method __call__ (line 284) | def __call__(self, img, target):
  class RandomClockwiseRotation (line 289) | class RandomClockwiseRotation(object):
    method __init__ (line 290) | def __init__(self, p=0.5):
    method __call__ (line 293) | def __call__(self, img, target):
  class RandomResize (line 298) | class RandomResize(object):
    method __init__ (line 299) | def __init__(self, sizes, max_size=None):
    method __call__ (line 304) | def __call__(self, img, target=None):
  class RandomPad (line 309) | class RandomPad(object):
    method __init__ (line 310) | def __init__(self, max_pad):
    method __call__ (line 313) | def __call__(self, img, target):
  class RandomErasing (line 318) | class RandomErasing(object):
    method __init__ (line 319) | def __init__(self, p=0.5, scale=(0.02, 0.33), ratio=(0.3, 3.3), value=...
    method get_params (line 343) | def get_params(img: Tensor, scale: Tuple[float, float], ratio: Tuple[f...
    method __call__ (line 377) | def __call__(self, img, target):
  class ColorJitter (line 385) | class ColorJitter(object):
    method __init__ (line 386) | def __init__(self, brightness=0.4, contrast=0.4, saturation=0.4, hue=0...
    method _check_input (line 393) | def _check_input(self, value, name, center=1, bound=(0, float('inf')),...
    method __call__ (line 412) | def __call__(self, img, target):
  class RandomSelect (line 437) | class RandomSelect(object):
    method __init__ (line 438) | def __init__(self, transforms1, transforms2, p=0.5):
    method __call__ (line 443) | def __call__(self, img, target):
  class ToTensor (line 449) | class ToTensor(object):
    method __call__ (line 450) | def __call__(self, img, target):
  class Normalize (line 453) | class Normalize(object):
    method __init__ (line 454) | def __init__(self, mean, std):
    method __call__ (line 458) | def __call__(self, image, target=None):
  class Compose (line 473) | class Compose(object):
    method __init__ (line 474) | def __init__(self, transforms):
    method __call__ (line 477) | def __call__(self, image, target):
    method __repr__ (line 482) | def __repr__(self):

FILE: src/engine.py
  function train_one_epoch (line 18) | def train_one_epoch(model, criterion, postprocessors, data_loader, optim...
  function evaluate (line 77) | def evaluate(model, criterion, postprocessors, data_loader, base_ds, dev...

FILE: src/main.py
  function main (line 20) | def main(args):

FILE: src/models/__init__.py
  function build_model (line 5) | def build_model(args):

FILE: src/models/backbone.py
  class FrozenBatchNorm2d (line 19) | class FrozenBatchNorm2d(torch.nn.Module):
    method __init__ (line 28) | def __init__(self, n):
    method _load_from_state_dict (line 35) | def _load_from_state_dict(self, state_dict, prefix, local_metadata, st...
    method forward (line 45) | def forward(self, x):
  class BackboneBase (line 58) | class BackboneBase(nn.Module):
    method __init__ (line 60) | def __init__(self, backbone: nn.Module, train_backbone: bool, num_chan...
    method forward (line 72) | def forward(self, tensor_list: NestedTensor):
  class Backbone (line 84) | class Backbone(BackboneBase):
    method __init__ (line 86) | def __init__(self, name: str,
  class Joiner (line 97) | class Joiner(nn.Sequential):
    method __init__ (line 98) | def __init__(self, backbone, position_embedding):
    method forward (line 101) | def forward(self, tensor_list: NestedTensor):
  function build_backbone (line 113) | def build_backbone(args):

FILE: src/models/letr.py
  class LETR (line 19) | class LETR(nn.Module):
    method __init__ (line 21) | def __init__(self, backbone, transformer, num_classes, num_queries, ar...
    method forward (line 38) | def forward(self, samples, postprocessors=None, targets=None, criterio...
    method _set_aux_loss (line 57) | def _set_aux_loss(self, outputs_class, outputs_coord):
  class SetCriterion (line 60) | class SetCriterion(nn.Module):
    method __init__ (line 62) | def __init__(self, num_classes, weight_dict, eos_coef, losses, args, m...
    method loss_lines_labels (line 81) | def loss_lines_labels(self, outputs, targets,  num_items,  log=False, ...
    method label_focal_loss (line 104) | def label_focal_loss(self, input, target, weight, gamma=2.0):
    method loss_cardinality (line 124) | def loss_cardinality(self, outputs, targets,  num_items, origin_indice...
    method loss_lines_POST (line 137) | def loss_lines_POST(self, outputs, targets, num_items, origin_indices=...
    method loss_lines (line 158) | def loss_lines(self, outputs, targets, num_items, origin_indices=None):
    method _get_src_permutation_idx (line 173) | def _get_src_permutation_idx(self, indices):
    method _get_tgt_permutation_idx (line 179) | def _get_tgt_permutation_idx(self, indices):
    method get_loss (line 185) | def get_loss(self, loss, outputs, targets, num_items, **kwargs):
    method forward (line 197) | def forward(self, outputs, targets, origin_indices=None):
  class PostProcess_Line (line 245) | class PostProcess_Line(nn.Module):
    method forward (line 249) | def forward(self, outputs, target_sizes, output_type):
  class MLP (line 302) | class MLP(nn.Module):
    method __init__ (line 305) | def __init__(self, input_dim, hidden_dim, output_dim, num_layers):
    method forward (line 311) | def forward(self, x):
  function build (line 317) | def build(args):

FILE: src/models/letr_stack.py
  class LETRstack (line 18) | class LETRstack(nn.Module):
    method __init__ (line 19) | def __init__(self, letr, args):
    method forward (line 48) | def forward(self, samples, postprocessors=None, targets=None, criterio...
    method _set_aux_loss (line 83) | def _set_aux_loss(self, outputs_class, outputs_coord):
    method _set_aux_loss_POST (line 91) | def _set_aux_loss_POST(self, outputs_class, outputs_coord):
  function _expand (line 97) | def _expand(tensor, length: int):
  class MLP (line 100) | class MLP(nn.Module):
    method __init__ (line 103) | def __init__(self, input_dim, hidden_dim, output_dim, num_layers):
    method forward (line 109) | def forward(self, x):
  class Transformer (line 115) | class Transformer(nn.Module):
    method __init__ (line 117) | def __init__(self, d_model=512, nhead=8, num_encoder_layers=6,
    method _reset_parameters (line 137) | def _reset_parameters(self):
    method forward (line 142) | def forward(self, src, mask, query_embed, pos_embed):
  class TransformerEncoder (line 155) | class TransformerEncoder(nn.Module):
    method __init__ (line 157) | def __init__(self, encoder_layer, num_layers, norm=None):
    method forward (line 163) | def forward(self, src,
  class TransformerDecoder (line 178) | class TransformerDecoder(nn.Module):
    method __init__ (line 180) | def __init__(self, decoder_layer, num_layers, norm=None, return_interm...
    method forward (line 187) | def forward(self, tgt, memory,
  class TransformerEncoderLayer (line 219) | class TransformerEncoderLayer(nn.Module):
    method __init__ (line 221) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
    method with_pos_embed (line 238) | def with_pos_embed(self, tensor, pos: Optional[Tensor]):
    method forward_post (line 241) | def forward_post(self,
    method forward_pre (line 256) | def forward_pre(self, src,
    method forward (line 270) | def forward(self, src,
  class TransformerDecoderLayer (line 279) | class TransformerDecoderLayer(nn.Module):
    method __init__ (line 281) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
    method with_pos_embed (line 301) | def with_pos_embed(self, tensor, pos: Optional[Tensor]):
    method forward_post (line 304) | def forward_post(self, tgt, memory,
    method forward_pre (line 327) | def forward_pre(self, tgt, memory,
    method forward (line 350) | def forward(self, tgt, memory,
  function _get_clones (line 364) | def _get_clones(module, N):
  function _get_activation_fn (line 367) | def _get_activation_fn(activation):

FILE: src/models/matcher.py
  class HungarianMatcher_Line (line 8) | class HungarianMatcher_Line(nn.Module):
    method __init__ (line 16) | def __init__(self, cost_class: float = 1, cost_line: float = 1):
    method forward (line 29) | def forward(self, outputs, targets):
  function build_matcher (line 80) | def build_matcher(args, type=None):

FILE: src/models/multi_head_attention.py
  function dropout (line 22) | def dropout(input, p=0.5, training=True, inplace=False):
  function _get_softmax_dim (line 48) | def _get_softmax_dim(name, ndim, stacklevel):
  function softmax (line 58) | def softmax(input, dim=None, _stacklevel=3, dtype=None):
  function linear (line 90) | def linear(input, weight, bias=None):
  function multi_head_attention_forward (line 116) | def multi_head_attention_forward(query: Tensor, key: Tensor, value: Tens...
  class MultiheadAttention (line 381) | class MultiheadAttention(Module):
    method __init__ (line 412) | def __init__(self, embed_dim, num_heads, dropout=0., bias=True, add_bi...
    method _reset_parameters (line 451) | def _reset_parameters(self):
    method __setstate__ (line 467) | def __setstate__(self, state):
    method forward (line 474) | def forward(self, query, key, value, key_padding_mask=None,

FILE: src/models/position_encoding.py
  class PositionEmbeddingSine (line 12) | class PositionEmbeddingSine(nn.Module):
    method __init__ (line 17) | def __init__(self, num_pos_feats=64, temperature=10000, normalize=Fals...
    method forward (line 28) | def forward(self, tensor_list: NestedTensor):
  class PositionEmbeddingLearned (line 51) | class PositionEmbeddingLearned(nn.Module):
    method __init__ (line 55) | def __init__(self, num_pos_feats=256):
    method reset_parameters (line 61) | def reset_parameters(self):
    method forward (line 65) | def forward(self, tensor_list: NestedTensor):
  function build_position_encoding (line 79) | def build_position_encoding(args):

FILE: src/models/transformer.py
  class Transformer (line 18) | class Transformer(nn.Module):
    method __init__ (line 20) | def __init__(self, d_model=512, nhead=8, num_encoder_layers=6,
    method _reset_parameters (line 42) | def _reset_parameters(self):
    method forward (line 47) | def forward(self, src, mask, query_embed, pos_embed):
  class TransformerEncoder (line 64) | class TransformerEncoder(nn.Module):
    method __init__ (line 66) | def __init__(self, encoder_layer, num_layers, norm=None):
    method forward (line 72) | def forward(self, src,
  class TransformerDecoder (line 87) | class TransformerDecoder(nn.Module):
    method __init__ (line 89) | def __init__(self, decoder_layer, num_layers, norm=None, return_interm...
    method forward (line 96) | def forward(self, tgt, memory,
  class TransformerEncoderLayer (line 128) | class TransformerEncoderLayer(nn.Module):
    method __init__ (line 130) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
    method with_pos_embed (line 146) | def with_pos_embed(self, tensor, pos: Optional[Tensor]):
    method forward_post (line 149) | def forward_post(self,
    method forward_pre (line 164) | def forward_pre(self, src,
    method forward (line 178) | def forward(self, src,
  class TransformerDecoderLayer (line 187) | class TransformerDecoderLayer(nn.Module):
    method __init__ (line 189) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
    method with_pos_embed (line 209) | def with_pos_embed(self, tensor, pos: Optional[Tensor]):
    method forward_post (line 212) | def forward_post(self, tgt, memory,
    method forward_pre (line 235) | def forward_pre(self, tgt, memory,
    method forward (line 258) | def forward(self, tgt, memory,
  function _get_clones (line 272) | def _get_clones(module, N):
  function build_transformer (line 276) | def build_transformer(args):
  function _get_activation_fn (line 289) | def _get_activation_fn(activation):

FILE: src/util/misc.py
  class SmoothedValue (line 26) | class SmoothedValue(object):
    method __init__ (line 31) | def __init__(self, window_size=20, fmt=None):
    method update (line 39) | def update(self, value, n=1):
    method synchronize_between_processes (line 44) | def synchronize_between_processes(self):
    method median (line 58) | def median(self):
    method avg (line 63) | def avg(self):
    method global_avg (line 68) | def global_avg(self):
    method max (line 72) | def max(self):
    method value (line 76) | def value(self):
    method __str__ (line 79) | def __str__(self):
  function all_gather (line 88) | def all_gather(data):
  function reduce_dict (line 131) | def reduce_dict(input_dict, average=True):
  class MetricLogger (line 158) | class MetricLogger(object):
    method __init__ (line 159) | def __init__(self, delimiter="\t"):
    method update (line 163) | def update(self, **kwargs):
    method __getattr__ (line 170) | def __getattr__(self, attr):
    method __str__ (line 178) | def __str__(self):
    method synchronize_between_processes (line 186) | def synchronize_between_processes(self):
    method add_meter (line 190) | def add_meter(self, name, meter):
    method log_every (line 193) | def log_every(self, iterable, print_freq, header=None):
  function get_sha (line 248) | def get_sha():
  function collate_fn (line 268) | def collate_fn(batch):
  function _max_by_axis (line 274) | def _max_by_axis(the_list):
  function nested_tensor_from_tensor_list (line 283) | def nested_tensor_from_tensor_list(tensor_list: List[Tensor]):
  function _onnx_nested_tensor_from_tensor_list (line 311) | def _onnx_nested_tensor_from_tensor_list(tensor_list):
  class NestedTensor (line 339) | class NestedTensor(object):
    method __init__ (line 340) | def __init__(self, tensors, mask: Optional[Tensor]):
    method to (line 344) | def to(self, device):
    method decompose (line 355) | def decompose(self):
    method __repr__ (line 358) | def __repr__(self):
  function setup_for_distributed (line 362) | def setup_for_distributed(is_master):
  function is_dist_avail_and_initialized (line 377) | def is_dist_avail_and_initialized():
  function get_world_size (line 385) | def get_world_size():
  function get_rank (line 391) | def get_rank():
  function is_main_process (line 397) | def is_main_process():
  function save_on_master (line 401) | def save_on_master(*args, **kwargs):
  function init_distributed_mode (line 406) | def init_distributed_mode(args):
  function accuracy (line 432) | def accuracy(output, target, topk=(1,)):
  function interpolate (line 450) | def interpolate(input, size=None, scale_factor=None, mode="nearest", ali...