问题描述
如题,使用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()
甚至可以只在终端中输入该函数,一样可以正常显示。