SYMBOL INDEX (447 symbols across 52 files) FILE: Face_Detection/align_warp_back_multiple_dlib.py function calculate_cdf (line 26) | def calculate_cdf(histogram): function calculate_lookup (line 42) | def calculate_lookup(src_cdf, ref_cdf): function match_histograms (line 62) | def match_histograms(src_image, ref_image): function _standard_face_pts (line 112) | def _standard_face_pts(): function _origin_face_pts (line 121) | def _origin_face_pts(): function compute_transformation_matrix (line 127) | def compute_transformation_matrix(img, landmark, normalize, target_face_... function compute_inverse_transformation_matrix (line 148) | def compute_inverse_transformation_matrix(img, landmark, normalize, targ... function show_detection (line 169) | def show_detection(image, box, landmark): function affine2theta (line 185) | def affine2theta(affine, input_w, input_h, target_w, target_h): function blur_blending (line 198) | def blur_blending(im1, im2, mask): function blur_blending_cv2 (line 217) | def blur_blending_cv2(im1, im2, mask): function Poisson_blending (line 239) | def Poisson_blending(im1, im2, mask): function Poisson_B (line 259) | def Poisson_B(im1, im2, mask, center): function seamless_clone (line 270) | def seamless_clone(old_face, new_face, raw_mask): function get_landmark (line 309) | def get_landmark(face_landmarks, id): function search (line 317) | def search(face_landmarks): FILE: Face_Detection/align_warp_back_multiple_dlib_HR.py function calculate_cdf (line 26) | def calculate_cdf(histogram): function calculate_lookup (line 42) | def calculate_lookup(src_cdf, ref_cdf): function match_histograms (line 62) | def match_histograms(src_image, ref_image): function _standard_face_pts (line 112) | def _standard_face_pts(): function _origin_face_pts (line 121) | def _origin_face_pts(): function compute_transformation_matrix (line 127) | def compute_transformation_matrix(img, landmark, normalize, target_face_... function compute_inverse_transformation_matrix (line 148) | def compute_inverse_transformation_matrix(img, landmark, normalize, targ... function show_detection (line 169) | def show_detection(image, box, landmark): function affine2theta (line 185) | def affine2theta(affine, input_w, input_h, target_w, target_h): function blur_blending (line 198) | def blur_blending(im1, im2, mask): function blur_blending_cv2 (line 217) | def blur_blending_cv2(im1, im2, mask): function Poisson_blending (line 239) | def Poisson_blending(im1, im2, mask): function Poisson_B (line 259) | def Poisson_B(im1, im2, mask, center): function seamless_clone (line 270) | def seamless_clone(old_face, new_face, raw_mask): function get_landmark (line 309) | def get_landmark(face_landmarks, id): function search (line 317) | def search(face_landmarks): FILE: Face_Detection/detect_all_dlib.py function _standard_face_pts (line 27) | def _standard_face_pts(): function _origin_face_pts (line 36) | def _origin_face_pts(): function get_landmark (line 42) | def get_landmark(face_landmarks, id): function search (line 50) | def search(face_landmarks): function compute_transformation_matrix (line 80) | def compute_transformation_matrix(img, landmark, normalize, target_face_... function show_detection (line 101) | def show_detection(image, box, landmark): function affine2theta (line 117) | def affine2theta(affine, input_w, input_h, target_w, target_h): FILE: Face_Detection/detect_all_dlib_HR.py function _standard_face_pts (line 27) | def _standard_face_pts(): function _origin_face_pts (line 36) | def _origin_face_pts(): function get_landmark (line 42) | def get_landmark(face_landmarks, id): function search (line 50) | def search(face_landmarks): function compute_transformation_matrix (line 80) | def compute_transformation_matrix(img, landmark, normalize, target_face_... function show_detection (line 101) | def show_detection(image, box, landmark): function affine2theta (line 117) | def affine2theta(affine, input_w, input_h, target_w, target_h): FILE: Face_Enhancement/data/__init__.py function create_dataloader (line 10) | def create_dataloader(opt): FILE: Face_Enhancement/data/base_dataset.py class BaseDataset (line 11) | class BaseDataset(data.Dataset): method __init__ (line 12) | def __init__(self): method modify_commandline_options (line 16) | def modify_commandline_options(parser, is_train): method initialize (line 19) | def initialize(self, opt): function get_params (line 23) | def get_params(opt, size): function get_transform (line 45) | def get_transform(opt, params, method=Image.BICUBIC, normalize=True, toT... function normalize (line 78) | def normalize(): function __resize (line 82) | def __resize(img, w, h, method=Image.BICUBIC): function __make_power_2 (line 86) | def __make_power_2(img, base, method=Image.BICUBIC): function __scale_width (line 95) | def __scale_width(img, target_width, method=Image.BICUBIC): function __scale_shortside (line 104) | def __scale_shortside(img, target_width, method=Image.BICUBIC): function __crop (line 115) | def __crop(img, pos, size): function __flip (line 122) | def __flip(img, flip): FILE: Face_Enhancement/data/custom_dataset.py class CustomDataset (line 8) | class CustomDataset(Pix2pixDataset): method modify_commandline_options (line 15) | def modify_commandline_options(parser, is_train): method get_paths (line 39) | def get_paths(self, opt): FILE: Face_Enhancement/data/face_dataset.py class FaceTestDataset (line 11) | class FaceTestDataset(BaseDataset): method modify_commandline_options (line 13) | def modify_commandline_options(parser, is_train): method initialize (line 23) | def initialize(self, opt): method __getitem__ (line 60) | def __getitem__(self, index): method __len__ (line 100) | def __len__(self): FILE: Face_Enhancement/data/image_folder.py function is_image_file (line 24) | def is_image_file(filename): function make_dataset_rec (line 28) | def make_dataset_rec(dir, images): function make_dataset (line 38) | def make_dataset(dir, recursive=False, read_cache=False, write_cache=Fal... function default_loader (line 69) | def default_loader(path): class ImageFolder (line 73) | class ImageFolder(data.Dataset): method __init__ (line 74) | def __init__(self, root, transform=None, return_paths=False, loader=de... method __getitem__ (line 90) | def __getitem__(self, index): method __len__ (line 100) | def __len__(self): FILE: Face_Enhancement/data/pix2pix_dataset.py class Pix2pixDataset (line 10) | class Pix2pixDataset(BaseDataset): method modify_commandline_options (line 12) | def modify_commandline_options(parser, is_train): method initialize (line 20) | def initialize(self, opt): method get_paths (line 48) | def get_paths(self, opt): method paths_match (line 55) | def paths_match(self, path1, path2): method __getitem__ (line 60) | def __getitem__(self, index): method postprocess (line 104) | def postprocess(self, input_dict): method __len__ (line 107) | def __len__(self): FILE: Face_Enhancement/models/__init__.py function find_model_using_name (line 8) | def find_model_using_name(model_name): function get_option_setter (line 34) | def get_option_setter(model_name): function create_model (line 39) | def create_model(opt): FILE: Face_Enhancement/models/networks/__init__.py function find_network_using_name (line 11) | def find_network_using_name(target_network_name, filename): function modify_commandline_options (line 21) | def modify_commandline_options(parser, is_train): function create_network (line 35) | def create_network(cls, opt): function define_G (line 45) | def define_G(opt): function define_D (line 50) | def define_D(opt): function define_E (line 55) | def define_E(opt): FILE: Face_Enhancement/models/networks/architecture.py class SPADEResnetBlock (line 19) | class SPADEResnetBlock(nn.Module): method __init__ (line 20) | def __init__(self, fin, fout, opt): method forward (line 49) | def forward(self, x, seg, degraded_image): method shortcut (line 59) | def shortcut(self, x, seg, degraded_image): method actvn (line 66) | def actvn(self, x): class ResnetBlock (line 72) | class ResnetBlock(nn.Module): method __init__ (line 73) | def __init__(self, dim, norm_layer, activation=nn.ReLU(False), kernel_... method forward (line 85) | def forward(self, x): class VGG19 (line 92) | class VGG19(torch.nn.Module): method __init__ (line 93) | def __init__(self, requires_grad=False): method forward (line 115) | def forward(self, X): class SPADEResnetBlock_non_spade (line 125) | class SPADEResnetBlock_non_spade(nn.Module): method __init__ (line 126) | def __init__(self, fin, fout, opt): method forward (line 155) | def forward(self, x, seg, degraded_image): method shortcut (line 165) | def shortcut(self, x, seg, degraded_image): method actvn (line 172) | def actvn(self, x): FILE: Face_Enhancement/models/networks/base_network.py class BaseNetwork (line 8) | class BaseNetwork(nn.Module): method __init__ (line 9) | def __init__(self): method modify_commandline_options (line 13) | def modify_commandline_options(parser, is_train): method print_network (line 16) | def print_network(self): method init_weights (line 27) | def init_weights(self, init_type="normal", gain=0.02): FILE: Face_Enhancement/models/networks/encoder.py class ConvEncoder (line 11) | class ConvEncoder(BaseNetwork): method __init__ (line 14) | def __init__(self, opt): method forward (line 36) | def forward(self, x): FILE: Face_Enhancement/models/networks/generator.py class SPADEGenerator (line 14) | class SPADEGenerator(BaseNetwork): method modify_commandline_options (line 16) | def modify_commandline_options(parser, is_train): method __init__ (line 27) | def __init__(self, opt): method compute_latent_vector_size (line 90) | def compute_latent_vector_size(self, opt): method forward (line 105) | def forward(self, input, degraded_image, z=None): class Pix2PixHDGenerator (line 151) | class Pix2PixHDGenerator(BaseNetwork): method modify_commandline_options (line 153) | def modify_commandline_options(parser, is_train): method __init__ (line 172) | def __init__(self, opt): method forward (line 231) | def forward(self, input, degraded_image, z=None): FILE: Face_Enhancement/models/networks/normalization.py function get_nonspade_norm_layer (line 12) | def get_nonspade_norm_layer(opt, norm_type="instance"): class SPADE (line 49) | class SPADE(nn.Module): method __init__ (line 50) | def __init__(self, config_text, norm_nc, label_nc, opt): method forward (line 81) | def forward(self, x, segmap, degraded_image): FILE: Face_Enhancement/models/pix2pix_model.py class Pix2PixModel (line 9) | class Pix2PixModel(torch.nn.Module): method modify_commandline_options (line 11) | def modify_commandline_options(parser, is_train): method __init__ (line 15) | def __init__(self, opt): method forward (line 36) | def forward(self, data, mode): method create_optimizers (line 55) | def create_optimizers(self, opt): method save (line 73) | def save(self, epoch): method initialize_networks (line 83) | def initialize_networks(self, opt): method preprocess_input (line 101) | def preprocess_input(self, data): method compute_generator_loss (line 127) | def compute_generator_loss(self, input_semantics, degraded_image, real... method compute_discriminator_loss (line 157) | def compute_discriminator_loss(self, input_semantics, degraded_image, ... method encode_z (line 171) | def encode_z(self, real_image): method generate_fake (line 176) | def generate_fake(self, input_semantics, degraded_image, real_image, c... method discriminate (line 195) | def discriminate(self, input_semantics, fake_image, real_image): method divide_pred (line 217) | def divide_pred(self, pred): method get_edges (line 232) | def get_edges(self, t): method reparameterize (line 240) | def reparameterize(self, mu, logvar): method use_gpu (line 245) | def use_gpu(self): FILE: Face_Enhancement/options/base_options.py class BaseOptions (line 14) | class BaseOptions: method __init__ (line 15) | def __init__(self): method initialize (line 18) | def initialize(self, parser): method gather_options (line 185) | def gather_options(self): method print_options (line 215) | def print_options(self, opt): method option_file_path (line 227) | def option_file_path(self, opt, makedir=False): method save_options (line 234) | def save_options(self, opt): method update_options_from_file (line 247) | def update_options_from_file(self, parser, opt): method load_options (line 255) | def load_options(self, opt): method parse (line 260) | def parse(self, save=False): FILE: Face_Enhancement/options/test_options.py class TestOptions (line 7) | class TestOptions(BaseOptions): method initialize (line 8) | def initialize(self, parser): FILE: Face_Enhancement/util/iter_counter.py class IterationCounter (line 10) | class IterationCounter: method __init__ (line 11) | def __init__(self, opt, dataset_size): method training_epochs (line 33) | def training_epochs(self): method record_epoch_start (line 36) | def record_epoch_start(self, epoch): method record_one_iteration (line 42) | def record_one_iteration(self): method record_epoch_end (line 52) | def record_epoch_end(self): method record_current_iter (line 63) | def record_current_iter(self): method needs_saving (line 67) | def needs_saving(self): method needs_printing (line 70) | def needs_printing(self): method needs_displaying (line 73) | def needs_displaying(self): FILE: Face_Enhancement/util/util.py function save_obj (line 15) | def save_obj(obj, name): function load_obj (line 20) | def load_obj(name): function copyconf (line 25) | def copyconf(default_opt, **kwargs): function tensor2im (line 35) | def tensor2im(image_tensor, imtype=np.uint8, normalize=True, tile=False): function tensor2label (line 67) | def tensor2label(label_tensor, n_label, imtype=np.uint8, tile=False): function save_image (line 97) | def save_image(image_numpy, image_path, create_dir=False): function mkdirs (line 110) | def mkdirs(paths): function mkdir (line 118) | def mkdir(path): function atoi (line 123) | def atoi(text): function natural_keys (line 127) | def natural_keys(text): function natural_sort (line 136) | def natural_sort(items): function str2bool (line 140) | def str2bool(v): function find_class_in_module (line 149) | def find_class_in_module(target_cls_name, module): function save_network (line 167) | def save_network(net, label, epoch, opt): function load_network (line 175) | def load_network(net, label, epoch, opt): function uint82bin (line 190) | def uint82bin(n, count=8): class Colorize (line 195) | class Colorize(object): method __init__ (line 196) | def __init__(self, n=35): method __call__ (line 200) | def __call__(self, gray_image): FILE: Face_Enhancement/util/visualizer.py class Visualizer (line 20) | class Visualizer: method __init__ (line 21) | def __init__(self, opt): method display_current_results (line 49) | def display_current_results(self, visuals, epoch, step): method plot_current_errors (line 72) | def plot_current_errors(self, errors, step): method print_current_errors (line 93) | def print_current_errors(self, epoch, i, errors, t): method convert_visuals_to_numpy (line 103) | def convert_visuals_to_numpy(self, visuals): method save_images (line 114) | def save_images(self, webpage, visuals, image_path): FILE: GUI.py function modify (line 11) | def modify(image_filename=None, cv2_frame=None): FILE: Global/data/Create_Bigfile.py function is_image_file (line 14) | def is_image_file(filename): function make_dataset (line 18) | def make_dataset(dir): FILE: Global/data/Load_Bigfile.py class BigFileMemoryLoader (line 9) | class BigFileMemoryLoader(object): method __load_bigfile (line 10) | def __load_bigfile(self): method __init__ (line 27) | def __init__(self, file_path): method __getitem__ (line 32) | def __getitem__(self, index): method __len__ (line 41) | def __len__(self): FILE: Global/data/base_data_loader.py class BaseDataLoader (line 4) | class BaseDataLoader(): method __init__ (line 5) | def __init__(self): method initialize (line 8) | def initialize(self, opt): method load_data (line 12) | def load_data(): FILE: Global/data/base_dataset.py class BaseDataset (line 10) | class BaseDataset(data.Dataset): method __init__ (line 11) | def __init__(self): method name (line 14) | def name(self): method initialize (line 17) | def initialize(self, opt): function get_params (line 20) | def get_params(opt, size): function get_transform (line 46) | def get_transform(opt, params, method=Image.BICUBIC, normalize=True): function normalize (line 84) | def normalize(): function __make_power_2 (line 87) | def __make_power_2(img, base, method=Image.BICUBIC): function __scale_width (line 95) | def __scale_width(img, target_width, method=Image.BICUBIC): function __crop (line 103) | def __crop(img, pos, size): function __flip (line 111) | def __flip(img, flip): FILE: Global/data/custom_dataset_data_loader.py function CreateDataset (line 10) | def CreateDataset(opt): class CustomDatasetDataLoader (line 23) | class CustomDatasetDataLoader(BaseDataLoader): method name (line 24) | def name(self): method initialize (line 27) | def initialize(self, opt): method load_data (line 37) | def load_data(self): method __len__ (line 40) | def __len__(self): FILE: Global/data/data_loader.py function CreateDataLoader (line 4) | def CreateDataLoader(opt): FILE: Global/data/image_folder.py function is_image_file (line 14) | def is_image_file(filename): function make_dataset (line 18) | def make_dataset(dir): function default_loader (line 31) | def default_loader(path): class ImageFolder (line 35) | class ImageFolder(data.Dataset): method __init__ (line 37) | def __init__(self, root, transform=None, return_paths=False, method __getitem__ (line 51) | def __getitem__(self, index): method __len__ (line 61) | def __len__(self): FILE: Global/data/online_dataset_for_old_photos.py function pil_to_np (line 17) | def pil_to_np(img_PIL): function np_to_pil (line 32) | def np_to_pil(img_np): function synthesize_salt_pepper (line 46) | def synthesize_salt_pepper(image,amount,salt_vs_pepper): function synthesize_gaussian (line 67) | def synthesize_gaussian(image,std_l,std_r): function synthesize_speckle (line 81) | def synthesize_speckle(image,std_l,std_r): function synthesize_low_resolution (line 96) | def synthesize_low_resolution(img): function convertToJpeg (line 112) | def convertToJpeg(im,quality): function blur_image_v2 (line 119) | def blur_image_v2(img): function online_add_degradation_v2 (line 132) | def online_add_degradation_v2(img): function irregular_hole_synthesize (line 156) | def irregular_hole_synthesize(img,mask): function zero_mask (line 168) | def zero_mask(size): class UnPairOldPhotos_SR (line 175) | class UnPairOldPhotos_SR(BaseDataset): ## Synthetic + Real Old method initialize (line 176) | def initialize(self, opt): method __getitem__ (line 213) | def __getitem__(self, index): method __len__ (line 278) | def __len__(self): method name (line 281) | def name(self): class PairOldPhotos (line 285) | class PairOldPhotos(BaseDataset): method initialize (line 286) | def initialize(self, opt): method __getitem__ (line 313) | def __getitem__(self, index): method __len__ (line 370) | def __len__(self): method name (line 377) | def name(self): class PairOldPhotos_with_hole (line 381) | class PairOldPhotos_with_hole(BaseDataset): method initialize (line 382) | def initialize(self, opt): method __getitem__ (line 411) | def __getitem__(self, index): method __len__ (line 476) | def __len__(self): method name (line 484) | def name(self): FILE: Global/detection.py function data_transforms (line 25) | def data_transforms(img, full_size, method=Image.BICUBIC): function scale_tensor (line 51) | def scale_tensor(img_tensor, default_scale=256): function blend_mask (line 66) | def blend_mask(img, mask): function main (line 73) | def main(config): FILE: Global/detection_models/antialiasing.py class Downsample (line 11) | class Downsample(nn.Module): method __init__ (line 14) | def __init__(self, pad_type="reflect", filt_size=3, stride=2, channels... method forward (line 51) | def forward(self, inp): function get_pad_layer (line 61) | def get_pad_layer(pad_type): FILE: Global/detection_models/networks.py class UNet (line 11) | class UNet(nn.Module): method __init__ (line 12) | def __init__( method forward (line 109) | def forward(self, x): class UNetConvBlock (line 124) | class UNetConvBlock(nn.Module): method __init__ (line 125) | def __init__(self, conv_num, in_size, out_size, padding, batch_norm): method forward (line 139) | def forward(self, x): class UNetUpBlock (line 144) | class UNetUpBlock(nn.Module): method __init__ (line 145) | def __init__(self, conv_num, in_size, out_size, up_mode, padding, batc... method center_crop (line 158) | def center_crop(self, layer, target_size): method forward (line 164) | def forward(self, x, bridge): class UnetGenerator (line 173) | class UnetGenerator(nn.Module): method __init__ (line 176) | def __init__(self, input_nc, output_nc, num_downs, ngf=64, norm_type="... method forward (line 223) | def forward(self, input): class UnetSkipConnectionBlock (line 227) | class UnetSkipConnectionBlock(nn.Module): method __init__ (line 234) | def __init__( method forward (line 291) | def forward(self, x): FILE: Global/detection_util/util.py function print_options (line 25) | def print_options(config_dict): function save_options (line 32) | def save_options(config_dict): function config_parse (line 46) | def config_parse(config_file, options, save=True): function to_np (line 70) | def to_np(x): function prepare_device (line 74) | def prepare_device(use_gpu, gpu_ids): function get_dir_size (line 94) | def get_dir_size(start_path="."): function mkdir_if_not (line 103) | def mkdir_if_not(dir_path): class Timer (line 109) | class Timer: method __init__ (line 110) | def __init__(self, msg): method __enter__ (line 114) | def __enter__(self): method __exit__ (line 117) | def __exit__(self, exc_type, exc_value, exc_tb): function get_size (line 123) | def get_size(start_path="."): function clean_tensorboard (line 132) | def clean_tensorboard(directory): function prepare_tensorboard (line 146) | def prepare_tensorboard(config, experiment_name=datetime.now().strftime(... function tb_loss_logger (line 159) | def tb_loss_logger(tb_writer, iter_index, loss_logger): function tb_image_logger (line 164) | def tb_image_logger(tb_writer, iter_index, images_info, config): function tb_image_logger_test (line 179) | def tb_image_logger_test(epoch, iter, images_info, config): function imshow (line 199) | def imshow(input_image, title=None, to_numpy=False): function vgg_preprocess (line 216) | def vgg_preprocess(tensor): function torch_vgg_preprocess (line 228) | def torch_vgg_preprocess(tensor): function network_gradient (line 238) | def network_gradient(net, gradient_on=True): FILE: Global/models/NonLocal_feature_mapping_model.py class Mapping_Model_with_mask (line 17) | class Mapping_Model_with_mask(nn.Module): method __init__ (line 18) | def __init__(self, nc, mc=64, n_blocks=3, norm="instance", padding_typ... method forward (line 71) | def forward(self, input, mask): class Mapping_Model_with_mask_2 (line 81) | class Mapping_Model_with_mask_2(nn.Module): ## Multi-Scale Patch Attention method __init__ (line 82) | def __init__(self, nc, mc=64, n_blocks=3, norm="instance", padding_typ... method forward (line 177) | def forward(self, input, mask): method inference_forward (line 187) | def inference_forward(self, input, mask): FILE: Global/models/base_model.py class BaseModel (line 9) | class BaseModel(torch.nn.Module): method name (line 10) | def name(self): method initialize (line 13) | def initialize(self, opt): method set_input (line 20) | def set_input(self, input): method forward (line 23) | def forward(self): method test (line 27) | def test(self): method get_image_paths (line 30) | def get_image_paths(self): method optimize_parameters (line 33) | def optimize_parameters(self): method get_current_visuals (line 36) | def get_current_visuals(self): method get_current_errors (line 39) | def get_current_errors(self): method save (line 42) | def save(self, label): method save_network (line 46) | def save_network(self, network, network_label, epoch_label, gpu_ids): method save_optimizer (line 53) | def save_optimizer(self, optimizer, optimizer_label, epoch_label): method load_optimizer (line 58) | def load_optimizer(self, optimizer, optimizer_label, epoch_label, save... method load_network (line 70) | def load_network(self, network, network_label, epoch_label, save_dir=""): method update_learning_rate (line 121) | def update_learning_rate(): FILE: Global/models/mapping_model.py class Mapping_Model (line 18) | class Mapping_Model(nn.Module): method __init__ (line 19) | def __init__(self, nc, mc=64, n_blocks=3, norm="instance", padding_typ... method forward (line 56) | def forward(self, input): class Pix2PixHDModel_Mapping (line 60) | class Pix2PixHDModel_Mapping(BaseModel): method name (line 61) | def name(self): method init_loss_filter (line 64) | def init_loss_filter(self, use_gan_feat_loss, use_vgg_loss, use_smooth... method initialize (line 78) | def initialize(self, opt): method encode_input (line 215) | def encode_input(self, label_map, inst_map=None, real_image=None, feat... method discriminate (line 240) | def discriminate(self, input_label, test_image, use_pool=False): method forward (line 248) | def forward(self, label, inst, image, feat, pair=True, infer=False, la... method inference (line 325) | def inference(self, label, inst): class InferenceModel (line 349) | class InferenceModel(Pix2PixHDModel_Mapping): method forward (line 350) | def forward(self, label, inst): FILE: Global/models/models.py function create_model (line 7) | def create_model(opt): function create_da_model (line 29) | def create_da_model(opt): FILE: Global/models/networks.py function weights_init (line 17) | def weights_init(m): function get_norm_layer (line 26) | def get_norm_layer(norm_type="instance"): function print_network (line 40) | def print_network(net): function define_G (line 50) | def define_G(input_nc, output_nc, ngf, netG, k_size=3, n_downsample_glob... function define_D (line 70) | def define_D(input_nc, ndf, n_layers_D, opt, norm='instance', use_sigmoi... class GlobalGenerator_DCDCv2 (line 82) | class GlobalGenerator_DCDCv2(nn.Module): method __init__ (line 83) | def __init__( method forward (line 283) | def forward(self, input, flow="enc_dec"): class ResnetBlock (line 295) | class ResnetBlock(nn.Module): method __init__ (line 296) | def __init__( method build_conv_block (line 304) | def build_conv_block(self, dim, padding_type, norm_layer, activation, ... method forward (line 337) | def forward(self, x): class Encoder (line 342) | class Encoder(nn.Module): method __init__ (line 343) | def __init__(self, input_nc, output_nc, ngf=32, n_downsampling=4, norm... method forward (line 376) | def forward(self, input, inst): function SN (line 394) | def SN(module, mode=True): class NonLocalBlock2D_with_mask_Res (line 401) | class NonLocalBlock2D_with_mask_Res(nn.Module): method __init__ (line 402) | def __init__( method forward (line 460) | def forward(self, x, mask): ## The shape of mask is Batch*1*H*W class MultiscaleDiscriminator (line 526) | class MultiscaleDiscriminator(nn.Module): method __init__ (line 527) | def __init__(self, input_nc, opt, ndf=64, n_layers=3, norm_layer=nn.Ba... method singleD_forward (line 544) | def singleD_forward(self, model, input): method forward (line 553) | def forward(self, input): class NLayerDiscriminator (line 568) | class NLayerDiscriminator(nn.Module): method __init__ (line 569) | def __init__(self, input_nc, opt, ndf=64, n_layers=3, norm_layer=nn.Ba... method forward (line 609) | def forward(self, input): class Patch_Attention_4 (line 621) | class Patch_Attention_4(nn.Module): ## While combine the feature map, u... method __init__ (line 622) | def __init__(self, in_channels, inter_channels, patch_size): method Hard_Compose (line 666) | def Hard_Compose(self, input, dim, index): method forward (line 678) | def forward(self, z, mask): ## The shape of mask is Batch*1*H*W method inference_forward (line 720) | def inference_forward(self,z,mask): ## Reduce the extra memory cost class GANLoss (line 781) | class GANLoss(nn.Module): method __init__ (line 782) | def __init__(self, use_lsgan=True, target_real_label=1.0, target_fake_... method get_target_tensor (line 795) | def get_target_tensor(self, input, target_is_real): method __call__ (line 813) | def __call__(self, input, target_is_real): class VGG19_torch (line 831) | class VGG19_torch(torch.nn.Module): method __init__ (line 832) | def __init__(self, requires_grad=False): method forward (line 854) | def forward(self, X): class VGGLoss_torch (line 863) | class VGGLoss_torch(nn.Module): method __init__ (line 864) | def __init__(self, gpu_ids): method forward (line 870) | def forward(self, x, y): FILE: Global/models/pix2pixHD_model.py class Pix2PixHDModel (line 12) | class Pix2PixHDModel(BaseModel): method name (line 13) | def name(self): method init_loss_filter (line 16) | def init_loss_filter(self, use_gan_feat_loss, use_vgg_loss,use_smooth_... method initialize (line 22) | def initialize(self, opt): method encode_input (line 112) | def encode_input(self, label_map, inst_map=None, real_image=None, feat... method discriminate (line 145) | def discriminate(self, input_label, test_image, use_pool=False): method forward (line 156) | def forward(self, label, inst, image, feat, infer=False): method inference (line 221) | def inference(self, label, inst, image=None, feat=None): method sample_features (line 245) | def sample_features(self, inst): method encode_features (line 266) | def encode_features(self, image, inst): method get_edges (line 288) | def get_edges(self, t): method save (line 299) | def save(self, which_epoch): method update_fixed_params (line 309) | def update_fixed_params(self): method update_learning_rate (line 318) | def update_learning_rate(self): class InferenceModel (line 330) | class InferenceModel(Pix2PixHDModel): method forward (line 331) | def forward(self, inp): FILE: Global/models/pix2pixHD_model_DA.py class Pix2PixHDModel (line 13) | class Pix2PixHDModel(BaseModel): method name (line 14) | def name(self): method init_loss_filter (line 17) | def init_loss_filter(self, use_gan_feat_loss, use_vgg_loss): method initialize (line 25) | def initialize(self, opt): method encode_input (line 118) | def encode_input(self, label_map, inst_map=None, real_image=None, feat... method discriminate (line 151) | def discriminate(self, input_label, test_image, use_pool=False): method feat_discriminate (line 162) | def feat_discriminate(self,input): method forward (line 167) | def forward(self, label, inst, image, feat, infer=False): method inference (line 256) | def inference(self, label, inst, image=None, feat=None): method sample_features (line 280) | def sample_features(self, inst): method encode_features (line 301) | def encode_features(self, image, inst): method get_edges (line 323) | def get_edges(self, t): method save (line 334) | def save(self, which_epoch): method update_fixed_params (line 346) | def update_fixed_params(self): method update_learning_rate (line 355) | def update_learning_rate(self): class InferenceModel (line 369) | class InferenceModel(Pix2PixHDModel): method forward (line 370) | def forward(self, inp): FILE: Global/options/base_options.py class BaseOptions (line 10) | class BaseOptions: method __init__ (line 11) | def __init__(self): method initialize (line 15) | def initialize(self): method parse (line 338) | def parse(self, save=True): FILE: Global/options/test_options.py class TestOptions (line 7) | class TestOptions(BaseOptions): method initialize (line 8) | def initialize(self): FILE: Global/options/train_options.py class TrainOptions (line 6) | class TrainOptions(BaseOptions): method initialize (line 7) | def initialize(self): FILE: Global/test.py function data_transforms (line 18) | def data_transforms(img, method=Image.BILINEAR, scale=False): function data_transforms_rgb_old (line 39) | def data_transforms_rgb_old(img): function irregular_hole_synthesize (line 47) | def irregular_hole_synthesize(img, mask): function parameter_set (line 59) | def parameter_set(opt): FILE: Global/util/image_pool.py class ImagePool (line 9) | class ImagePool: method __init__ (line 10) | def __init__(self, pool_size): method query (line 16) | def query(self, images): FILE: Global/util/util.py function tensor2im (line 14) | def tensor2im(image_tensor, imtype=np.uint8, normalize=True): function tensor2label (line 32) | def tensor2label(label_tensor, n_label, imtype=np.uint8): function save_image (line 43) | def save_image(image_numpy, image_path): function mkdirs (line 48) | def mkdirs(paths): function mkdir (line 56) | def mkdir(path): FILE: Global/util/visualizer.py class Visualizer (line 16) | class Visualizer(): method __init__ (line 17) | def __init__(self, opt): method display_current_results (line 40) | def display_current_results(self, visuals, epoch, step): method plot_current_errors (line 98) | def plot_current_errors(self, errors, step): method print_current_errors (line 105) | def print_current_errors(self, epoch, i, errors, t, lr): method print_save (line 116) | def print_save(self,message): method save_images (line 125) | def save_images(self, webpage, visuals, image_path): FILE: predict.py class Predictor (line 12) | class Predictor(cog.Predictor): method setup (line 13) | def setup(self): method predict (line 51) | def predict(self, image, HR=False, with_scratch=False): function clean_folder (line 213) | def clean_folder(folder): FILE: run.py function run_cmd (line 10) | def run_cmd(command):