SYMBOL INDEX (125 symbols across 12 files)

FILE: backbones.py
  function load_ref_wrn50 (line 5) | def load_ref_wrn50():
  function load (line 61) | def load(name):

FILE: common.py
  class _BaseMerger (line 10) | class _BaseMerger:
    method __init__ (line 11) | def __init__(self):
    method merge (line 14) | def merge(self, features: list):
  class AverageMerger (line 19) | class AverageMerger(_BaseMerger):
    method _reduce (line 21) | def _reduce(features):
  class ConcatMerger (line 28) | class ConcatMerger(_BaseMerger):
    method _reduce (line 30) | def _reduce(features):
  class Preprocessing (line 35) | class Preprocessing(torch.nn.Module):
    method __init__ (line 36) | def __init__(self, input_dims, output_dim):
    method forward (line 46) | def forward(self, features):
  class MeanMapper (line 53) | class MeanMapper(torch.nn.Module):
    method __init__ (line 54) | def __init__(self, preprocessing_dim):
    method forward (line 58) | def forward(self, features):
  class Aggregator (line 63) | class Aggregator(torch.nn.Module):
    method __init__ (line 64) | def __init__(self, target_dim):
    method forward (line 68) | def forward(self, features):
  class RescaleSegmentor (line 76) | class RescaleSegmentor:
    method __init__ (line 77) | def __init__(self, device, target_size=224):
    method convert_to_segmentation (line 82) | def convert_to_segmentation(self, patch_scores, features):
  class NetworkFeatureAggregator (line 124) | class NetworkFeatureAggregator(torch.nn.Module):
    method __init__ (line 127) | def __init__(self, backbone, layers_to_extract_from, device, train_bac...
    method forward (line 173) | def forward(self, images, eval=True):
    method feature_dimensions (line 187) | def feature_dimensions(self, input_shape):
  class ForwardHook (line 194) | class ForwardHook:
    method __init__ (line 195) | def __init__(self, hook_dict, layer_name: str, last_layer_to_extract: ...
    method __call__ (line 202) | def __call__(self, module, input, output):
  class LastLayerToExtractReachedException (line 209) | class LastLayerToExtractReachedException(Exception):

FILE: datasets/btad.py
  class DatasetSplit (line 18) | class DatasetSplit(Enum):
  class BTADDataset (line 24) | class BTADDataset(torch.utils.data.Dataset):
    method __init__ (line 29) | def __init__(
    method __getitem__ (line 98) | def __getitem__(self, idx):
    method __len__ (line 119) | def __len__(self):
    method get_image_data (line 122) | def get_image_data(self):

FILE: datasets/cifar10.py
  class DatasetSplit (line 13) | class DatasetSplit(Enum):
  class Cifar10Dataset (line 19) | class Cifar10Dataset(torch.utils.data.Dataset):
    method __init__ (line 26) | def __init__(
    method __getitem__ (line 95) | def __getitem__(self, idx):
    method __len__ (line 110) | def __len__(self):
    method get_image_data (line 113) | def get_image_data(self):

FILE: datasets/mvtec.py
  class DatasetSplit (line 30) | class DatasetSplit(Enum):
  class MVTecDataset (line 36) | class MVTecDataset(torch.utils.data.Dataset):
    method __init__ (line 41) | def __init__(
    method __getitem__ (line 110) | def __getitem__(self, idx):
    method __len__ (line 131) | def __len__(self):
    method get_image_data (line 134) | def get_image_data(self):

FILE: datasets/sdd.py
  class DatasetSplit (line 15) | class DatasetSplit(Enum):
  class SDDDataset (line 21) | class SDDDataset(torch.utils.data.Dataset):
    method __init__ (line 26) | def __init__(
    method __getitem__ (line 99) | def __getitem__(self, idx):
    method __len__ (line 119) | def __len__(self):
    method get_image_data (line 122) | def get_image_data(self):

FILE: datasets/sdd2.py
  class DatasetSplit (line 15) | class DatasetSplit(Enum):
  class SDD2Dataset (line 21) | class SDD2Dataset(torch.utils.data.Dataset):
    method __init__ (line 26) | def __init__(
    method __getitem__ (line 98) | def __getitem__(self, idx):
    method __len__ (line 118) | def __len__(self):
    method get_image_data (line 121) | def get_image_data(self):

FILE: main.py
  function main (line 39) | def main(**kwargs):
  function run (line 44) | def run(
  function net (line 150) | def net(
  function dataset (line 245) | def dataset(

FILE: metrics.py
  function compute_imagewise_retrieval_metrics (line 7) | def compute_imagewise_retrieval_metrics(
  function compute_pixelwise_retrieval_metrics (line 35) | def compute_pixelwise_retrieval_metrics(anomaly_segmentations, ground_tr...
  function compute_pro (line 88) | def compute_pro(masks, amaps, num_th=200):

FILE: resnet.py
  function conv3x3 (line 30) | def conv3x3(in_planes: int, out_planes: int, stride: int = 1, groups: in...
  function conv1x1 (line 36) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1):
  class BasicBlock (line 41) | class BasicBlock(nn.Module):
    method __init__ (line 44) | def __init__(
    method forward (line 71) | def forward(self, x: Tensor):
  class Bottleneck (line 90) | class Bottleneck(nn.Module):
    method __init__ (line 99) | def __init__(
    method forward (line 125) | def forward(self, x: Tensor):
  class ResNet (line 148) | class ResNet(nn.Module):
    method __init__ (line 150) | def __init__(
    method _make_layer (line 207) | def _make_layer(self, block: Type[Union[BasicBlock, Bottleneck]], plan...
    method _forward_impl (line 231) | def _forward_impl(self, x: Tensor):
    method forward (line 248) | def forward(self, x: Tensor):
  function _resnet (line 252) | def _resnet(
  function resnet18 (line 268) | def resnet18(pretrained: bool = False, progress: bool = True, **kwargs: ...
  function resnet34 (line 280) | def resnet34(pretrained: bool = False, progress: bool = True, **kwargs: ...
  function resnet50 (line 292) | def resnet50(pretrained: bool = False, progress: bool = True, **kwargs: ...
  function resnet101 (line 304) | def resnet101(pretrained: bool = False, progress: bool = True, **kwargs:...
  function resnet152 (line 316) | def resnet152(pretrained: bool = False, progress: bool = True, **kwargs:...
  function resnext50_32x4d (line 328) | def resnext50_32x4d(pretrained: bool = False, progress: bool = True, **k...
  function resnext101_32x8d (line 342) | def resnext101_32x8d(pretrained: bool = False, progress: bool = True, **...
  function wide_resnet50_2 (line 356) | def wide_resnet50_2(pretrained: bool = False, progress: bool = True, **k...
  function wide_resnet101_2 (line 374) | def wide_resnet101_2(pretrained: bool = False, progress: bool = True, **...

FILE: simplenet.py
  function init_weight (line 27) | def init_weight(m):
  class Discriminator (line 35) | class Discriminator(torch.nn.Module):
    method __init__ (line 36) | def __init__(self, in_planes, n_layers=1, hidden=None):
    method forward (line 53) | def forward(self,x):
  class Projection (line 59) | class Projection(torch.nn.Module):
    method __init__ (line 61) | def __init__(self, in_planes, out_planes=None, n_layers=1, layer_type=0):
    method forward (line 83) | def forward(self, x):
  class TBWrapper (line 90) | class TBWrapper:
    method __init__ (line 92) | def __init__(self, log_dir):
    method step (line 96) | def step(self):
  class SimpleNet (line 99) | class SimpleNet(torch.nn.Module):
    method __init__ (line 100) | def __init__(self, device):
    method load (line 105) | def load(
    method set_model_dir (line 205) | def set_model_dir(self, model_dir, dataset_name):
    method embed (line 216) | def embed(self, data):
    method _embed (line 228) | def _embed(self, images, detach=True, provide_patch_shapes=False, eval...
    method test (line 289) | def test(self, training_data, test_data, save_segmentation_images):
    method _evaluate (line 344) | def _evaluate(self, test_data, scores, segmentations, features, labels...
    method train (line 389) | def train(self, training_data, test_data):
    method _train_discriminator (line 455) | def _train_discriminator(self, input_data):
    method predict (line 543) | def predict(self, data, prefix=""):
    method _predict_dataloader (line 548) | def _predict_dataloader(self, dataloader, prefix):
    method _predict (line 576) | def _predict(self, images):
    method _params_file (line 615) | def _params_file(filepath, prepend=""):
    method save_to_path (line 618) | def save_to_path(self, save_path: str, prepend: str = ""):
    method save_segmentation_images (line 640) | def save_segmentation_images(self, data, segmentations, scores):
  class PatchMaker (line 674) | class PatchMaker:
    method __init__ (line 675) | def __init__(self, patchsize, top_k=0, stride=None):
    method patchify (line 680) | def patchify(self, features, return_spatial_info=False):
    method unpatch_scores (line 708) | def unpatch_scores(self, x, batchsize):
    method score (line 711) | def score(self, x):

FILE: utils.py
  function plot_segmentation_images (line 15) | def plot_segmentation_images(
  function create_storage_folder (line 77) | def create_storage_folder(
  function set_torch_device (line 96) | def set_torch_device(gpu_ids):
  function fix_seeds (line 109) | def fix_seeds(seed, with_torch=True, with_cuda=True):
  function compute_and_store_final_results (line 127) | def compute_and_store_final_results(