Browse Source

Use cunet

nagadomi 6 năm trước cách đây
mục cha
commit
07ac8b8559
1 tập tin đã thay đổi với 2 bổ sung1 xóa
  1. 2 1
      web.lua

+ 2 - 1
web.lua

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