在学习CIFAR10时,在 PyCharm中无法显示图片
跟着慕课网上的视频教程,仔细对比代码,无效,百度一番
解决办法如下
plt.imshow(npimg) # 显示图像
plt.show()
果然有效
附上完整代码
import torch # 张量库
import torchvision # 计算机视觉库
import torchvision.transforms as transforms # 数据转换模块
# 数据转换
transform = transforms.Compose(
[
transforms.ToTensor(), # 转换为张量
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) # 标准化
]
)
# 训练数据集
trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) # 训练集
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4, shuffle=True, num_workers=2) # 训练加载器
# 测试数据集
testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform) # 测试集
testloader = torch.utils.data.DataLoader(testset, batch_size=4, shuffle=False, num_workers=2) # 测试加载器
import matplotlib.pyplot as plt # 绘图库
import numpy as np # 数组库
def imshow(img):
img = img / 2 + 0.5 # 反标准化
npimg = img.numpy() # 转换为numpy数组
npimg = np.transpose(npimg, (1, 2, 0)) # 转置
plt.imshow(npimg) # 显示图像
plt.show()
# 随机获取一些训练图像
dataiter = iter(trainloader) # 迭代器
images, labels = next(dataiter) # 获取数据
# 显示图像
imshow(torchvision.utils.make_grid(images))
您可以选择一种方式赞助本站
支付宝扫一扫赞助
微信钱包扫描赞助
赏