快捷導(dǎo)航

pytorch 數(shù)據(jù)集圖片顯示方法

更新時(shí)間：2018年07月26日 15:59:26 作者：zzw小凡

今天小編就為大家分享一篇pytorch 數(shù)據(jù)集圖片顯示方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

圖片顯示

pytorch 載入的數(shù)據(jù)集是元組tuple 形式，里面包括了數(shù)據(jù)及標(biāo)簽（train_data,label），其中的train_data數(shù)據(jù)可以轉(zhuǎn)換為torch.Tensor形式，方便后面計(jì)算使用。

同樣給一些剛?cè)腴T(mén)的同學(xué)在使用載入的數(shù)據(jù)顯示圖片的時(shí)候帶來(lái)一些難以理解的地方，這里主要是將Tensor與numpy轉(zhuǎn)換的過(guò)程，理解了這些就可以就行轉(zhuǎn)換了

CIAFA10數(shù)據(jù)集

首先載入數(shù)據(jù)集，這里做了一些數(shù)據(jù)處理，包括圖片尺寸、數(shù)據(jù)歸一化等

import torch
from torch.autograd import Variable 
import matplotlib.pyplot as plt 
import torchvision.datasets as dset
import torchvision.transforms as transforms
from autoencoder import AutoEncoder
import torch.nn as nn
import torchvision
import numpy as np
dataset = dset.CIFAR10(root='../train/data', download=True, 
    transform=transforms.Compose([
    transforms.Scale(200),
    transforms.ToTensor(),
    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
    transforms.Gray()
    ]))

在這里 dataset 是一個(gè)CIFAR10對(duì)象，(大家可以查看一下他的源代碼）

方式一

dataset[1] = ([torch.FloatTensor of size 1x200x200]，9)

載入的第二個(gè)數(shù)據(jù)是個(gè)tensor格式，包含一個(gè)標(biāo)簽 9

這里我們做的就是將torch.FloatTensor 轉(zhuǎn)換為numpy，然后顯示

b = dataset[1][0].numpy()
#取數(shù)據(jù)，不取標(biāo)簽

因?yàn)檫@里的b仍然是1*200*200的大小，所以要重新reshape一下，適合輸出圖像

plt.imshow(b.reshape(200,200),cmap = 'gray')
plt.show()

然后可以顯示圖像了

方式二

利用torch的接口

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)換，才能顯示

以上這篇pytorch 數(shù)據(jù)集圖片顯示方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python讀寫(xiě)刪除復(fù)制文件操作方法詳細(xì)實(shí)例總結(jié)
這篇文章主要介紹了python讀寫(xiě)刪除復(fù)制文件操作方法詳細(xì)實(shí)例總結(jié),需要的朋友可以參考下
2021-04-04
Python處理電子表格的Pandas、OpenPyXL、xlrd和xlwt庫(kù)
在Python中處理表格數(shù)據(jù),有幾個(gè)非常流行且功能強(qiáng)大的庫(kù),Pandas在數(shù)據(jù)分析方面提供了廣泛的功能,而OpenPyXL、xlrd和xlwt則在處理Excel文件方面各有所長(zhǎng),以下是一些最常用的庫(kù)及其示例代碼
2024-01-01
解決PyCharm不在run輸出運(yùn)行結(jié)果而不是再Console里輸出的問(wèn)題
這篇文章主要介紹了解決PyCharm不在run輸出運(yùn)行結(jié)果而不是再Console里輸出的問(wèn)題,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2020-09-09
python圖形界面教程Tkinter詳解
Python自帶了tkinter模塊,實(shí)質(zhì)上是一種流行的面向?qū)ο蟮腉UI工具包?TK?的Python編程接口,提供了快速便利地創(chuàng)建GUI應(yīng)用程序的方法,下面這篇文章主要給大家介紹了關(guān)于python圖形界面教程Tkinter的相關(guān)資料,需要的朋友可以參考下
2022-12-12
pycharm下配置pyqt5的教程(anaconda虛擬環(huán)境下+tensorflow)
這篇文章主要介紹了pycharm下配置pyqt5的教程(anaconda虛擬環(huán)境下+tensorflow),本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2020-03-03
Python繪制的二項(xiàng)分布概率圖示例
這篇文章主要介紹了Python繪制的二項(xiàng)分布概率圖,涉及Python基于numpy、math的數(shù)值運(yùn)算及matplotlib圖形繪制相關(guān)操作技巧,需要的朋友可以參考下
2018-08-08
Python繪制箭頭向量圖的示例代碼
箭頭向量圖十分常見(jiàn)，比如天氣預(yù)報(bào)在顯示風(fēng)場(chǎng)的時(shí)候，就會(huì)貼心地用箭頭指明風(fēng)的方向，下面就跟隨小編一起學(xué)習(xí)一下如何利用Python繪制箭頭向量圖吧
2023-08-08
python GUI庫(kù)圖形界面開(kāi)發(fā)之PyQt5菜單欄控件QMenuBar的詳細(xì)使用方法與實(shí)例
這篇文章主要介紹了python GUI庫(kù)圖形界面開(kāi)發(fā)之PyQt5菜單欄控件QMenuBar的詳細(xì)使用方法與實(shí)例,需要的朋友可以參考下
2020-02-02
Python中的二叉樹(shù)查找算法模塊使用指南
二叉樹(shù)查找算法，在開(kāi)發(fā)實(shí)踐中，會(huì)經(jīng)常用到。按照慣例，對(duì)于這么一個(gè)常用的東西，Python一定會(huì)提供輪子的。是的，python就是這樣，一定會(huì)讓開(kāi)發(fā)者省心，降低開(kāi)發(fā)者的工作壓力。
2014-07-07
幾種實(shí)用的pythonic語(yǔ)法實(shí)例代碼
在我理解，Pythonic 就是很 Python 的 Python 代碼。下面這篇文章主要給大家分享介紹了幾種實(shí)用的pythonic語(yǔ)法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧。
2018-02-02