summaryrefslogtreecommitdiff
path: root/cli/app/search
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-09 20:58:04 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-09 20:58:04 +0100
commit062d272b5ae1417e3d77a11eacfff2f2042b493b (patch)
tree9f9263d96ec5bbc6d39b5519544734f7e5424d4d /cli/app/search
parentdc809d31769d6bb910394b0a872134876ae22334 (diff)
video
Diffstat (limited to 'cli/app/search')
-rw-r--r--cli/app/search/search_class.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py
index 0afa762..f290908 100644
--- a/cli/app/search/search_class.py
+++ b/cli/app/search/search_class.py
@@ -198,7 +198,9 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la
if opt_video and opt_snapshot_interval != 0 and (i % opt_snapshot_interval) == 0:
phi_guess = sess.run(output)
guess_im = imgrid(imconvert_uint8(phi_guess), cols=1)
- imwrite(join(app_cfg.DIR_OUTPUTS, fp_frames, 'frame_{:04d}.png'.format(int(i / opt_snapshot_interval))), guess_im)
+ fp_out_im = join(app_cfg.DIR_OUTPUTS, fp_frames, 'frame_{:04d}.png'.format(int(i / opt_snapshot_interval)))
+ print(fp_out_im)
+ imwrite(fp_out_im, guess_im)
except KeyboardInterrupt:
pass