diff --git a/appendix/waifu2x-nocuda/lib/settings.lua b/appendix/waifu2x-nocuda/lib/settings.lua index 56f894d..3ea818c 100644 --- a/appendix/waifu2x-nocuda/lib/settings.lua +++ b/appendix/waifu2x-nocuda/lib/settings.lua @@ -21,7 +21,7 @@ cmd:text("Options:") cmd:option("-seed", 11, 'fixed input seed') cmd:option("-data_dir", "./data", 'data directory') cmd:option("-test", "images/miku_small.png", 'test image file') -cmd:option("-model_dir", "./models", 'model directory') +cmd:option("-model_dir", "/usr/share/waifu2x/models", 'model directory') cmd:option("-method", "scale", '(noise|scale)') cmd:option("-noise_level", 1, '(1|2)') cmd:option("-scale", 2.0, 'scale') diff --git a/appendix/waifu2x-nocuda/waifu2x.lua b/appendix/waifu2x-nocuda/waifu2x.lua index 7afe284..d0b4f37 100644 --- a/appendix/waifu2x-nocuda/waifu2x.lua +++ b/appendix/waifu2x-nocuda/waifu2x.lua @@ -19,7 +19,7 @@ local function waifu2x() cmd:text("Options:") cmd:option("-i", "images/miku_small.png", 'path of input image') cmd:option("-o", "(auto)", 'path of output') - cmd:option("-model_dir", "./models", 'model directory') + cmd:option("-model_dir", "/usr/share/waifu2x/models", 'model directory') cmd:option("-m", "noise_scale", 'method (noise|scale|noise_scale)') cmd:option("-noise_level", 1, '(1|2)') cmd:option("-crop_size", 128, 'crop size') diff --git a/src/main.cpp b/src/main.cpp index ad1a2d4..b2a7ecc 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -52,7 +52,7 @@ int main(int argc, char** argv) { TCLAP::ValueArg cmdModelPath("", "model_dir", "path to custom model directory (don't append last / )", false, - "models_rgb", "string", cmd); + "/usr/share/waifu2x/models_rgb", "string", cmd); TCLAP::ValueArg cmdNumberOfJobs("j", "jobs", "number of threads launching at the same time", false, 0, "integer",