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ì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
這邊我是需要得到圖片在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)文章
全網(wǎng)最詳細(xì)的PyCharm+Anaconda的安裝過(guò)程圖解
這篇文章主要介紹了全網(wǎng)最詳細(xì)的PyCharm+Anaconda的安裝過(guò)程圖解,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-01-01
LeetCode189輪轉(zhuǎn)數(shù)組python示例
這篇文章主要為大家介紹了LeetCode189輪轉(zhuǎn)數(shù)組python解法示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-08-08
os.environ是Python中的一個(gè)字典對(duì)象，它包含了當(dāng)前進(jìn)程的環(huán)境變量，通過(guò)使用os.environ，可以獲取和設(shè)置環(huán)境變量的值，這篇文章主要介紹了python中os.environ基本介紹及使用方法,需要的朋友可以參考下2023-08-08
python中nan與inf轉(zhuǎn)為特定數(shù)字方法示例
這篇文章主要給大家介紹了將python中nan與inf轉(zhuǎn)為特定數(shù)字的方法，文中給出了詳細(xì)的示例代碼和運(yùn)行結(jié)果，對(duì)大家的理解和學(xué)習(xí)具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起看看吧。2017-05-05
如何基于Django實(shí)現(xiàn)上下文章跳轉(zhuǎn)
這篇文章主要介紹了如何基于Django實(shí)現(xiàn)上下文章跳轉(zhuǎn),文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-09-09
解決Python訪問(wèn)MySQL數(shù)據(jù)庫(kù)速度慢的問(wèn)題
這篇文章主要介紹了解決Python訪問(wèn)MySQL數(shù)據(jù)庫(kù)速度慢的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-04-04
這篇文章主要介紹了python調(diào)試神器PySnooper的使用，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
python selenium執(zhí)行所有測(cè)試用例并生成報(bào)告的方法
今天小編就為大家分享一篇python selenium執(zhí)行所有測(cè)試用例并生成報(bào)告的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-02-02
這篇文章主要介紹了python裝飾器decorator介紹,decorator設(shè)計(jì)模式允許動(dòng)態(tài)地對(duì)現(xiàn)有的對(duì)象或函數(shù)包裝以至于修改現(xiàn)有的職責(zé)和行為,簡(jiǎn)單地講用來(lái)動(dòng)態(tài)地?cái)U(kuò)展現(xiàn)有的功能,需要的朋友可以參考下2014-11-11
對(duì)python:print打印時(shí)加u的含義詳解
今天小編就為大家分享一篇對(duì)python:print打印時(shí)加u的含義詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12