diff options
| -rw-r--r-- | live-mogrify.py | 6 | ||||
| -rw-r--r-- | options/dataset_options.py | 4 |
2 files changed, 8 insertions, 2 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index b511af1..fedb246 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -199,6 +199,8 @@ class Listener(): print('load sequence: {}'.format(payload)) self.data_opt.sequence_name = payload self.data_opt.load_sequence = True + if cmd == 'seek': + self.data_opt.seek_to = payload if cmd == 'get_status': return { 'processing': self.data_opt.processing, @@ -259,6 +261,10 @@ def process_live_input(opt, data_opt, rpc_client, model): print("Got sequence {}, {} images, first: {}".format(data_opt.sequence_name, len(sequence), sequence[0])) sequence = new_sequence sequence_i = 1 + if data_opt.seek_to != 1: + if data_opt.seek_to > 0 and data_opt.seek_to < len(sequence_i): + sequence_i = data_opt.seek_to + data_opt.seek_to = 1 model.set_input(data) model.test() diff --git a/options/dataset_options.py b/options/dataset_options.py index de1dd57..7ac5a8a 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -115,9 +115,9 @@ class DatasetOptions(BaseOptions): ) self.parser.add_argument( - '--sequence-offset', + '--seek_to', type=int, - default=0, + default=1, help='initial offset into sequence' ) |