diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-09 20:58:04 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-09 20:58:04 +0100 |
| commit | 062d272b5ae1417e3d77a11eacfff2f2042b493b (patch) | |
| tree | 9f9263d96ec5bbc6d39b5519544734f7e5424d4d /cli/app/search/search_class.py | |
| parent | dc809d31769d6bb910394b0a872134876ae22334 (diff) | |
video
Diffstat (limited to 'cli/app/search/search_class.py')
| -rw-r--r-- | cli/app/search/search_class.py | 4 |
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 |