“ToPILImage”对象没有属性“show”

Question

我正在做一个图像处理任务，我想连接两个网站的图片。为了连接，我先将图像转换为张量，然后将张量转换为PIL图像来显示它，但报错了。有人可以帮助我吗？这是我的代码：

import skimage.io as io
import torch
from torchvision import  models, transforms
from PIL import Image
import matplotlib.pyplot as plt
from torchvision.transforms import ToPILImage
import numpy as np
from skimage import data_dir,io,color

coll1 = io.ImageCollection('F:\\code1/*.jpg')
coll2 = io.ImageCollection('F:\\code2/*.jpg')
a = torch.tensor(coll1)
print(a)
print(a.shape)
b = torch.tensor(coll2)
print(b)
print(b.shape)
c=torch.cat((a,b),1)
print(c.shape)
print(c)
img= transforms.ToPILImage()
img.show()

这是错误代码：

回溯（最近一次调用最后一次）：文件“F:/filelist.py”，第 39 行，在 img.show() AttributeError: 'ToPILImage' 对象没有属性 'show'

Answer 1

该ToPILImage方法接受张量或 ndarray 作为输入source。

您需要将单个图像张量投射到该ToPILImage方法。从您的帖子中，我怀疑您正在传递一批图像张量而不是一批，因此会出现错误。

假设如果你想可视化来自张量的图像c，

img = transforms.ToPILImage()(c)
img.show()