add resume option
you can resume video upscaling with `-resume 1` option.
This commit is contained in:
parent
11ff1d54ce
commit
2a30216798
1 changed files with 2 additions and 1 deletions
|
@ -59,7 +59,7 @@ local function convert_frames(opt)
|
||||||
end
|
end
|
||||||
fp:close()
|
fp:close()
|
||||||
for i = 1, #lines do
|
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 x = image_loader.load_float(lines[i])
|
||||||
local new_x = nil
|
local new_x = nil
|
||||||
if opt.m == "noise" and opt.noise_level == 1 then
|
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("-m", "noise_scale", 'method (noise|scale|noise_scale)')
|
||||||
cmd:option("-noise_level", 1, '(1|2)')
|
cmd:option("-noise_level", 1, '(1|2)')
|
||||||
cmd:option("-crop_size", 128, 'patch size per process')
|
cmd:option("-crop_size", 128, 'patch size per process')
|
||||||
|
cmd:option("-resume", 0, "skip existing files (0|1)")
|
||||||
|
|
||||||
local opt = cmd:parse(arg)
|
local opt = cmd:parse(arg)
|
||||||
if string.len(opt.l) == 0 then
|
if string.len(opt.l) == 0 then
|
||||||
|
|
Loading…
Reference in a new issue