Pytorch中的variable, tensor與numpy相互轉(zhuǎn)化的方法
在使用pytorch作為深度學(xué)習(xí)的框架時(shí),經(jīng)常會(huì)遇到變量variable、張量tensor與矩陣numpy的類型的相互轉(zhuǎn)化的問題,本章結(jié)合這實(shí)際圖像對(duì)此轉(zhuǎn)化方法進(jìn)行實(shí)現(xiàn)。
1.加載需要用到的模塊
import torch from torch.autograd import Variable import matplotlib.pyplot as plt import matplotlib.image as mpimg
2.顯示圖片與圖片中的一部分區(qū)域
test_img = mpimg.imread('example1.jpg')
i_x = 20
i_y = 85
sub_img = test_img[i_y:i_y + 100,i_x:i_x + 100,:] #numpy類型

3.將numpy矩陣轉(zhuǎn)換為Tensor張量
sub_ts = torch.from_numpy(sub_img) #sub_img為numpy類型
4.將Tensor張量轉(zhuǎn)化為numpy矩陣
sub_np1 = sub_ts.numpy() #sub_ts為tensor張量
5.將numpy轉(zhuǎn)換為Variable
sub_va = Variable(torch.from_numpy(sub_img))
6.將Variable張量轉(zhuǎn)化為numpy
sub_np2 = sub_va.data.numpy()
以上就是本文的全部內(nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
在終端啟動(dòng)Python時(shí)報(bào)錯(cuò)的解決方案
這篇文章主要介紹了在終端啟動(dòng)Python時(shí)報(bào)錯(cuò)的解決方案,幫助大家更好的理解和使用python,感興趣的朋友可以了解下2020-11-11
Python字典創(chuàng)建 遍歷 添加等實(shí)用基礎(chǔ)操作技巧
字段是Python是字典中唯一的鍵-值類型,本文講述了Python中字典如何創(chuàng)建 遍歷 添加等實(shí)用基礎(chǔ)操作技巧,內(nèi)容非?;A(chǔ)但非常重要,一定要熟練掌握2018-09-09
python獲取外網(wǎng)IP并發(fā)郵件的實(shí)現(xiàn)方法
下面小編就為大家?guī)硪黄猵ython獲取外網(wǎng)IP并發(fā)郵件的實(shí)現(xiàn)方法。小編覺得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧2017-10-10
Python編程使用tkinter模塊實(shí)現(xiàn)計(jì)算器軟件完整代碼示例
這篇文章主要介紹了Python編程實(shí)現(xiàn)一個(gè)計(jì)算器軟件完整代碼示例,簡單介紹了Tkinter的相關(guān)內(nèi)容,然后分享了通過tkinter模塊開發(fā)一個(gè)計(jì)算器的完整Python代碼,具有一定借鑒價(jià)值,需要的朋友可以參考下。2017-11-11
Python使用Pandas對(duì)比兩列數(shù)據(jù)取最大值的五種方法
本文主要介紹使用 Pandas 對(duì)比兩列數(shù)據(jù)取最大值的五種方法,包括使用 max 方法、apply 方法結(jié)合 lambda 函數(shù)、函數(shù)、clip 方法、where 方法結(jié)合條件賦值,并通過案例進(jìn)行詳細(xì)說明,需要的朋友可以參考下2025-02-02
python如何修改PYTHONPATH環(huán)境變量
這篇文章主要介紹了python如何修改PYTHONPATH環(huán)境變量問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-08-08
pycharm如何debug for循環(huán)里面的錯(cuò)誤值(推薦)
一般debug時(shí),在for循環(huán)里面的話,需要自己一步一步點(diǎn),如果循環(huán)幾百次那種就比較麻煩,此時(shí)可以采用try except的方式來解決,這篇文章主要介紹了pycharm如何debug for循環(huán)里面的錯(cuò)誤值,需要的朋友可以參考下2024-07-07

