From 8d2d1639e282ebe6e490c1bd20a3f3f490826a36 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 21 May 2018 02:49:26 +0200
Subject: more live code clean up and options

---
 options/base_options.py    |  1 +
 options/dataset_options.py | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+)

(limited to 'options')

diff --git a/options/base_options.py b/options/base_options.py
index f741927..f3e93c6 100644
--- a/options/base_options.py
+++ b/options/base_options.py
@@ -42,6 +42,7 @@ class BaseOptions():
         self.parser.add_argument('--no_flip', action='store_true', help='if specified, do not flip the images for data augmentation')
         self.parser.add_argument('--init_type', type=str, default='normal', help='network initialization [normal|xavier|kaiming|orthogonal]')
         self.parser.add_argument('--center_crop', action='store_true', help='center crop instead of random crop')
+        self.parser.add_argument('--poll_delay', type=float, default=0.01, help='time to wait before checking for the next frame')
 
         self.initialized = True
 
diff --git a/options/dataset_options.py b/options/dataset_options.py
index 356ce60..a201903 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -72,6 +72,27 @@ class DatasetOptions(BaseOptions):
           help='scp destination'
         )
 
+        ## LIVE IMAGE PROCESSING
+
+        self.parser.add_argument(
+          '--send_image',
+          type=str,
+          default='b',
+          help='which image to send... a, b, a_mix, sequence'
+        )
+
+        self.parser.add_argument(
+          '--store_a',
+          action='store_true',
+          help='dont remove the generated A image after processing it'
+        )
+
+        self.parser.add_argument(
+          '--store_b',
+          action='store_true',
+          help='after generating a B image, save it to disk'
+        )
+
         ## IMAGE FILTERS
 
         ### RECURSION
-- 
cgit v1.2.3-70-g09d2