update

app.py

@@ -109,11 +109,11 @@ cfg_arg = parse_args()
 temp_image = np.random.rand(224,224,3)
 temp_text = "start"
 temp_input = np.random.rand(224,224,3)
-def run_live(img, experiment_id, cfg_arg=cfg_arg):
+def run_live(img, experiment_id, num_iter, cfg_arg=cfg_arg):
     experiment = app_experiment_change(experiment_id)
     cfg_arg.target = img
     cfg_arg.experiment = experiment
-    img, text = main_func(img, experiment_id, cfg_arg=cfg_arg)
+    img, text = main_func(img, experiment_id, num_iter, cfg_arg=cfg_arg)
     return temp_image, text
 
 
@@ -287,9 +287,9 @@ def run_live(img, experiment_id, cfg_arg=cfg_arg):
 def main(args):
     gr.close_all()
     # -------------------Inputs-------------------
-    inputs_update = gr.inputs.Slider(
-        label=" ",
-        default=50, maximum=100, minimum=1, step=1)
+    inputs_iteration = gr.inputs.Slider(
+        label="Optimization Iteration",
+        default=500, maximum=600, minimum=100, step=100)
     inputs_img = gr.inputs.Image(type="pil", label="Input Image")
     experiment_id = gr.inputs.Radio(
         choices=[
@@ -304,7 +304,7 @@ def main(args):
 
         inputs_img,  # input image
         experiment_id, # path adding scheduler
-        # inputs_update, # update output
+        inputs_iteration, # input iteration
 
     ]
     # outputs
@@ -322,22 +322,27 @@ def main(args):
         [
             "./examples/1.png",
             "add [1, 1, 1, 1, 1] total 5 paths one by one",
+            300,
         ],
         [
             "./examples/2.png",
             "add [1, 1, 1, 1, 1] total 5 paths one by one",
+            300,
         ],
         [
             "./examples/3.jpg",
             "add [1,2,4,8,16,32, ...] total 128 paths",
+            300,
         ],
         [
             "./examples/4.png",
             "add [1,2,4,8,16,32, ...] total 256 paths",
+            300,
         ],
         [
             "./examples/5.png",
             "add [1, 1, 1, 1, 1] total 5 paths one by one",
+            300,
         ],
     ]
 

main.py

@@ -345,7 +345,7 @@ class linear_decay_lrlambda_f(object):
         lr = lr_s * (1-r) + lr_e * r
         return lr
 
-def main_func(target, experiment, cfg_arg):
+def main_func(target, experiment, num_iter, cfg_arg):
     with open(cfg_arg.config, 'r') as f:
         cfg = yaml.load(f, Loader=yaml.FullLoader)
     cfg_default = edict(cfg['default'])
@@ -358,6 +358,7 @@ def main_func(target, experiment, cfg_arg):
         osp.join(cfg.log_dir, '{}_{}'.format(cfg.exid, '_'.join(cfg.signature)))
     cfg.target = target
     cfg.experiment = experiment
+    cfg.num_iter = num_iter
 
     configfile = osp.join(cfg.experiment_dir, 'config.yaml')
     check_and_create_dir(configfile)