diff options
| -rw-r--r-- | live.py | 3 | ||||
| -rw-r--r-- | options/dataset_options.py | 9 |
2 files changed, 10 insertions, 2 deletions
@@ -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' ) @@ -253,6 +253,13 @@ class DatasetOptions(BaseOptions): ) 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', help='pause sequence iterator' |