File size: 1,816 Bytes
ed697ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from PIL import Image
import matplotlib.pyplot as plt
# Log images
def log_image(x, opts):
return tensor2im(x)
def tensor2im(var):
var = var.cpu().detach().transpose(0, 2).transpose(0, 1).numpy()
var = ((var + 1) / 2)
var[var < 0] = 0
var[var > 1] = 1
var = var * 255
return Image.fromarray(var.astype('uint8'))
def vis_faces(log_hooks):
display_count = len(log_hooks)
fig = plt.figure(figsize=(12, 4 * display_count))
gs = fig.add_gridspec(display_count, 4)
for i in range(display_count):
hooks_dict = log_hooks[i]
vis_faces_with_age(hooks_dict, fig, gs, i)
plt.tight_layout()
return fig
def vis_faces_with_age(hooks_dict, fig, gs, i):
fig.add_subplot(gs[i, 0])
plt.imshow(hooks_dict['input_face'])
plt.title('Input\nOut Sim={:.2f}\nInput Age={:.2f}'.format(float(hooks_dict['diff_input_real']),
float(hooks_dict['input_age_real'])))
fig.add_subplot(gs[i, 1])
plt.imshow(hooks_dict['target_face'])
plt.title('Target\nIn={:.2f},Out={:.2f}\nTarget Age={:.2f}'.format(float(hooks_dict['diff_views_real']),
float(hooks_dict['diff_target_real']),
float(hooks_dict['target_age_real'])))
fig.add_subplot(gs[i, 2])
plt.imshow(hooks_dict['output_face'])
plt.title('Output\nTarget Sim={:.2f}\nOuput Age={:.2f}'.format(float(hooks_dict['diff_target_real']),
float(hooks_dict['output_age_real'])))
fig.add_subplot(gs[i, 3])
plt.imshow(hooks_dict['recovered_face'])
plt.title('Recovered\nTarget Sim={:.2f}\nOuput Age={:.2f}'.format(float(hooks_dict['diff_target_cycle']),
float(hooks_dict['output_age_cycle'])))
|