From 2a3021679880e2c4932ae40384b47cb7c374f5e2 Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Tue, 26 May 2015 00:35:34 +0900
Subject: [PATCH] add resume option

you can resume video upscaling with `-resume 1` option.
---
 waifu2x.lua | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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