pytorch 加載(.pth)格式的模型實(shí)例
有一些非常流行的網(wǎng)絡(luò)如 resnet、squeezenet、densenet等在pytorch里面都有，包括網(wǎng)絡(luò)結(jié)構(gòu)和訓(xùn)練好的模型。
pytorch自帶模型網(wǎng)址：https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-models/
按官網(wǎng)加載預(yù)訓(xùn)練好的模型：
import torchvision.models as models # pretrained=True就可以使用預(yù)訓(xùn)練的模型 resnet18 = models.resnet18(pretrained=True) print(resnet18)
報(bào)錯(cuò)如下：
requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于連接方在一段時(shí)間后沒(méi)有正確答復(fù)或連接的主機(jī)沒(méi)有反應(yīng)，連接嘗試失敗。', None, 10060, None))
主要是因?yàn)榇a會(huì)去遠(yuǎn)端下載模型的參數(shù)，而國(guó)內(nèi)的網(wǎng)一般連接不上，這是我們需要手動(dòng)去下載你要的預(yù)訓(xùn)練網(wǎng)絡(luò)。
通過(guò)地址下載，地址有兩種獲取方式：
1.從報(bào)錯(cuò)里面獲取，上述代碼運(yùn)行時(shí)會(huì)出現(xiàn)這樣一行信息：
Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to C:\Users\Luo/.torch\models\resnet18-5c106cde.pth
復(fù)制這個(gè)網(wǎng)址到瀏覽器，有可能打不開(kāi)，去掉https://，直接輸入download.pytorch.org/models/resnet18-5c106cde.pth就可以下載了。
2.從pytorch的github下找模型的地址：https://github.com/pytorch/vision/tree/master/torchvision/models
找到對(duì)應(yīng)模型名稱(chēng)點(diǎn)進(jìn)去找地址
下載好后自行保存，我是直接存在pytorch models里面
接下來(lái)就是運(yùn)行這個(gè).pth文件。首先要判斷是保存的整個(gè)網(wǎng)絡(luò)結(jié)構(gòu)加參數(shù)呢，還是只保存了參數(shù)，可以測(cè)試一下。這是我的模型是squeezenet1_1，你可以測(cè)試自己下載的模型
import torch pthfile = r'E:\anaconda\app\envs\luo\Lib\site-packages\torchvision\models\squeezenet1_1.pth' net = torch.load(pthfile) print(net)
結(jié)果為
很明顯就是只保存了參數(shù)，這是我們要換個(gè)方法加載模型
import torch import torchvision.models as models # pretrained=True就可以使用預(yù)訓(xùn)練的模型 net = models.squeezenet1_1(pretrained=False) pthfile = r'E:\anaconda\app\envs\luo\Lib\site-packages\torchvision\models\squeezenet1_1.pth' net.load_state_dict(torch.load(pthfile)) print(net)
結(jié)果;
這下就加載好預(yù)訓(xùn)練模型了
以上這篇pytorch 加載(.pth)格式的模型實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
- 淺談pytorch 模型 .pt, .pth, .pkl的區(qū)別及模型保存方式
- pytorch實(shí)現(xiàn)從本地加載 .pth 格式模型
- pytorch模型轉(zhuǎn)onnx模型的方法詳解
- PyTorch?模型?onnx?文件導(dǎo)出及調(diào)用詳情
- Pytorch轉(zhuǎn)onnx、torchscript方式
- Pytorch通過(guò)保存為ONNX模型轉(zhuǎn)TensorRT5的實(shí)現(xiàn)
- Pytorch模型轉(zhuǎn)onnx模型實(shí)例
- Pytorch pth 格式轉(zhuǎn)ONNX 格式的詳細(xì)過(guò)程
相關(guān)文章
matplotlib多子圖實(shí)現(xiàn)共享坐標(biāo)軸的示例詳解
這篇文章主要為大家詳細(xì)介紹了matplotlib繪制多子圖師如何實(shí)現(xiàn)共享坐標(biāo)軸,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-02-02
這篇文章主要介紹了如何在python字符串中輸入純粹的{}以及python字符串連接的三種方法,需要的朋友可以參考下2018-08-08
Python編程中對(duì)super函數(shù)的正確理解和用法解析
可能有人會(huì)想到,Python中既然可以直接通過(guò)父類(lèi)名調(diào)用父類(lèi)方法為什么還會(huì)存在super函數(shù)?其實(shí),很多人對(duì)Python中的super函數(shù)的認(rèn)識(shí)存在誤區(qū),本文我們就帶來(lái)在Python編程中對(duì)super函數(shù)的正確理解和用法解析2016-07-07
python實(shí)現(xiàn)查找excel里某一列重復(fù)數(shù)據(jù)并且剔除后打印的方法
這篇文章主要介紹了python實(shí)現(xiàn)查找excel里某一列重復(fù)數(shù)據(jù)并且剔除后打印的方法,涉及Python使用xlrd模塊操作Excel的相關(guān)技巧,需要的朋友可以參考下2015-05-05
本文主要介紹了python實(shí)現(xiàn)矩陣的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07
這篇文章主要介紹了python基于property()函數(shù)定義屬性,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-01-01
計(jì)算機(jī)二級(jí)python學(xué)習(xí)教程（3） python語(yǔ)言基本數(shù)據(jù)類(lèi)型
這篇文章主要為大家詳細(xì)介紹了計(jì)算機(jī)二級(jí)python學(xué)習(xí)教程的第3篇，python語(yǔ)言基本數(shù)據(jù)類(lèi)型，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-05-05
一起解密Python中的*args和**kwargs無(wú)限可能的函數(shù)參數(shù)
這篇文章主要來(lái)跟大家一起解密Python中的*args和**kwargs無(wú)限可能的函數(shù)參數(shù)使用的靈活性，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-06-06
今天小編就為大家分享一篇python獲取代理IP的實(shí)例分享，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-05-05
Django Admin后臺(tái)模型列表頁(yè)面如何添加自定義操作按鈕
這篇文章主要介紹了Django Admin后臺(tái)模型列表頁(yè)面如何添加自定義操作按鈕,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11