Spaces:
Running
Running
change the network
Browse files
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
|
13 |
from options.options import parse
|
14 |
|
15 |
-
path_opt = './options/
|
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 =
|
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 |
-
|
34 |
-
|
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)
|