more general

1 files changed, 3 insertions, 1 deletions

diff --git a/options/test_options.py b/options/test_options.py
index 6b79860..f8c3a31 100644
--- a/options/test_options.py
+++ b/options/test_options.py
@@ -1,9 +1,11 @@
 from .base_options import BaseOptions
-
+import random
 
 class TestOptions(BaseOptions):
     def initialize(self):
         BaseOptions.initialize(self)
+        self.parser.add_argument('--start_img', type=str, default="test.jpg", help='starting image for recursive generation')
+        self.parser.add_argument('--experiment', type=str, default="exp_{:06d}".format(random.randint(1,1000001)), help='experiment name (name of directory)')
         self.parser.add_argument('--ntest', type=int, default=float("inf"), help='# of test examples.')
         self.parser.add_argument('--results_dir', type=str, default='./results/', help='saves results here.')
         self.parser.add_argument('--aspect_ratio', type=float, default=1.0, help='aspect ratio of result images')