summaryrefslogtreecommitdiff
path: root/live.py
diff options
context:
space:
mode:
Diffstat (limited to 'live.py')
-rw-r--r--live.py3
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