diff options
Diffstat (limited to 'cli')
| -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)) |