Pytorch抽取网络层的Feature Map(Vgg)实例
更新时间：2019年08月20日 10:51:32 作者：seeney
今天小编就为大家分享一篇Pytorch抽取网络层的Feature Map(Vgg)实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
这边我是需要得到图片在Vgg的5个block里relu后的Feature Map （其余网络只需要替换就可以了）
索引可以这样获得
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抽取网络层的Feature Map(Vgg)实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python下实现的RSA加密/解密及签名/验证功能,结合具体实例形式分析了Python中RSA加密、解密的实现方法及签名、验证功能的使用技巧,需要的朋友可以参考下2017-07-07
这篇文章主要介绍了Python文本生成的Beam Search解码,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-07-07
这篇文章主要介绍了Python多进程并发与多线程并发编程,结合实例形式总结分析了Python编程中的多进程并发与多线程并发相关概念、使用方法与操作注意事项,需要的朋友可以参考下2018-02-02
这篇文章主要介绍了Python数组遍历的简单实现方法,结合实例总结分析了Python针对数组的元素,索引常用遍历技巧,需要的朋友可以参考下2016-04-04
这篇文章主要介绍了利用python代码写的12306订票代码，自己写的python代码，是非常实用的技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-12-12
这篇文章主要介绍了python re模块findall()函数实例解析，具有一定借鉴价值,需要的朋友可以参考下2018-01-01
tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例
这篇文章主要介绍了tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例，具有很好的参考价值，希望对大家有所帮助。一起跟随想过来看看吧2020-06-06
在本篇内容里小编给大家整理一篇关于python爬取抖音视频的实例分析的相关内容，有兴趣的朋友可以测试下实例内容。2021-01-01
如果你打算从过去的一个版本升级Django, 你需要先删除老版本的Django之后，再安装新的版本。下面这篇文章主要给大家介绍了在Django卸载之后重新安装的方法，文中给出了详细的步骤，需要的朋友可以参考借鉴，下面来一起看看吧。2017-03-03
Expected conditions模块使用方法汇总代码解析
这篇文章主要介绍了Expected conditions模块使用方法汇总代码解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
最新评论