diff options
Diffstat (limited to 'live-mogrify.py')
| -rw-r--r-- | live-mogrify.py | 6 |
1 files changed, 6 insertions, 0 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() |