|
|
--- |
|
|
default: |
|
|
generator: rdn |
|
|
feature_extractor: false |
|
|
discriminator: false |
|
|
training_set: div2k |
|
|
test_set: sample |
|
|
log_dirs: |
|
|
logs: ./logs |
|
|
weights: ./weights |
|
|
feature_extractor: |
|
|
vgg19: |
|
|
layers_to_extract: |
|
|
- 5 |
|
|
- 9 |
|
|
generators: |
|
|
rrdn: |
|
|
C: 4 |
|
|
D: 3 |
|
|
G: 32 |
|
|
G0: 32 |
|
|
T: 4 |
|
|
x: 4 |
|
|
rdn: |
|
|
C: 6 |
|
|
D: 20 |
|
|
G: 64 |
|
|
G0: 64 |
|
|
x: 2 |
|
|
loss_weights: |
|
|
generator: 1.0 |
|
|
feature_extractor: 0.0833 |
|
|
discriminator: 0.01 |
|
|
losses: |
|
|
generator: mae |
|
|
discriminator: binary_crossentropy |
|
|
feature_extractor: mse |
|
|
session: |
|
|
prediction: |
|
|
patch_size: |
|
|
training: |
|
|
steps_per_epoch: 1000 |
|
|
patch_size: 32 |
|
|
batch_size: 16 |
|
|
epochs: 300 |
|
|
n_validation_samples: 100 |
|
|
learning_rate: |
|
|
initial_value: 0.0004 |
|
|
decay_frequency: 50 |
|
|
decay_factor: 0.5 |
|
|
fallback_save_every_n_epochs: 2 |
|
|
flatness: |
|
|
min: 0.0 |
|
|
increase_frequency: null |
|
|
increase: 0.0 |
|
|
max: 0.0 |
|
|
metrics: |
|
|
generator: PSNR_Y |
|
|
monitored_metrics: |
|
|
val_loss: min |
|
|
val_PSNR_Y: max |
|
|
adam_optimizer: |
|
|
beta1: 0.9 |
|
|
beta2: 0.999 |
|
|
epsilon: null |
|
|
test_sets: |
|
|
sample: ./data/input/sample |
|
|
training_sets: |
|
|
custom data: |
|
|
lr_train_dir: ./data/custom/lr/train |
|
|
hr_train_dir: ./data/custom/hr/train |
|
|
lr_valid_dir: ./data/custom/lr/validation |
|
|
hr_valid_dir: ./data/custom/hr/validation |
|
|
data_name: custom |
|
|
div2k: |
|
|
lr_train_dir: ./data/DIV2K/DIV2K_train_LR_bicubic/X2 |
|
|
hr_train_dir: ./data/DIV2K/DIV2K_train_HR |
|
|
lr_valid_dir: ./data/DIV2K/DIV2K_valid_LR_bicubic/X2 |
|
|
hr_valid_dir: ./data/DIV2K/DIV2K_valid_HR |
|
|
data_name: div2k |
|
|
weights_paths: |
|
|
discriminator: |
|
|
generator: |
|
|
|
