Spaces:
Running
Running
Delete options
Browse files- options/options.py +0 -44
- options/predict/LOLBlur.yml +0 -27
- options/train/All_LOL.yml +0 -82
- options/train/LOL.yml +0 -79
- options/train/LOLBlur.yml +0 -86
options/options.py
DELETED
@@ -1,44 +0,0 @@
|
|
1 |
-
import os
|
2 |
-
import os.path as osp
|
3 |
-
import logging
|
4 |
-
import yaml
|
5 |
-
from collections import OrderedDict
|
6 |
-
try:
|
7 |
-
from yaml import CLoader as Loader, CDumper as Dumper
|
8 |
-
except ImportError:
|
9 |
-
from yaml import Loader, Dumper
|
10 |
-
|
11 |
-
|
12 |
-
def OrderedYaml():
|
13 |
-
'''yaml orderedDict support'''
|
14 |
-
_mapping_tag = yaml.resolver.BaseResolver.DEFAULT_MAPPING_TAG
|
15 |
-
|
16 |
-
def dict_representer(dumper, data):
|
17 |
-
return dumper.represent_dict(data.items())
|
18 |
-
|
19 |
-
def dict_constructor(loader, node):
|
20 |
-
return OrderedDict(loader.construct_pairs(node))
|
21 |
-
|
22 |
-
Dumper.add_representer(OrderedDict, dict_representer)
|
23 |
-
Loader.add_constructor(_mapping_tag, dict_constructor)
|
24 |
-
return Loader, Dumper
|
25 |
-
|
26 |
-
#-----------------------
|
27 |
-
Loader, Dumper = OrderedYaml()
|
28 |
-
|
29 |
-
def parse(opt_path):
|
30 |
-
with open(opt_path, mode='r') as f:
|
31 |
-
opt = yaml.load(f, Loader=Loader)
|
32 |
-
return opt
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
if __name__ == '__main__':
|
37 |
-
|
38 |
-
path_yaml = './train/NBDN.yml'
|
39 |
-
with open(path_yaml, mode='r') as f:
|
40 |
-
opt = yaml.load(f, Loader=Loader)
|
41 |
-
opt = parse(path_yaml)
|
42 |
-
# print(opt)
|
43 |
-
print(type(opt['network']['width']))
|
44 |
-
# print(opt['gpu'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
options/predict/LOLBlur.yml
DELETED
@@ -1,27 +0,0 @@
|
|
1 |
-
#### network structures
|
2 |
-
network:
|
3 |
-
name: Network
|
4 |
-
img_channels: 3
|
5 |
-
width: 32
|
6 |
-
middle_blk_num_enc: 2
|
7 |
-
middle_blk_num_dec: 2
|
8 |
-
enc_blk_nums: [1, 2, 3]
|
9 |
-
dec_blk_nums: [3, 1, 1]
|
10 |
-
enc_blk_nums_map: None
|
11 |
-
middle_blk_num_map: None
|
12 |
-
residual_layers: None
|
13 |
-
dilations: [1, 4, 9]
|
14 |
-
spatial: False
|
15 |
-
extra_depth_wise: True
|
16 |
-
|
17 |
-
#### save model
|
18 |
-
save:
|
19 |
-
best: ./models/bests/Network_noFAC_LOLBlur.pt
|
20 |
-
path: ./models/Network_noFAC_LOLBlur.pt
|
21 |
-
|
22 |
-
LOLBlur:
|
23 |
-
inputs_path: ./inputs/synthetic
|
24 |
-
results_path: ./results/synthetic
|
25 |
-
RealBlur:
|
26 |
-
inputs_path: ./inputs/real
|
27 |
-
results_path: ./results/real
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
options/train/All_LOL.yml
DELETED
@@ -1,82 +0,0 @@
|
|
1 |
-
#### general settings
|
2 |
-
resume_training:
|
3 |
-
resume_training: True
|
4 |
-
resume: must
|
5 |
-
id: 2oceqxkt
|
6 |
-
|
7 |
-
#### training devices
|
8 |
-
device:
|
9 |
-
cuda: True
|
10 |
-
gpus: 0
|
11 |
-
|
12 |
-
#### datasets
|
13 |
-
datasets:
|
14 |
-
name: All_LOL
|
15 |
-
train:
|
16 |
-
train_path: /home/leadergpu/Datasets
|
17 |
-
|
18 |
-
n_workers: 4 # per GPU
|
19 |
-
batch_size_train: 12
|
20 |
-
cropsize: 256 # size you want to crop out as input sample.
|
21 |
-
flips: True
|
22 |
-
verbose: True
|
23 |
-
crop_type: Random
|
24 |
-
val:
|
25 |
-
test_path: /home/leadergpu/Datasets
|
26 |
-
batch_size_test: 1
|
27 |
-
|
28 |
-
#### network structures
|
29 |
-
network:
|
30 |
-
name: Network
|
31 |
-
img_channels: 3
|
32 |
-
width: 32
|
33 |
-
middle_blk_num: 3
|
34 |
-
enc_blk_nums: [1, 2, 3]
|
35 |
-
dec_blk_nums: [3, 1, 1]
|
36 |
-
enc_blk_nums_map: None
|
37 |
-
middle_blk_num_map: None
|
38 |
-
residual_layers: 2
|
39 |
-
dilations: [1, 4, 9]
|
40 |
-
spatial: None
|
41 |
-
extra_depth_wise: True
|
42 |
-
|
43 |
-
|
44 |
-
#### training settings: learning rate scheme, loss
|
45 |
-
train:
|
46 |
-
lr_initial: !!float 5e-4
|
47 |
-
lr_scheme: CosineAnnealing
|
48 |
-
betas: [0.9, 0.9]
|
49 |
-
epochs: 500
|
50 |
-
lr_gamma: 0.5
|
51 |
-
weight_decay: !!float 1e-3
|
52 |
-
eta_min: !!float 1e-6
|
53 |
-
|
54 |
-
pixel_criterion: l1
|
55 |
-
pixel_weight: 1.0
|
56 |
-
|
57 |
-
perceptual: True
|
58 |
-
perceptual_criterion: l1
|
59 |
-
perceptual_weight: 0.01
|
60 |
-
perceptual_reduction: mean
|
61 |
-
|
62 |
-
edge: True
|
63 |
-
edge_criterion: l2
|
64 |
-
edge_weight: 50.0
|
65 |
-
edge_reduction: mean
|
66 |
-
|
67 |
-
frequency: True
|
68 |
-
frequency_criterion: l2
|
69 |
-
frequency_weight: 0.01
|
70 |
-
frequency_reduction: mean
|
71 |
-
|
72 |
-
#### save model
|
73 |
-
save:
|
74 |
-
path: ./models/Network_all_LOL.pt
|
75 |
-
best: ./models/bests/
|
76 |
-
#### wandb:
|
77 |
-
wandb:
|
78 |
-
init: True
|
79 |
-
project: LOLBlur
|
80 |
-
entity: cidautai
|
81 |
-
name: Network_all_LOL
|
82 |
-
save_code: True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
options/train/LOL.yml
DELETED
@@ -1,79 +0,0 @@
|
|
1 |
-
#### general settings
|
2 |
-
resume_training:
|
3 |
-
resume_training: True
|
4 |
-
resume: must
|
5 |
-
id: waye95l7
|
6 |
-
|
7 |
-
|
8 |
-
#### datasets
|
9 |
-
datasets:
|
10 |
-
name: LOL
|
11 |
-
train:
|
12 |
-
train_path: /mnt/valab-datasets/LOL/train
|
13 |
-
|
14 |
-
n_workers: 4 # per GPU
|
15 |
-
batch_size_train: 4
|
16 |
-
cropsize: 256 # size you want to crop out as input sample.
|
17 |
-
flips: True
|
18 |
-
verbose: True
|
19 |
-
crop_type: Random
|
20 |
-
val:
|
21 |
-
test_path: /mnt/valab-datasets/LOL/test
|
22 |
-
cropsize: 256
|
23 |
-
batch_size_test: 1
|
24 |
-
|
25 |
-
#### network structures
|
26 |
-
network:
|
27 |
-
name: Network_v3
|
28 |
-
img_channels: 3
|
29 |
-
width: 32
|
30 |
-
middle_blk_num: 3
|
31 |
-
enc_blk_nums: [1, 2, 3]
|
32 |
-
dec_blk_nums: [3, 1, 1]
|
33 |
-
enc_blk_nums_map: None
|
34 |
-
middle_blk_num_map: None
|
35 |
-
residual_layers: 1
|
36 |
-
dilations: [1, 4]
|
37 |
-
spatial: None
|
38 |
-
extra_depth_wise: False
|
39 |
-
|
40 |
-
|
41 |
-
#### training settings: learning rate scheme, loss
|
42 |
-
train:
|
43 |
-
lr_initial: !!float 5e-4
|
44 |
-
lr_scheme: CosineAnnealing
|
45 |
-
betas: [0.9, 0.9]
|
46 |
-
epochs: 500
|
47 |
-
lr_gamma: 0.5
|
48 |
-
weight_decay: !!float 1e-3
|
49 |
-
eta_min: !!float 1e-6
|
50 |
-
|
51 |
-
pixel_criterion: l1
|
52 |
-
pixel_weight: 1.0
|
53 |
-
|
54 |
-
perceptual: True
|
55 |
-
perceptual_criterion: l1
|
56 |
-
perceptual_weight: 0.01
|
57 |
-
perceptual_reduction: mean
|
58 |
-
|
59 |
-
edge: True
|
60 |
-
edge_criterion: l2
|
61 |
-
edge_weight: 50.0
|
62 |
-
edge_reduction: mean
|
63 |
-
|
64 |
-
frequency: True
|
65 |
-
frequency_criterion: l2
|
66 |
-
frequency_weight: 0.01
|
67 |
-
frequency_reduction: mean
|
68 |
-
|
69 |
-
#### save model
|
70 |
-
save:
|
71 |
-
path: ./models/Network_v3_interpolate.pt
|
72 |
-
best: ./models/bests/
|
73 |
-
#### wandb:
|
74 |
-
wandb:
|
75 |
-
init: True
|
76 |
-
project: LOLBlur
|
77 |
-
entity: cidautai
|
78 |
-
name: Network_v3_interpolate_extraDW_LOLBlur
|
79 |
-
save_code: True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
options/train/LOLBlur.yml
DELETED
@@ -1,86 +0,0 @@
|
|
1 |
-
#### general settings
|
2 |
-
resume_training:
|
3 |
-
resume_training: True
|
4 |
-
resume: must
|
5 |
-
id: qa5ramgk
|
6 |
-
|
7 |
-
#### training devices
|
8 |
-
device:
|
9 |
-
cuda: True
|
10 |
-
gpus: 0
|
11 |
-
|
12 |
-
#### datasets
|
13 |
-
datasets:
|
14 |
-
name: LOLBlur
|
15 |
-
train:
|
16 |
-
train_path: /home/leadergpu/Datasets/LOLBlur_temp/train
|
17 |
-
|
18 |
-
n_workers: 4 # per GPU
|
19 |
-
batch_size_train: 12
|
20 |
-
cropsize: 256 # size you want to crop out as input sample.
|
21 |
-
flips: True
|
22 |
-
verbose: True
|
23 |
-
crop_type: Random
|
24 |
-
val:
|
25 |
-
test_path: /home/leadergpu/Datasets/LOLBlur_temp/test
|
26 |
-
batch_size_test: 1
|
27 |
-
|
28 |
-
#### network structures
|
29 |
-
network:
|
30 |
-
name: Network
|
31 |
-
img_channels: 3
|
32 |
-
width: 32
|
33 |
-
middle_blk_num_enc: 2
|
34 |
-
middle_blk_num_dec: 2
|
35 |
-
enc_blk_nums: [1, 2, 3]
|
36 |
-
dec_blk_nums: [3, 1, 1]
|
37 |
-
enc_blk_nums_map: None
|
38 |
-
middle_blk_num_map: None
|
39 |
-
residual_layers: None
|
40 |
-
dilations: [1, 4, 9]
|
41 |
-
spatial: None
|
42 |
-
extra_depth_wise: True
|
43 |
-
|
44 |
-
|
45 |
-
#### training settings: learning rate scheme, loss
|
46 |
-
train:
|
47 |
-
lr_initial: !!float 5e-4
|
48 |
-
lr_scheme: CosineAnnealing
|
49 |
-
betas: [0.9, 0.9]
|
50 |
-
epochs: 500
|
51 |
-
lr_gamma: 0.5
|
52 |
-
weight_decay: !!float 1e-3
|
53 |
-
eta_min: !!float 1e-6
|
54 |
-
|
55 |
-
pixel_criterion: l1
|
56 |
-
pixel_weight: 1.0
|
57 |
-
|
58 |
-
perceptual: True
|
59 |
-
perceptual_criterion: l1
|
60 |
-
perceptual_weight: 0.01
|
61 |
-
perceptual_reduction: mean
|
62 |
-
|
63 |
-
edge: True
|
64 |
-
edge_criterion: l2
|
65 |
-
edge_weight: 50.0
|
66 |
-
edge_reduction: mean
|
67 |
-
|
68 |
-
frequency: True
|
69 |
-
frequency_criterion: l2
|
70 |
-
frequency_weight: 0.01
|
71 |
-
frequency_reduction: sum
|
72 |
-
|
73 |
-
#### save model
|
74 |
-
save:
|
75 |
-
path: ./models/Network_noFAC_LOLBlur.pt
|
76 |
-
best: ./models/bests/
|
77 |
-
new: ./models/Network_noFAC_LOLBlur.pt
|
78 |
-
#### wandb:
|
79 |
-
wandb:
|
80 |
-
init: True
|
81 |
-
project: LOLBlur
|
82 |
-
entity: cidautai
|
83 |
-
name: Network_v3_noFAC_LOLBlur
|
84 |
-
save_code: True
|
85 |
-
|
86 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|