diff options
Diffstat (limited to 'cli/app/search')
| -rw-r--r-- | cli/app/search/live.py | 4 |
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) |