從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安裝路徑
這篇文章主要介紹了如何查看Mac本機(jī)的Python3安裝路徑問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2023-03-03
Jupyter Notebook遠(yuǎn)程登錄及密碼設(shè)置操作
今天小編就為大家分享一篇Jupyter Notebook遠(yuǎn)程登錄及密碼設(shè)置操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2020-04-04
Python 爬蟲爬取指定博客的所有文章
因?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鏡像源的方法示例
這篇文章主要介紹了Python更改pip鏡像源的方法示例，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-12-12