-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
38 lines (31 loc) · 1.29 KB
/
config.py
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
from easydict import EasyDict as edict
import json
config = edict()
config.TRAIN = edict()
## Adam
config.TRAIN.batch_size = 21 #16
config.TRAIN.lr_init = 1e-4
config.TRAIN.beta1 = 0.9
## initialize G
config.TRAIN.n_epoch_init = 50 #100
# config.TRAIN.lr_decay_init = 0.1
# config.TRAIN.decay_every_init = int(config.TRAIN.n_epoch_init / 2)
## adversarial learning (SRGAN)
config.TRAIN.n_epoch = 500 #2000
config.TRAIN.lr_decay = 0.1
config.TRAIN.decay_every = max(1, int(config.TRAIN.n_epoch / 2))
## train set location
config.TRAIN.hr_img_path = 'datasets/RELEASE_2015-08-31/DATA/TRAIN/HD/'
config.TRAIN.lr_img_path = 'datasets/RELEASE_2015-08-31/DATA/TRAIN/LR/'
config.VALID = edict()
## test set location
#config.VALID.hr_img_path = 'datasets/DIV2K_valid_HR/'
#config.VALID.lr_img_path = 'datasets/DIV2K_valid_LR_bicubic/X4/'
config.VALID.hr_img_path = 'datasets/RELEASE_2015-08-31/DATA/TEST/HD/'
config.VALID.lr_img_path = 'datasets/RELEASE_2015-08-31/DATA/TEST/LR/'
config.VALID.annot_path = 'datasets/RELEASE_2015-08-31/DATA/TEST/ANNOTATION/'
def log_config(filename, cfg):
with open(filename, 'w') as f:
f.write("================================================\n")
f.write(json.dumps(cfg, indent=4))
f.write("\n================================================\n")