Pytorch抽取網(wǎng)絡(luò)層的Feature Map(Vgg)實(shí)例
更新時(shí)間：2019年08月20日 10:51:32 作者：seeney
今天小編就為大家分享一篇Pytorch抽取網(wǎng)絡(luò)層的Feature Map(Vgg)實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
這邊我是需要得到圖片在Vgg的5個(gè)block里relu后的Feature Map （其余網(wǎng)絡(luò)只需要替換就可以了）
索引可以這樣獲得
vgg = models.vgg19(pretrained=True).features.eval() print (vgg)
Feature Map可利用下面的class
class Vgg16(nn.Module):
def __init__(self, pretrained=True):
super(Vgg16, self).__init__()
self.net = models.vgg16(pretrained).features.eval()
def forward(self, x):
out = []
for i in range(len(self.net)):
x = self.net[i](x)
if i in [3, 8, 15, 22, 29]:
# print(self.net[i])
out.append(x
return out
以上這篇Pytorch抽取網(wǎng)絡(luò)層的Feature Map(Vgg)實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python下實(shí)現(xiàn)的RSA加密/解密及簽名/驗(yàn)證功能示例
這篇文章主要介紹了Python下實(shí)現(xiàn)的RSA加密/解密及簽名/驗(yàn)證功能,結(jié)合具體實(shí)例形式分析了Python中RSA加密、解密的實(shí)現(xiàn)方法及簽名、驗(yàn)證功能的使用技巧,需要的朋友可以參考下2017-07-07
關(guān)于Python文本生成的Beam?Search解碼問題
這篇文章主要介紹了Python文本生成的Beam?Search解碼,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-07-07
Python多進(jìn)程并發(fā)與多線程并發(fā)編程實(shí)例總結(jié)
這篇文章主要介紹了Python多進(jìn)程并發(fā)與多線程并發(fā)編程,結(jié)合實(shí)例形式總結(jié)分析了Python編程中的多進(jìn)程并發(fā)與多線程并發(fā)相關(guān)概念、使用方法與操作注意事項(xiàng),需要的朋友可以參考下2018-02-02
Python數(shù)組遍歷的簡(jiǎn)單實(shí)現(xiàn)方法小結(jié)
這篇文章主要介紹了Python數(shù)組遍歷的簡(jiǎn)單實(shí)現(xiàn)方法,結(jié)合實(shí)例總結(jié)分析了Python針對(duì)數(shù)組的元素,索引常用遍歷技巧,需要的朋友可以參考下2016-04-04
這篇文章主要介紹了利用python代碼寫的12306訂票代碼，自己寫的python代碼，是非常實(shí)用的技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-12-12
python re模塊findall()函數(shù)實(shí)例解析
這篇文章主要介紹了python re模塊findall()函數(shù)實(shí)例解析，具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01
tensorflow 2.0模式下訓(xùn)練的模型轉(zhuǎn)成 tf1.x 版本的pb模型實(shí)例
這篇文章主要介紹了tensorflow 2.0模式下訓(xùn)練的模型轉(zhuǎn)成 tf1.x 版本的pb模型實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨想過來看看吧2020-06-06
在本篇內(nèi)容里小編給大家整理一篇關(guān)于python爬取抖音視頻的實(shí)例分析的相關(guān)內(nèi)容，有興趣的朋友可以測(cè)試下實(shí)例內(nèi)容。2021-01-01
如果你打算從過去的一個(gè)版本升級(jí)Django, 你需要先刪除老版本的Django之后，再安裝新的版本。下面這篇文章主要給大家介紹了在Django卸載之后重新安裝的方法，文中給出了詳細(xì)的步驟，需要的朋友可以參考借鑒，下面來一起看看吧。2017-03-03
Expected conditions模塊使用方法匯總代碼解析
這篇文章主要介紹了Expected conditions模塊使用方法匯總代碼解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-08-08