Spaces:
Runtime error
Runtime error
File size: 991 Bytes
f8ea2c9 |
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 39 40 |
llm:
model: 'TaylorAI/bge-micro-v2' # See Paper Sec. 3.2 and Appendix
model_dim: 384
embd_dim: 256
nclasses: 7 # noise, blur, rain, haze, lol, enhancement, upsampling (Paper Sec. 4.3)
weights: False
model:
arch: "instructir"
use_text: True
in_ch: 3
out_ch: 3
width : 32
enc_blks: [2, 2, 4, 8]
middle_blk_num: 4
dec_blks: [2, 2, 2, 2]
textdim: 256
weights: False
test:
batch_size: 1
num_workers: 3
dn_datapath: "data/denoising_testsets/"
dn_datasets: ["CBSD68", "urban100", "Kodak24", "McMaster"]
dn_sigmas: [15, 25, 50]
rain_targets: ["data/Rain/rain_test/Rain100L/target/"]
rain_inputs: ["data/Rain/rain_test/Rain100L/input/"]
haze_targets: "data/SOTS-OUT/GT/"
haze_inputs : "data/SOTS-OUT/IN/"
lol_targets: "data/LOL/eval15/high/"
lol_inputs : "data/LOL/eval15/low/"
gopro_targets: "data/gopro_test/GoPro/target/"
gopro_inputs: "data/gopro_test/GoPro/input/"
|