從Pytorch模型pth文件中讀取參數(shù)成numpy矩陣的操作
更新時間：2021年03月04日 14:46:27 作者：木盞
這篇文章主要介紹了從Pytorch模型pth文件中讀取參數(shù)成numpy矩陣的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
目的：
把訓(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()是以一個生成器的形式迭代返回每一層的參數(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矩陣。
方便又好用，爆贊~
補充：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)#生成一個隨機輸入維度的輸入
traced_script_module = torch.jit.trace(model, example)
traced_script_module.save("model.pt")
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
淺談python中常用的8種經(jīng)典數(shù)據(jù)結(jié)構(gòu)
這篇文章主要介紹了python中常用的8種經(jīng)典數(shù)據(jù)結(jié)構(gòu),包括原生數(shù)據(jù)結(jié)構(gòu),NumPy包中的數(shù)據(jù)結(jié)構(gòu),以及Pandas包中的數(shù)據(jù)結(jié)構(gòu),需要的朋友可以參考下2023-03-03
這篇文章主要介紹了python動態(tài)網(wǎng)頁批量爬取的方法，主要針對四六級成績批量爬取，感興趣的小伙伴們可以參考一下2016-02-02
將SVG（可縮放矢量圖形）文件插入到PDF（便攜式文檔格式）文件中不僅能夠保留SVG圖像的矢量特性,確保圖像在任何分辨率下都保持清晰,使得技術(shù)文檔、手冊、報告等內(nèi)容更加豐富多樣且易于傳播,本文將介紹如何使用Python插入SVG文件到PDF文檔中,需要的朋友可以參考下2024-08-08
本篇文章通過代碼實例給大家分享了python3爬取淘寶信息的過程以及實例分析，對此有興趣的朋友學(xué)習(xí)下。2018-02-02
這篇文章主要介紹了如何在Python中對文件進行操作，文章圍繞主題展開內(nèi)容，即使用Python中內(nèi)置的open()函數(shù)來打開文件，返回文件對象，并對文件進行處理2022-08-08
這篇文章主要介紹了Python WebSocket長連接心跳與短連接的示例，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下2020-11-11
python 如何將office文件轉(zhuǎn)換為PDF
這篇文章主要介紹了python 如何將office文件轉(zhuǎn)換為PDF，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-09-09
django 解決自定義序列化返回處理數(shù)據(jù)為null的問題
這篇文章主要介紹了django 解決自定義序列化返回處理數(shù)據(jù)為null的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要介紹了在?Python?中讀取?gzip?文件,本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-05-05
今天小編就為大家分享一篇關(guān)于Python常用的json標(biāo)準(zhǔn)庫，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2019-02-02