danifei commited on
Commit
262c416
·
verified ·
1 Parent(s): efdb65a

Delete options

Browse files
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
-