From 64ffd2400c098eeee7483799b9a3d03839c70e9c Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 21 Aug 2018 15:01:42 +0200 Subject: skip frames --- live.py | 3 ++- options/dataset_options.py | 9 ++++++++- 2 files changed, 10 insertions(+), 2 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 diff --git a/options/dataset_options.py b/options/dataset_options.py index b1c453a..f8b7fe7 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -96,7 +96,7 @@ class DatasetOptions(BaseOptions): self.parser.add_argument( '--store-b', - action='store_true', + action='store_false', help='after generating a B image, save it to disk' ) @@ -252,6 +252,13 @@ class DatasetOptions(BaseOptions): help='amount to advance sequence after each frame' ) + self.parser.add_argument( + '--sequence-skip', + default=0.0, + type=float, + help='sequence frames to skip' + ) + self.parser.add_argument( '--sequence-playing', action='store_false', -- cgit v1.2.3-70-g09d2