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)实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了全网最详细的PyCharm+Anaconda的安装过程图解,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-01-01
这篇文章主要为大家介绍了LeetCode189轮转数组python解法示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-08-08
os.environ是Python中的一个字典对象，它包含了当前进程的环境变量，通过使用os.environ，可以获取和设置环境变量的值，这篇文章主要介绍了python中os.environ基本介绍及使用方法,需要的朋友可以参考下2023-08-08
这篇文章主要给大家介绍了将python中nan与inf转为特定数字的方法，文中给出了详细的示例代码和运行结果，对大家的理解和学习具有一定的参考学习价值，需要的朋友们下面来一起看看吧。2017-05-05
这篇文章主要介绍了如何基于Django实现上下文章跳转,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-09-09
这篇文章主要介绍了解决Python访问MySQL数据库速度慢的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-04-04
这篇文章主要介绍了python调试神器PySnooper的使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
python selenium执行所有测试用例并生成报告的方法
今天小编就为大家分享一篇python selenium执行所有测试用例并生成报告的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
这篇文章主要介绍了python装饰器decorator介绍,decorator设计模式允许动态地对现有的对象或函数包装以至于修改现有的职责和行为,简单地讲用来动态地扩展现有的功能,需要的朋友可以参考下2014-11-11
今天小编就为大家分享一篇对python:print打印时加u的含义详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
最新评论