upload video script

1 files changed, 3 insertions, 3 deletions

diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index b09575d..4c66873 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -374,6 +374,8 @@ class Listener:
     interpolator.build()
     self.rpc_client.send_status('processing', True)
     tag = "biggan_" + timestamp()
+    path_out = os.path.join(app_cfg.RESULTS_DIR, tag)
+    os.makedirs(path_out, exist_ok=True)
     dt = 1 / FPS
     for i in range(99999):
       if i == 0:
@@ -386,12 +388,10 @@ class Listener:
         print("Exiting...")
         break
       if (i % 100) == 0 or i == 1:
-        # print(gen_images.shape)
         print("Step {}. Generation time: {:.2f}s".format(i, time.time() - gen_time))
       out_img = vs.data2pil(gen_images[0])
       if out_img is not None:
-        # save image
-        #if out_img.resize_before_sending:
+        out_img.save(os.path.join(path_out, "image_{:5d}.png".format(i)), format='png')
         img_to_send = out_img.resize((256, 256), Image.BICUBIC)
         meta = {
           'i': i,