pytorch 實現(xiàn)張量tensor,圖片,CPU,GPU,數(shù)組等的轉(zhuǎn)換
更新時間：2020年01月13日 09:08:13 作者：小星星*
今天小編就為大家分享一篇pytorch 實現(xiàn)張量tensor,圖片,CPU,GPU,數(shù)組等的轉(zhuǎn)換，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
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 實現(xiàn)張量tensor,圖片,CPU,GPU,數(shù)組等的轉(zhuǎn)換就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了sklearn中make_blobs的用法詳情，sklearn中的make_blobs函數(shù)主要是為了生成數(shù)據(jù)集的，下面文章的學(xué)習(xí)內(nèi)容，需要的小伙伴可以參考一下2022-01-01
今天小編就為大家分享一篇python實現(xiàn)逆濾波與維納濾波示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
基于Python實現(xiàn)剪切板實時監(jiān)控方法解析
這篇文章主要介紹了基于Python實現(xiàn)剪切板實時監(jiān)控方法解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-09-09
python結(jié)合shell自動創(chuàng)建kafka的連接器實戰(zhàn)教程
這篇文章主要介紹了python結(jié)合shell自動創(chuàng)建kafka的連接器,需要安裝連接oracle的python包，獲取oracle表信息，本文給大家介紹的非常詳細(xì)，需要的朋友可以參考下2022-04-04
每天都是浪漫七夕！這篇文章主要為大家詳細(xì)介紹了python愛心表白，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-08-08
非極大值抑制（Non-Maximum Suppression，NMS），顧名思義就是抑制不是極大值的元素，可以理解為局部最大搜索。這個局部代表的是一個鄰域，鄰域有兩個參數(shù)可變，一是鄰域的維數(shù)，二是鄰域的大小2021-06-06
ORM Django 終端打印 SQL 語句實現(xiàn)解析
這篇文章主要介紹了ORM Django 終端打印 SQL 語句實現(xiàn)解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-08-08
詳解Python3之?dāng)?shù)據(jù)指紋MD5校驗與對比
這篇文章主要介紹了Python3之?dāng)?shù)據(jù)指紋MD5校驗與對比，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-06-06
這篇文章主要介紹了詳解python中executemany和序列的使用方法的相關(guān)資料,需要的朋友可以參考下2017-08-08
python3實現(xiàn)語音轉(zhuǎn)文字(語音識別)和文字轉(zhuǎn)語音(語音合成)
這篇文章主要介紹了python3實現(xiàn)語音轉(zhuǎn)文字(語音識別)和文字轉(zhuǎn)語音(語音合成)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10