pytorch 實(shí)現(xiàn)張量tensor,圖片,CPU,GPU,數(shù)組等的轉(zhuǎn)換
更新時(shí)間：2020年01月13日 09:08:13 作者：小星星*
今天小編就為大家分享一篇pytorch 實(shí)現(xiàn)張量tensor,圖片,CPU,GPU,數(shù)組等的轉(zhuǎn)換，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
1, 創(chuàng)建pytorch 的Tensor張量：
torch.rand((3,224,224)) #創(chuàng)建隨機(jī)值的三維張量，大小為（3,224,224） torch.Tensor([3,2]) #創(chuàng)建張量，[3,2]
2, cpu上的tensor和GPU即pytorch創(chuàng)建的tensor的相互轉(zhuǎn)化
b = a.cpu() # GPU → CPU a = b.cuda() #CPU → GPU
3, tensor和numpy的轉(zhuǎn)化
b = a.numpy() # tensor轉(zhuǎn)化為 numpy數(shù)組 a = b.from_numpy() # numpy數(shù)組轉(zhuǎn)化為tensor
4, torch的GPU tensor保存為圖片
import scipy.misc scipy.misc.imsave(‘pic_name',img) #img為二維張量，比如（224,224），保存為黑白圖
5, 堆疊矩陣，形成彩色圖片
img = np.stack((ia,b,c),dim) #堆疊矩陣a,b,c 可用于三通道圖像的保存 dim表示要增加的維度， #比如a,b,c均為（224,224）大小的矩陣，那么令dim=-1，則 img的維度為（224,224,3)
6, 從numpy數(shù)組保存圖片
from PIL import Image
im = Image.fromarray(A)
im.save("your_file.jpeg")
7, 讀取圖片為矩陣:
import matplotlib.image
im = matplotlib.image.imread('0_0.jpg')
8, 保存矩陣為圖片:
import numpy as np
import scipy.misc
x = np.random.random((600,800,3))
scipy.misc.imsave('meelo.jpg', x)
以上這篇pytorch 實(shí)現(xiàn)張量tensor,圖片,CPU,GPU,數(shù)組等的轉(zhuǎn)換就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python3加密解密庫Crypto的RSA加解密和簽名/驗(yàn)簽實(shí)現(xiàn)方法實(shí)例
這篇文章主要介紹了Python3加密解密庫Crypto的RSA加解密和簽名/驗(yàn)簽實(shí)現(xiàn)方法實(shí)例,需要的朋友可以參考下2020-02-02
詳解Python3.8+PyQt5+pyqt5-tools+Pycharm配置詳細(xì)教程
這篇文章主要介紹了Python3.8+PyQt5+pyqt5-tools+Pycharm配置詳細(xì)教程，本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友參考下吧2020-11-11
今天小編就為大家分享一篇python 字符串追加實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
100 個(gè) Python 小例子（練習(xí)題二）
這篇文章主要繼續(xù)上一篇文章100 個(gè) Python 小例子（練習(xí)題一）來完成100 個(gè) Python 小例子，本文包括字母識(shí)詞、反向輸出II、表轉(zhuǎn)字符串、設(shè)置輸出顏色、算素?cái)?shù)等例子,需要的朋友可以參考一下2021-10-10
這篇文章主要介紹了在Python的Django框架中包裝視圖函數(shù)的方法,即requires_login的相關(guān)方法,需要的朋友可以參考下2015-07-07
親手教你用Python打造一款摸魚倒計(jì)時(shí)界面
前段時(shí)間在微博看到一段摸魚人的倒計(jì)時(shí)模板，感覺很有意思，于是我用了一個(gè)小時(shí)的時(shí)間寫一個(gè)頁面出來，下面小編把實(shí)現(xiàn)過程分享給大家，對Python摸魚倒計(jì)時(shí)界面感興趣的朋友一起看看吧2021-12-12
如何用C代碼給Python寫擴(kuò)展庫(Cython)
這篇文章主要介紹了如何用C代碼給Python寫擴(kuò)展庫(Cython)，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2019-05-05
Python使用三種方法實(shí)現(xiàn)PCA算法
這篇文章主要介紹了Python使用三種方法實(shí)現(xiàn)PCA算法，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2017-12-12
這篇文章主要為大家詳細(xì)介紹了如何通過Python語言實(shí)現(xiàn)視頻分幀的功能，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以跟隨小編一起嘗試一下2023-03-03
這篇文章主要介紹了Python中的切片函數(shù)Slice詳解,切片是對有序的集合而言,意思從有序集合里提取數(shù)據(jù)構(gòu)成子集集合,給定提取的起點(diǎn)start、終點(diǎn)end以及方向上的步長step,能否切出非空子集,起點(diǎn)start需能沿步長方向上到達(dá)終點(diǎn),需要的朋友可以參考下2023-09-09