CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

Use cunet

Browse source
This commit is contained in:
nagadomi 2018-11-12 08:43:47 +00:00
parent 581c27a2cb
commit 07ac8b8559
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
web.lua
View file

@ -50,7 +50,8 @@ if cudnn then
end
opt.force_cudnn = opt.force_cudnn == 1
opt.enable_tta = opt.enable_tta == 1
local ART_MODEL_DIR = path.join(ROOT, "models", "upconv_7", "art")
--local ART_MODEL_DIR = path.join(ROOT, "models", "upconv_7", "art")
local ART_MODEL_DIR = path.join(ROOT, "models", "cunet", "art")
local PHOTO_MODEL_DIR = path.join(ROOT, "models", "upconv_7", "photo")
local art_model = {
scale = w2nn.load_model(path.join(ART_MODEL_DIR, "scale2.0x_model.t7"), opt.force_cudnn),