learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-29 17:57:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 normalize images in sample

Browse Source
This commit is contained in:
Varuna Jayasiri
2020-10-28 09:23:35 +05:30
parent 98b659f439
commit c21d7cab57
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
labml_nn/gan/cycle_gan.py
View File

@ -692,6 +692,10 @@ def sample():
data = x_image.unsqueeze(0).to(conf.device) data = x_image.unsqueeze(0).to(conf.device)
generated_y = conf.generator_xy(data) generated_y = conf.generator_xy(data)
# Normalize the image
mm_range = generated_y.min(), generated_y.max()
generated_y = (generated_y - mm_range[0]) / (mm_range[1] - mm_range[0] + 1e-5)
# Display the generated image. We have to change the order of dimensions to HWC. # Display the generated image. We have to change the order of dimensions to HWC.
plt.imshow(generated_y[0].cpu().permute(1, 2, 0)) plt.imshow(generated_y[0].cpu().permute(1, 2, 0))
plt.show() plt.show()