SYMBOL INDEX (211 symbols across 29 files)

FILE: data/__init__.py
  function find_dataset_using_name (line 22) | def find_dataset_using_name(dataset_name):
  function get_option_setter (line 44) | def get_option_setter(dataset_name):
  function create_dataset (line 50) | def create_dataset(opt):
  class CustomDatasetDataLoader (line 65) | class CustomDatasetDataLoader:
    method __init__ (line 68) | def __init__(self, opt):
    method load_data (line 90) | def load_data(self):
    method __len__ (line 93) | def __len__(self):
    method __iter__ (line 97) | def __iter__(self):
    method set_epoch (line 104) | def set_epoch(self, epoch):

FILE: data/aligned_dataset.py
  class AlignedDataset (line 7) | class AlignedDataset(BaseDataset):
    method __init__ (line 14) | def __init__(self, opt):
    method __getitem__ (line 27) | def __getitem__(self, index):
    method __len__ (line 58) | def __len__(self):

FILE: data/base_dataset.py
  class BaseDataset (line 14) | class BaseDataset(data.Dataset, ABC):
    method __init__ (line 24) | def __init__(self, opt):
    method modify_commandline_options (line 34) | def modify_commandline_options(parser, is_train):
    method __len__ (line 47) | def __len__(self):
    method __getitem__ (line 52) | def __getitem__(self, index):
  function get_params (line 64) | def get_params(opt, size):
  function get_transform (line 82) | def get_transform(opt, params=None, grayscale=False, method=transforms.I...
  function __transforms2pil_resize (line 116) | def __transforms2pil_resize(method):
  function __make_power_2 (line 126) | def __make_power_2(img, base, method=transforms.InterpolationMode.BICUBIC):
  function __scale_width (line 138) | def __scale_width(img, target_size, crop_size, method=transforms.Interpo...
  function __crop (line 148) | def __crop(img, pos, size):
  function __flip (line 157) | def __flip(img, flip):
  function __print_size_warning (line 163) | def __print_size_warning(ow, oh, w, h):

FILE: data/colorization_dataset.py
  class ColorizationDataset (line 10) | class ColorizationDataset(BaseDataset):
    method modify_commandline_options (line 17) | def modify_commandline_options(parser, is_train):
    method __init__ (line 33) | def __init__(self, opt):
    method __getitem__ (line 45) | def __getitem__(self, index):
    method __len__ (line 67) | def __len__(self):

FILE: data/image_folder.py
  function is_image_file (line 29) | def is_image_file(filename):
  function make_dataset (line 33) | def make_dataset(dir, max_dataset_size=float("inf")):
  function default_loader (line 44) | def default_loader(path):
  class ImageFolder (line 48) | class ImageFolder(data.Dataset):
    method __init__ (line 50) | def __init__(self, root, transform=None, return_paths=False, loader=de...
    method __getitem__ (line 61) | def __getitem__(self, index):
    method __len__ (line 71) | def __len__(self):

FILE: data/single_dataset.py
  class SingleDataset (line 6) | class SingleDataset(BaseDataset):
    method __init__ (line 12) | def __init__(self, opt):
    method __getitem__ (line 23) | def __getitem__(self, index):
    method __len__ (line 38) | def __len__(self):

FILE: data/template_dataset.py
  class TemplateDataset (line 21) | class TemplateDataset(BaseDataset):
    method modify_commandline_options (line 25) | def modify_commandline_options(parser, is_train):
    method __init__ (line 39) | def __init__(self, opt):
    method __getitem__ (line 57) | def __getitem__(self, index):
    method __len__ (line 76) | def __len__(self):

FILE: data/unaligned_dataset.py
  class UnalignedDataset (line 8) | class UnalignedDataset(BaseDataset):
    method __init__ (line 19) | def __init__(self, opt):
    method __getitem__ (line 39) | def __getitem__(self, index):
    method __len__ (line 65) | def __len__(self):

FILE: models/__init__.py
  function find_model_using_name (line 25) | def find_model_using_name(model_name: str):
  function get_option_setter (line 47) | def get_option_setter(model_name: str):
  function create_model (line 53) | def create_model(opt):

FILE: models/base_model.py
  class BaseModel (line 10) | class BaseModel(ABC):
    method __init__ (line 20) | def __init__(self, opt):
    method modify_commandline_options (line 49) | def modify_commandline_options(parser, is_train):
    method set_input (line 62) | def set_input(self, input):
    method forward (line 71) | def forward(self):
    method optimize_parameters (line 76) | def optimize_parameters(self):
    method setup (line 80) | def setup(self, opt):
    method eval (line 132) | def eval(self):
    method test (line 139) | def test(self):
    method compute_visuals (line 149) | def compute_visuals(self):
    method get_image_paths (line 153) | def get_image_paths(self):
    method update_learning_rate (line 157) | def update_learning_rate(self):
    method get_current_visuals (line 169) | def get_current_visuals(self):
    method get_current_losses (line 177) | def get_current_losses(self):
    method save_networks (line 185) | def save_networks(self, epoch):
    method __patch_instance_norm_state_dict (line 209) | def __patch_instance_norm_state_dict(self, state_dict, module, keys, i...
    method load_networks (line 221) | def load_networks(self, epoch):
    method print_networks (line 248) | def print_networks(self, verbose):
    method set_requires_grad (line 266) | def set_requires_grad(self, nets, requires_grad=False):
    method init_networks (line 279) | def init_networks(self, init_type="normal", init_gain=0.02):

FILE: models/colorization_model.py
  class ColorizationModel (line 7) | class ColorizationModel(Pix2PixModel):
    method modify_commandline_options (line 16) | def modify_commandline_options(parser, is_train=True):
    method __init__ (line 33) | def __init__(self, opt):
    method lab2rgb (line 49) | def lab2rgb(self, L, AB):
    method compute_visuals (line 66) | def compute_visuals(self):

FILE: models/cycle_gan_model.py
  class CycleGANModel (line 8) | class CycleGANModel(BaseModel):
    method modify_commandline_options (line 21) | def modify_commandline_options(parser, is_train=True):
    method __init__ (line 53) | def __init__(self, opt):
    method set_input (line 101) | def set_input(self, input):
    method forward (line 114) | def forward(self):
    method backward_D_basic (line 121) | def backward_D_basic(self, netD, real, fake):
    method backward_D_A (line 143) | def backward_D_A(self):
    method backward_D_B (line 148) | def backward_D_B(self):
    method backward_G (line 153) | def backward_G(self):
    method optimize_parameters (line 182) | def optimize_parameters(self):

FILE: models/networks.py
  class Identity (line 13) | class Identity(nn.Module):
    method forward (line 14) | def forward(self, x):
  function get_norm_layer (line 18) | def get_norm_layer(norm_type="instance"):
  function get_scheduler (line 43) | def get_scheduler(optimizer, opt):
  function init_weights (line 74) | def init_weights(net, init_type="normal", init_gain=0.02):
  function init_net (line 109) | def init_net(net, init_type="normal", init_gain=0.02):
  function define_G (line 132) | def define_G(input_nc, output_nc, ngf, netG, norm="batch", use_dropout=F...
  function define_D (line 163) | def define_D(input_nc, ndf, netD, n_layers_D=3, norm="batch", init_type=...
  class GANLoss (line 209) | class GANLoss(nn.Module):
    method __init__ (line 216) | def __init__(self, gan_mode, target_real_label=1.0, target_fake_label=...
    method get_target_tensor (line 240) | def get_target_tensor(self, prediction, target_is_real):
    method __call__ (line 257) | def __call__(self, prediction, target_is_real):
  function cal_gradient_penalty (line 278) | def cal_gradient_penalty(netD, real_data, fake_data, device, type="mixed...
  class ResnetGenerator (line 313) | class ResnetGenerator(nn.Module):
    method __init__ (line 319) | def __init__(self, input_nc, output_nc, ngf=64, norm_layer=nn.BatchNor...
    method forward (line 359) | def forward(self, input):
  class ResnetBlock (line 364) | class ResnetBlock(nn.Module):
    method __init__ (line 367) | def __init__(self, dim, padding_type, norm_layer, use_dropout, use_bias):
    method build_conv_block (line 378) | def build_conv_block(self, dim, padding_type, norm_layer, use_dropout,...
    method forward (line 418) | def forward(self, x):
  class UnetGenerator (line 424) | class UnetGenerator(nn.Module):
    method __init__ (line 427) | def __init__(self, input_nc, output_nc, num_downs, ngf=64, norm_layer=...
    method forward (line 451) | def forward(self, input):
  class UnetSkipConnectionBlock (line 456) | class UnetSkipConnectionBlock(nn.Module):
    method __init__ (line 462) | def __init__(self, outer_nc, inner_nc, input_nc=None, submodule=None, ...
    method forward (line 511) | def forward(self, x):
  class NLayerDiscriminator (line 518) | class NLayerDiscriminator(nn.Module):
    method __init__ (line 521) | def __init__(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNo...
    method forward (line 553) | def forward(self, input):
  class PixelDiscriminator (line 558) | class PixelDiscriminator(nn.Module):
    method __init__ (line 561) | def __init__(self, input_nc, ndf=64, norm_layer=nn.BatchNorm2d):
    method forward (line 586) | def forward(self, input):

FILE: models/pix2pix_model.py
  class Pix2PixModel (line 6) | class Pix2PixModel(BaseModel):
    method modify_commandline_options (line 18) | def modify_commandline_options(parser, is_train=True):
    method __init__ (line 40) | def __init__(self, opt):
    method set_input (line 73) | def set_input(self, input):
    method forward (line 86) | def forward(self):
    method backward_D (line 90) | def backward_D(self):
    method backward_G (line 104) | def backward_G(self):
    method optimize_parameters (line 116) | def optimize_parameters(self):

FILE: models/template_model.py
  class TemplateModel (line 24) | class TemplateModel(BaseModel):
    method modify_commandline_options (line 26) | def modify_commandline_options(parser, is_train=True):
    method __init__ (line 42) | def __init__(self, opt):
    method set_input (line 73) | def set_input(self, input):
    method forward (line 84) | def forward(self):
    method backward (line 88) | def backward(self):
    method optimize_parameters (line 95) | def optimize_parameters(self):

FILE: models/test_model.py
  class TestModel (line 5) | class TestModel(BaseModel):
    method modify_commandline_options (line 13) | def modify_commandline_options(parser, is_train=True):
    method __init__ (line 32) | def __init__(self, opt):
    method set_input (line 52) | def set_input(self, input):
    method forward (line 63) | def forward(self):
    method optimize_parameters (line 67) | def optimize_parameters(self):

FILE: options/base_options.py
  class BaseOptions (line 9) | class BaseOptions:
    method __init__ (line 16) | def __init__(self):
    method initialize (line 20) | def initialize(self, parser):
    method gather_options (line 62) | def gather_options(self):
    method print_options (line 90) | def print_options(self, opt):
    method parse (line 115) | def parse(self):

FILE: options/test_options.py
  class TestOptions (line 4) | class TestOptions(BaseOptions):
    method initialize (line 10) | def initialize(self, parser):

FILE: options/train_options.py
  class TrainOptions (line 4) | class TrainOptions(BaseOptions):
    method initialize (line 10) | def initialize(self, parser):

FILE: scripts/edges/batch_hed.py
  function parse_args (line 21) | def parse_args():

FILE: scripts/eval_cityscapes/cityscapes.py
  class cityscapes (line 9) | class cityscapes:
    method __init__ (line 10) | def __init__(self, data_path):
    method get_dset (line 24) | def get_dset(self, split):
    method load_image (line 37) | def load_image(self, split, city, idx):
    method assign_trainIds (line 41) | def assign_trainIds(self, label):
    method load_label (line 55) | def load_label(self, split, city, idx):
    method preprocess (line 66) | def preprocess(self, im):
    method palette (line 80) | def palette(self, label):
    method make_boundaries (line 95) | def make_boundaries(label, thickness=None):
    method list_label_frames (line 110) | def list_label_frames(self, split):
    method collect_frame_sequence (line 126) | def collect_frame_sequence(self, split, idx, length):

FILE: scripts/eval_cityscapes/evaluate.py
  function main (line 21) | def main():

FILE: scripts/eval_cityscapes/util.py
  function get_out_scoremap (line 5) | def get_out_scoremap(net):
  function feed_net (line 9) | def feed_net(net, in_):
  function segrun (line 17) | def segrun(net, in_):
  function fast_hist (line 23) | def fast_hist(a, b, n):
  function get_scores (line 32) | def get_scores(hist):

FILE: scripts/test_before_push.py
  class TestBeforePush (line 7) | class TestBeforePush:
    method setup_datasets (line 11) | def setup_datasets(self):
    method test_pretrained_cyclegan_model (line 22) | def test_pretrained_cyclegan_model(self):
    method test_pretrained_pix2pix_model (line 34) | def test_pretrained_pix2pix_model(self):
    method test_cyclegan_train_test (line 49) | def test_cyclegan_train_test(self):
    method test_pix2pix_train_test (line 68) | def test_pix2pix_train_test(self):
    method test_template_train_test (line 87) | def test_template_train_test(self):
    method test_colorization_train_test (line 106) | def test_colorization_train_test(self):

FILE: util/get_data.py
  class GetData (line 10) | class GetData(object):
    method __init__ (line 26) | def __init__(self, technique="cyclegan", verbose=True):
    method _print (line 34) | def _print(self, text):
    method _get_options (line 39) | def _get_options(r):
    method _present_options (line 44) | def _present_options(self):
    method _download_data (line 53) | def _download_data(self, dataset_url, save_path):
    method get (line 77) | def get(self, save_path, dataset=None):

FILE: util/html.py
  class HTML (line 6) | class HTML:
    method __init__ (line 14) | def __init__(self, web_dir, title, refresh=0):
    method get_image_dir (line 34) | def get_image_dir(self):
    method add_header (line 38) | def add_header(self, text):
    method add_images (line 47) | def add_images(self, ims, txts, links, width=400):
    method save (line 67) | def save(self):

FILE: util/image_pool.py
  class ImagePool (line 5) | class ImagePool:
    method __init__ (line 12) | def __init__(self, pool_size):
    method query (line 23) | def query(self, images):

FILE: util/util.py
  function tensor2im (line 12) | def tensor2im(input_image, imtype=np.uint8):
  function diagnose_network (line 33) | def diagnose_network(net, name="network"):
  function init_ddp (line 53) | def init_ddp():
  function cleanup_ddp (line 73) | def cleanup_ddp():
  function save_image (line 78) | def save_image(image_numpy, image_path, aspect_ratio=1.0):
  function print_numpy (line 96) | def print_numpy(x, val=True, shp=False):
  function mkdirs (line 111) | def mkdirs(paths):
  function mkdir (line 124) | def mkdir(path):

FILE: util/visualizer.py
  function save_images (line 12) | def save_images(webpage, visuals, image_path, aspect_ratio=1.0, width=256):
  class Visualizer (line 40) | class Visualizer:
    method __init__ (line 46) | def __init__(self, opt):
    method reset (line 85) | def reset(self):
    method set_dataset_size (line 89) | def set_dataset_size(self, dataset_size):
    method _calculate_global_step (line 93) | def _calculate_global_step(self, epoch, epoch_iter):
    method display_current_results (line 98) | def display_current_results(self, visuals, epoch: int, total_iters: in...
    method plot_current_losses (line 134) | def plot_current_losses(self, total_iters, losses):
    method print_current_losses (line 148) | def print_current_losses(self, epoch, iters, losses, t_comp, t_data):