update

app.py

@@ -297,13 +297,14 @@ def main(args):
         experiment_id, # path adding scheduler
     ]
     # outputs
-    outputs = gr.outputs.Image(type="pil", label="检测图片")
+    outputs = gr.outputs.Image(type="file", label="检测图片")
     outputs02 = gr.outputs.JSON(label="检测信息")
 
     # title
     title = "LIVE: Towards Layer-wise Image Vectorization"
     # description
-    description = "<div align='center'>(CVPR 2022 Oral Presentation)</div>"
+    description = "<div align='center'>(CVPR 2022 Oral Presentation)</div>" \
+                  "<div align='center'>Without GPUs, it will cost longer time.</div>"
 
     # examples
     examples = [

config/base.yaml

@@ -8,7 +8,7 @@ default:
     init: false
     image: false
     output: true
-    video: false
+    video: true
     loss: false
   trainable: 
     bg: False

main.py

@@ -503,6 +503,10 @@ def main_func(target, experiment, cfg_arg):
             img = img[:, :, 3:4] * img[:, :, :3] + \
                 para_bg * (1 - img[:, :, 3:4])
 
+
+
+
+
             if cfg.save.video:
                 filename = os.path.join(
                     cfg.experiment_dir, "video-png",
@@ -515,6 +519,10 @@ def main_func(target, experiment, cfg_arg):
                     imshow = img.detach().cpu()
                 pydiffvg.imwrite(imshow, filename, gamma=gamma)
 
+                ### added for app
+                if t%10==0:
+                    return filename, None
+
             x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW
 
             if cfg.use_ycrcb: