我的代码如下:
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}")
我试图通过matploblib.pylot打印一个图像,但是什么都没有发生。
另外,我在我的linux服务器上做这件事,而在我的vscode上,相同的代码在本地运行得很好。
发布于 2022-07-07 10:42:52
作为远程查看matplolib窗口的另一种选择,您可以始终将绘图保存为图像文件并将其复制到本地计算机。这就像使用plt.savefig
一样简单
plt.save(f'label:{label}.png')
https://stackoverflow.com/questions/72894769
复制相似问题