update

app.py

@@ -95,6 +95,8 @@ def parse_args():
 
 
 def app_experiment_change(experiment_id):
+    if experiment_id == "add [1] total 1 path for demonstration":
+        return "experiment_1x1"
     if experiment_id == "add [1, 1, 1, 1, 1] total 5 paths one by one":
         return "experiment_5x1"
     elif experiment_id == "add [1, 1, 1, 1, 1, 1, 1, 1] total 8 paths one by one":
@@ -293,6 +295,7 @@ def main(args):
     inputs_img = gr.inputs.Image(type="pil", label="Input Image")
     experiment_id = gr.inputs.Radio(
         choices=[
+            "add [1] total 1 path for demonstration",
             "add [1, 1, 1, 1, 1] total 5 paths one by one",
             "add [1, 1, 1, 1, 1, 1, 1, 1] total 8 paths one by one",
             "add [1,2,4,8,16,32, ...] total 128 paths",

config/base.yaml

@@ -31,12 +31,19 @@ default:
     xing_loss_weight: 0.01
     bis_loss_weight: null
 
-experiment_4x1:
+
+experiment_1x1:
   path_schedule:
     type: repeat
     max_path: 1
     schedule_each: 1
 
+experiment_4x1:
+  path_schedule:
+    type: repeat
+    max_path: 4
+    schedule_each: 1
+
 experiment_5x1:
   path_schedule:
     type: repeat

main.py

@@ -520,10 +520,10 @@ def main_func(target, experiment, num_iter, cfg_arg):
                     imshow = img.detach().cpu()
                 pydiffvg.imwrite(imshow, filename, gamma=gamma)
 
-            ### added for app
-            if t%30==0 and t !=0 :
-                # print(f"debug: {t}, {filename} {img.size()}")
-                return img.detach().cpu().numpy(), t
+            # ### added for app
+            # if t%30==0 and t !=0 :
+            #     # print(f"debug: {t}, {filename} {img.size()}")
+            #     return img.detach().cpu().numpy(), t
 
             x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW
 
@@ -693,6 +693,7 @@ def main_func(target, experiment, num_iter, cfg_arg):
             # shutil.rmtree(os.path.join(cfg.experiment_dir, "video-png"))
 
     print("The last loss is: {}".format(loss.item()))
+    return img.detach().cpu().numpy(), t
 
 
 if __name__ == "__main__":