File size: 1,515 Bytes
93871a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import os
class Options:
def __init__(self, checkpoints, gpu_ids):
self.serial_batches = True # no shuffle
self.no_flip = True # no flip
self.label_nc = 0
self.n_downsample_global = 3
self.mc = 64
self.k_size = 4
self.start_r = 1
self.mapping_n_block = 6
self.map_mc = 512
self.no_instance = True
self.checkpoints_dir = checkpoints
self.gpu_ids = gpu_ids
self.mapping_net_dilation = 1
self.use_segmentation_model = False
self.feat_dim = -1
self.spatio_size = 64
self.resize_or_crop = 'scale_width'
self.isTrain = False
self.input_nc = 3
self.output_nc = 3
self.ngf = 64
self.norm = "instance"
self.load_pretrain = ""
self.which_epoch = "latest"
self.load_pretrain = ""
self.no_load_VAE = False
self.use_vae_which_epoch = "latest"
self.NL_res = True
self.use_SN = True
self.correlation_renormalize = True
self.NL_use_mask = True
self.NL_fusion_method = "combine"
self.non_local = "Setting_42"
# self.name = "mapping_scratch"
self.load_pretrainA = os.path.join(self.checkpoints_dir, "VAE_A_quality")
self.load_pretrainB = os.path.join(self.checkpoints_dir, "VAE_B_scratch")
self.mapping_exp = 1
self.inference_optimize = True
self.mask_dilation = 3
self.name = "mapping_Patch_Attention"
|