config.yml

MODE: 1             # 1: train, 2: test, 3: eval
MODEL: 2            # 1: edge model, 2: inpaint model, 3: edge-inpaint model, 4: joint model
MASK: 3             # 1: random block, 2: half, 3: external, 4: (external, random block), 5: (external, random block, half)
EDGE: 1             # 1: canny, 2: external
NMS: 1              # 0: no non-max-suppression, 1: applies non-max-suppression on the external edges by multiplying by Canny
SEED: 42            # random seed
GPU: [0]            # list of gpu ids
DEBUG: 0            # turns on debugging mode
VERBOSE: 0          # turns on verbose mode in the output console

PATH: ./checkpoints/
TRAIN_EDGE: ./dataset/trainA/
TRAIN_IMG: ./dataset/trainB/
TEST_EDGE: ./dataset/testA/
TEST_IMG: ./dataset/testB/
matrics_path: ./checkpoints/metric.txt

LR: 0.0001                    # learning rate
D2G_LR: 0.1                   # discriminator/generator learning rate ratio
BETA1: 0.0                    # adam optimizer beta1
BETA2: 0.9                    # adam optimizer beta2
BATCH_SIZE: 8                # input batch size for training
INPUT_SIZE: 256              # input image size for training 0 for original size
SIGMA: 2                      # standard deviation of the Gaussian filter used in Canny edge detector (0: random, -1: no edge)
MAX_ITERS: 2e6                # maximum number of iterations to train the model
max_epoch : 1000
early_stopping : 5


EDGE_THRESHOLD: 0.5           # edge detection threshold
L1_LOSS_WEIGHT: 5            # l1 loss weight
FM_LOSS_WEIGHT: 10            # feature-matching loss weight
STYLE_LOSS_WEIGHT: 250        # style loss weight
CONTENT_LOSS_WEIGHT: 0.1      # perceptual loss weight
INPAINT_ADV_LOSS_WEIGHT: 0.5  # adversarial loss weight

GAN_LOSS: nsgan               # nsgan | lsgan | hinge
GAN_POOL_SIZE: 0              # fake images pool size

SAVE_INTERVAL: 1000          # how many iterations to wait before saving model (0: never)
SAMPLE_INTERVAL: 1000       # how many iterations to wait before sampling (0: never)
SAMPLE_SIZE: 4               # number of images to sample
EVAL_INTERVAL: 0              # how many iterations to wait before model evaluation (0: never)
LOG_INTERVAL: 10              # how many iterations to wait before logging training status (0: never)

scale_factor : 2
img_size_max : 256
img_size_min : 32
num_scale : 0
size_list : 0
stage : 0