Pytorch抽取網(wǎng)絡(luò)層的Feature Map(Vgg)實例

更新時間：2019年08月20日 10:51:32 作者：seeney

今天小編就為大家分享一篇Pytorch抽取網(wǎng)絡(luò)層的Feature Map(Vgg)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

這邊我是需要得到圖片在Vgg的5個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)實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python下實現(xiàn)的RSA加密/解密及簽名/驗證功能示例
這篇文章主要介紹了Python下實現(xiàn)的RSA加密/解密及簽名/驗證功能,結(jié)合具體實例形式分析了Python中RSA加密、解密的實現(xiàn)方法及簽名、驗證功能的使用技巧,需要的朋友可以參考下
2017-07-07
關(guān)于Python文本生成的Beam?Search解碼問題
這篇文章主要介紹了Python文本生成的Beam?Search解碼,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2022-07-07
Python多進程并發(fā)與多線程并發(fā)編程實例總結(jié)
這篇文章主要介紹了Python多進程并發(fā)與多線程并發(fā)編程,結(jié)合實例形式總結(jié)分析了Python編程中的多進程并發(fā)與多線程并發(fā)相關(guān)概念、使用方法與操作注意事項,需要的朋友可以參考下
2018-02-02
Python數(shù)組遍歷的簡單實現(xiàn)方法小結(jié)
這篇文章主要介紹了Python數(shù)組遍歷的簡單實現(xiàn)方法,結(jié)合實例總結(jié)分析了Python針對數(shù)組的元素,索引常用遍歷技巧,需要的朋友可以參考下
2016-04-04
利用python代碼寫的12306訂票代碼
這篇文章主要介紹了利用python代碼寫的12306訂票代碼，自己寫的python代碼，是非常實用的技巧,具有一定參考借鑒價值,需要的朋友可以參考下
2015-12-12
python re模塊findall()函數(shù)實例解析
這篇文章主要介紹了python re模塊findall()函數(shù)實例解析，具有一定借鑒價值,需要的朋友可以參考下
2018-01-01
tensorflow 2.0模式下訓(xùn)練的模型轉(zhuǎn)成 tf1.x 版本的pb模型實例
這篇文章主要介紹了tensorflow 2.0模式下訓(xùn)練的模型轉(zhuǎn)成 tf1.x 版本的pb模型實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨想過來看看吧
2020-06-06
python爬取抖音視頻的實例分析
在本篇內(nèi)容里小編給大家整理一篇關(guān)于python爬取抖音視頻的實例分析的相關(guān)內(nèi)容，有興趣的朋友可以測試下實例內(nèi)容。
2021-01-01
Django卸載之后重新安裝的方法
如果你打算從過去的一個版本升級Django, 你需要先刪除老版本的Django之后，再安裝新的版本。下面這篇文章主要給大家介紹了在Django卸載之后重新安裝的方法，文中給出了詳細(xì)的步驟，需要的朋友可以參考借鑒，下面來一起看看吧。
2017-03-03
Expected conditions模塊使用方法匯總代碼解析
這篇文章主要介紹了Expected conditions模塊使用方法匯總代碼解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2020-08-08