借鑒:http://www.manongjc.com/article/4333.html
pytorch 載入的數(shù)據(jù)集是元組tuple 形式顷蟀,里面包括了數(shù)據(jù)及標(biāo)簽(train_data,label)第步,其中的train_data數(shù)據(jù)可以轉(zhuǎn)換為torch.Tensor形式,方便后面計(jì)算使用梁钾。
img = torchvision.utils.make_grid(dataset[1][0]).numpy()
plt.imshow(np.transpose(img,(1,2,0)))
plt.show()
np.transpose 是因?yàn)閜lt.imshow在顯示 時(shí)候輸入的是(imgsize,imgsieze,channels),而這里得到的img是(3,200,200)的格式,所以進(jìn)行了轉(zhuǎn)換,才能顯示,如cifar-10