pycharm中debug时查看图像的变化

问题描述

如题，使用pycharm进行debug时，常常需要查看图像的变化。如果使用opencv，常常会卡死，如果使用matplotlib.pyplot， matplotlib.pyplot在ubuntu系统上常常报错。

优势

PIL不像opencv一样需要waitkey()，当需要显示多个图像时，PIL可以缓存在一个窗口里面；而且PIL不需要Qt库，对系统兼容性更好。
将PIL与torchviion提供的ToPILImage结合，可以将tensor格式的张量转变为PIL格式的图片，从而更加简便地查看数据增强后的图片或者特征图。

解决方法

使用PIL，可以在程序中加入

from PIL import Image
img_show = Image.fromarray(img)
img_show.show()

甚至可以只在终端中输入该函数，一样可以正常显示。