danifei commited on
Commit
81903d2
·
verified ·
1 Parent(s): 56563be

change the network

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -9,10 +9,10 @@ import numpy as np
9
  import yaml
10
  from huggingface_hub import hf_hub_download
11
 
12
- from archs import Network_v3
13
  from options.options import parse
14
 
15
- path_opt = './options/test/LOLBlur.yml'
16
 
17
  opt = parse(path_opt)
18
 
@@ -25,13 +25,13 @@ network = opt['network']['name']
25
 
26
  PATH_MODEL = opt['save']['path']
27
 
28
- model = Network_v3(img_channel=opt['network']['img_channels'],
29
  width=opt['network']['width'],
30
  middle_blk_num=opt['network']['middle_blk_num'],
31
  enc_blk_nums=opt['network']['enc_blk_nums'],
32
  dec_blk_nums=opt['network']['dec_blk_nums'],
33
- residual_layers=opt['network']['residual_layers'],
34
- dilations=opt['network']['dilations'])
35
 
36
  checkpoints = torch.load(opt['save']['best'])
37
  # print(checkpoints)
 
9
  import yaml
10
  from huggingface_hub import hf_hub_download
11
 
12
+ from archs import Network
13
  from options.options import parse
14
 
15
+ path_opt = './options/predict/LOLBlur.yml'
16
 
17
  opt = parse(path_opt)
18
 
 
25
 
26
  PATH_MODEL = opt['save']['path']
27
 
28
+ model = Network(img_channel=opt['network']['img_channels'],
29
  width=opt['network']['width'],
30
  middle_blk_num=opt['network']['middle_blk_num'],
31
  enc_blk_nums=opt['network']['enc_blk_nums'],
32
  dec_blk_nums=opt['network']['dec_blk_nums'],
33
+ dilations=opt['network']['dilations'],
34
+ extra_depth_wise = opt['network']['extra_depth_wise'])
35
 
36
  checkpoints = torch.load(opt['save']['best'])
37
  # print(checkpoints)