summaryrefslogtreecommitdiff
path: root/cli/app
diff options
context:
space:
mode:
Diffstat (limited to 'cli/app')
-rw-r--r--cli/app/search/live.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index d98af2f..4d8ad1b 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -479,6 +479,10 @@ class Listener:
out_img = vs.data2pil(gen_images[0])
to_pil_time_total += time.time() - to_pil_time
+ if out_img is None:
+ print("Got None instead of an image...?")
+ return
+
save_time = time.time()
# out_img.save(os.path.join(path_out, "frame_{:05d}.png".format(i)), format='png', compression_level=3)
out_img.save(pipe.stdin, format='png', compression_level=3)