diff --git a/waifu2x.lua b/waifu2x.lua index 3d8d364..1764c5a 100644 --- a/waifu2x.lua +++ b/waifu2x.lua @@ -59,7 +59,7 @@ local function convert_frames(opt) end fp:close() for i = 1, #lines do - if path.exists(string.format(opt.o, i)) == false then + if opt.resume == 0 or path.exists(string.format(opt.o, i)) == false then local x = image_loader.load_float(lines[i]) local new_x = nil if opt.m == "noise" and opt.noise_level == 1 then @@ -110,6 +110,7 @@ local function waifu2x() cmd:option("-m", "noise_scale", 'method (noise|scale|noise_scale)') cmd:option("-noise_level", 1, '(1|2)') cmd:option("-crop_size", 128, 'patch size per process') + cmd:option("-resume", 0, "skip existing files (0|1)") local opt = cmd:parse(arg) if string.len(opt.l) == 0 then