yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Tensorboard image broken: scipy.misc.toimage deprecated

biggoron opened this issue · comments

With the latest version of scipy.misc, scipy.misc.toimage is no longer available. To load and save an image as png we now have to use PIL, breaking tensorboard image summary.

Here is how I fixed the bug:
1./ At the end of main.py, log a uint8 image
logger.image_summary(tag, (images * 255).astype(np.uint8), step+1)
2./ In Logger class, package image as bytes with the PIL library (mode="L" read image as B&W uint8 image)
from PIL import Image
Image.fromarray(img, mode='L').save(s, "PNG")

[EDIT] I am not used to contributing to other people code but if you teach me how to do it I can fix the bug myself :)

Thank you ,it's helpful