plt.show()不显示图像

Question

代码如下:

import torch
import torchvision
import torchvision.transforms as transforms
import numpy
import matplotlib
import matplotlib.pyplot as plt

torch.set_printoptions(linewidth=120)
train_set = torchvision.datasets.FashionMNIST(
root='./data/FashionMNIST',
train=True,
download=True,
transform=transforms.Compose([
transforms.ToTensor()
])
) 
train_loader = torch.utils.data.DataLoader(
train_set, batch_size=10
)
sample = next(iter(train_set))
image, label = sample
plt.imshow(image.squeeze(), cmap='gray')
plt.show()
print(f"label:{label}")

我尝试通过matplob打印图像。但是什么也没发生

另外，我在我的linux服务器上这样做，而相同的代码在我的vscode本地工作得很好。

Answer 1

作为远程查看matplolib窗口的另一种选择，您可以始终将您的绘图保存为图像文件并将其复制到本地机器。这就像使用plt.savefig

一样简单

plt.save(f'label:{label}.png')