两种可视化图片数据的方法

一种是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)