diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-06 15:42:04 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-06 15:42:04 +0100 |
| commit | c40438a98578cd3d5a86320b0cd4fda73a2b9df3 (patch) | |
| tree | 5cf3861586ee5d91ffc3be2e3e179c867bb3c9a1 /cli/app | |
| parent | b6a4e8e477a4f00c94f5bb7afdfd0a68f995fe2d (diff) | |
getting it running..?
Diffstat (limited to 'cli/app')
| -rw-r--r-- | cli/app/commands/biggan/search_class.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index b0d957b..1420572 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -132,7 +132,7 @@ def find_nearest_vector(sess, opt_fp_in, opt_dims, out_images, out_labels, out_l # feed_dict = {input_z: z, input_y: y, input_trunc: truncation} phi_start = sess.run(output, feed_dict=feed_dict) - start_im = imconvert_uint8(phi_start) + start_im = imgrid(imconvert_uint8(phi_start), cols=1) imwrite(join(app_cfg.DIR_OUTPUTS, fp_frames, 'frame_0000_start.png'), start_im) try: @@ -141,7 +141,7 @@ def find_nearest_vector(sess, opt_fp_in, opt_dims, out_images, out_labels, out_l curr_loss, _, _ = sess.run([loss, train_step_z, train_step_y], feed_dict=feed_dict) phi_guess = sess.run(output) - guess_im = imconvert_uint8(phi_guess) + guess_im = imgrid(imconvert_uint8(phi_guess), cols=1) imwrite(join(app_cfg.DIR_OUTPUTS, fp_frames, 'frame_{:04d}.png'.format(i)), guess_im) if i % 20 == 0: print('iter: {}, loss: {}'.format(i, curr_loss)) |