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

