camvid数据集使用方法_使用PyTorch处理CIFAR10数据集并显示

在训练图像分类的时候，我们通常会使用CIFAR10数据集，今天就先写一下如何展示数据集的图片及预处理。

第一部分代码，展示原始图像：

import numpy as npimport torch#导入内置cifarfrom torchvision.datasets import cifar#预处理模块import torchvision.transforms as transformsfrom torch.utils.data import DataLoaderimport matplotlib.pyplot as pltclasses = ('plane', 'car', 'bird', 'cat','deer', 'dog', 'frog', 'horse', 'ship', 'truck')#Compose将一些转换函数组合在一起#ToTensor，原始数据是numpy，现在改成Tensor。会将数据从[0,255]归一化到[0,1] 除以255transforms=transforms.Compose([transforms.ToTensor()])trainData=cifar.CIFAR10('./picdata',train=True,transform=transforms,download=True)testData=cifar.CIFAR10('./picdata',train=False,transform=transforms)x=0for images, labels in trainData:    plt.subplot(3,3,x+1)    plt.tight_layout()    images = images.numpy().transpose(1, 2, 0)  # 把channel那一维放到最后    plt.title(str(classes[labels]))    plt.imshow(images)    plt.xticks([])    plt.yticks([])    x+=1    if x==9:        breakplt.show()

图片展示如下：

第二部分代码，灰度化图片：

import numpy as npimport torch#导入内置cifarfrom torchvision.datasets import cifar#预处理模块import torchvision.transforms as transformsfrom torch.utils.data import DataLoaderimport matplotlib.pyplot as pltclasses = ('plane', 'car', 'bird', 'cat','deer', 'dog', 'frog', 'horse', 'ship', 'truck')#Compose将一些转换函数组合在一起#ToTensor，原始数据是numpy，现在改成Tensor。会将数据从[0,255]归一化到[0,1] 除以255#Normalize则是将数据按照通道进行标准化，(输入[通道]-均值[通道])/标准差[通道]，将数据归一化到[-1,1]#如果数据在[0,1]之间，则实际的偏移量bias会很大。而一般模型初始化的时候，bias=0，这样收敛的就会慢。经过Normalize后加快收敛速度#后面两个0.5就是制定mean和std，原来[0,1]变成：(0-0.5)/0.5=-1，(1-0.5)/0.5=1。本例是要灰度化，就一个通道，如果是三通道RGB，则应该为[0.5,0.5,0.5]  ,transforms.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])transforms=transforms.Compose([transforms.ToTensor(),transforms.Normalize([0.5],[0.5])])trainData=cifar.CIFAR10('./picdata',train=True,transform=transforms,download=True)testData=cifar.CIFAR10('./picdata',train=False,transform=transforms)#shuffle随机打乱trainLoader=DataLoader(trainData,batch_size=64,shuffle=False)testLoader=DataLoader(testData,batch_size=128,shuffle=False)#enumerate组合成一个索引序列，同时列出数据下标和数据examples=enumerate(trainLoader)batchIndex,(imgData,labels)=next(examples)fig=plt.figure()for i in range(9):    plt.subplot(3,3,i+1)    plt.tight_layout()    plt.imshow(imgData[i][0],cmap='gray',interpolation='none')    plt.title("{}".format(classes[labels[i]]))    plt.xticks([])    plt.yticks([])plt.show()

图片展示如下：