從Pytorch模型pth文件中讀取參數(shù)成numpy矩陣的操作
更新時(shí)間：2021年03月04日 14:46:27 作者：木盞
這篇文章主要介紹了從Pytorch模型pth文件中讀取參數(shù)成numpy矩陣的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
目的：
把訓(xùn)練好的pth模型參數(shù)提取出來，然后用其他方式部署到邊緣設(shè)備。
Pytorch給了很方便的讀取參數(shù)接口：
nn.Module.parameters()
直接看demo：
from torchvision.models.alexnet import alexnet model = alexnet(pretrained=True).eval().cuda() parameters = model.parameters() for p in parameters: numpy_para = p.detach().cpu().numpy() print(type(numpy_para)) print(numpy_para.shape)
上面得到的numpy_para就是numpy參數(shù)了~
Note：
model.parameters()是以一個(gè)生成器的形式迭代返回每一層的參數(shù)。所以用for循環(huán)讀取到各層的參數(shù)，循環(huán)次數(shù)就表示層數(shù)。
而每一層的參數(shù)都是torch.nn.parameter.Parameter類型，是Tensor的子類，所以直接用tensor轉(zhuǎn)numpy（即p.detach().cpu().numpy()）的方法就可以直接轉(zhuǎn)成numpy矩陣。
方便又好用，爆贊~
補(bǔ)充：pytorch訓(xùn)練好的.pth模型轉(zhuǎn)換為.pt
將python訓(xùn)練好的.pth文件轉(zhuǎn)為.pt
import torch
import torchvision
from unet import UNet
model = UNet(3, 2)#自己定義的網(wǎng)絡(luò)模型
model.load_state_dict(torch.load("best_weights.pth"))#保存的訓(xùn)練模型
model.eval()#切換到eval（）
example = torch.rand(1, 3, 320, 480)#生成一個(gè)隨機(jī)輸入維度的輸入
traced_script_module = torch.jit.trace(model, example)
traced_script_module.save("model.pt")
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
在python3中pyqt5和mayavi不兼容問題的解決方法
今天小編就為大家分享一篇在python3中pyqt5和mayavi不兼容問題的解決方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-01-01
Python機(jī)器學(xué)習(xí)NLP自然語言處理基本操作關(guān)鍵詞
本文是Python機(jī)器學(xué)習(xí)NLP自然語言處理系列文章，帶大家開啟一段學(xué)習(xí)自然語言處理 (NLP) 的旅程. 本文主要學(xué)習(xí)NLP自然語言處理關(guān)鍵詞的操作2021-09-09
python3基于TCP實(shí)現(xiàn)CS架構(gòu)文件傳輸
這篇文章主要為大家詳細(xì)介紹了python3基于TCP實(shí)現(xiàn)CS架構(gòu)文件傳輸，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-07-07
這篇文章主要介紹了如何查看Mac本機(jī)的Python3安裝路徑問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-03-03
Jupyter Notebook遠(yuǎn)程登錄及密碼設(shè)置操作
今天小編就為大家分享一篇Jupyter Notebook遠(yuǎn)程登錄及密碼設(shè)置操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-04-04
因?yàn)镚oogle App Engine 被墻，我無法繼續(xù)完善我的Moven project 還有20＋天才回去，怕到時(shí)候會(huì)忘記project的進(jìn)度和細(xì)節(jié)就趁著個(gè)冷的什么都不想干的時(shí)候， 大概的總結(jié)一下2016-02-02
Python實(shí)現(xiàn)單例模式的四種方式詳解
單例模式可以保證一個(gè)類僅有一個(gè)實(shí)例，并提供一個(gè)訪問它的全局訪問點(diǎn)。本文為大家介紹了Python實(shí)現(xiàn)單例模式的四種方式，需要的可以參考一下2022-05-05
Python的Flask框架中@app.route的用法教程
這篇文章主要介紹了Python的Flask框架中@app.route的用法教程,包括相關(guān)的正則表達(dá)式講解，是Flask學(xué)習(xí)過程當(dāng)中的基礎(chǔ)知識(shí)，需要的朋友可以參考下2015-03-03
django美化后臺(tái)django-suit的安裝配置操作
這篇文章主要介紹了django美化后臺(tái)django-suit的安裝配置操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-07-07
這篇文章主要介紹了Python更改pip鏡像源的方法示例，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12