from matplotlib import pyplot as plt
# loader使用torchvision中自带的transforms函数
from torchvision import transforms
loader = transforms.Compose([
transforms.ToTensor()])
unloader = transforms.ToPILImage()
def imshow(tensor, title=None):
image = tensor.cpu().clone()
image = image.squeeze(0)
image = unloader(image)
plt.imshow(image)
if title is not None:
plt.title(title)
plt.pause(0.001)
pytorch展示Tensor类型的图片
最新推荐文章于 2023-01-06 13:02:38 发布