skip frames

1 files changed, 2 insertions, 1 deletions

diff --git a/live.py b/live.py
index 735c5ad..81daeb7 100644
--- a/live.py
+++ b/live.py
@@ -86,6 +86,7 @@ def process_live_input(opt, data_opt, rpc_client):
       print(model)
 
   sequence_i = 1
+  skip_i = 0
 
   print("generating...")
   for i, data in enumerate(dataset):
@@ -128,7 +129,7 @@ def process_live_input(opt, data_opt, rpc_client):
 
     im = util.tensor2im(generated.data[0])
 
-    sequence_i = mix_next_image(opt, data_opt, rpc_client, im, i, sequence, sequence_i)
+    sequence_i, skip_i = mix_next_image(opt, data_opt, rpc_client, im, i, sequence, sequence_i, skip_i)
 
     if data_opt.pause:
       data_opt.pause = False