小试牛刀—CIFAR-10

import torchvision as tv
import torchvision.transforms as transforms
from torchvision.transforms import ToPILImage
show=ToPILImage()#可以把Tensor转成Image,方便可视化


#定义对数据的预处理
transform=transforms.Compose([transforms.ToTensor(),#转化为Tensor
                             transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5),)#归一化
                             ])


#训练集
trainset=tv.datasets.CIFAR10(
    root='/media/z/CC/',
    train=True,
    download=True,
    transform=transform)

trainloader=t.utils.data.DataLoader(trainset,
                                   batch_size=4,
                                   shuffle=True,
                                   num_workers=1)

#测试集
testset=tv.datasets.CIFAR10('/media/z/CC/',
                           train=False,
                           download=True,
                           transform=transform)

testloader=t.utils.data.DataLoader(testset,
                                  batch_size=4,
                                  shuffle=False,
                                  num_workers=1)

classes=('plane','car','bird','cat',
        'deer','dog','frog','horse','ship','truck')
#(data,label)=trainset[100]
#print(classes[label])
#show((data+1)/2).resize((100,100))
dataiter=iter(trainloader)
images,labels=dataiter.next()#返回4张图片及标签
print(''.join('%11s'%classes[labels[j]] for j in range(4)))
show(tv.utils.make_grid((images+1)/2)).resize((400,100))
                                  

输出 

  plane        car      truck       deer