diff options
| -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 |