seek to a frame

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'
         )