一种是import matplotlib.pyplot as plt
假设我们自定义了图片Dataset，data_loader从Dataset取出来的是可迭代对象。
去第一个对象p1 = next(iter(data_loader))，p1是个[images,labels]的列表。而且images,labels都是一个batch的
p1[0].size()
torch.Size([12, 3, 224, 224])
在取第一个对象
p1[0][0]
是一个tensor，是我们可视化需要的
tensor([[[0.6706, 0.6784, 0.6941, …, 0.5137, 0.5294, 0.5490],
[0.7020, 0.7059, 0.6863, …, 0.5176, 0.5608, 0.5647],
[0.6863, 0.6706, 0.6353, …, 0.5412, 0.5647, 0.5686],
[0.9294, 0.9255, 0.9255, …, 0.8235, 0.8431, 0.8392],
[0.9333, 0.9255, 0.9255, …, 0.8196, 0.8431, 0.8275],
[0.9373, 0.9294, 0.9333, …, 0.8314, 0.8471, 0.8275]]])
def imshow(img):
版权声明：本文为zhangzhifu2019原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。