Pytorch抽取网络层的Feature Map(Vgg)实例


Posted in Python onAugust 20, 2019

这边我是需要得到图片在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 相关文章推荐
videocapture库制作python视频高速传输程序
Dec 23 Python
Python实现的飞速中文网小说下载脚本
Apr 23 Python
Python实现OpenCV的安装与使用示例
Mar 30 Python
解决Python运行文件出现out of memory框的问题
Dec 03 Python
使用Python+wxpy 找出微信里把你删除的好友实例
Feb 21 Python
python命令 -u参数用法解析
Oct 24 Python
Python API len函数操作过程解析
Mar 05 Python
使用 Python ssh 远程登陆服务器的最佳方案
Mar 06 Python
详解Windows下PyCharm安装Numpy包及无法安装问题解决方案
Jun 18 Python
Python如何实现后端自定义认证并实现多条件登陆
Jun 22 Python
Python在字符串中处理html和xml的方法
Jul 31 Python
matplotlib之pyplot模块实现添加子图subplot的使用
Apr 25 Python
python批量解压zip文件的方法
Aug 20 #Python
pytorch获取vgg16-feature层输出的例子
Aug 20 #Python
python 并发编程 非阻塞IO模型原理解析
Aug 20 #Python
Python实现某论坛自动签到功能
Aug 20 #Python
python函数的作用域及关键字详解
Aug 20 #Python
pytorch 可视化feature map的示例代码
Aug 20 #Python
python爬虫 基于requests模块的get请求实现详解
Aug 20 #Python
You might like
遍历指定目录下的所有目录和文件的php代码
2011/11/27 PHP
php注销代码(session注销)
2012/05/31 PHP
PHP中IP地址与整型数字互相转换详解
2014/08/20 PHP
php项目开发中用到的快速排序算法分析
2016/06/25 PHP
js或css实现滚动广告的几种方案
2010/01/28 Javascript
JQuery调webservice实现邮箱验证(检测是否可用)
2013/05/21 Javascript
关于jQuery判断元素是否存在的问题示例探讨
2014/07/21 Javascript
Bootstrap每天必学之模态框(Modal)插件
2016/04/26 Javascript
AngularJs表单验证实例详解
2016/05/30 Javascript
最丑的时钟效果!js canvas时钟制作方法
2016/08/15 Javascript
Angularjs CURD 详解及实例代码
2016/09/14 Javascript
用自定义图片代替原生checkbox实现全选,删除以及提交的方法
2016/10/18 Javascript
JS对象与JSON互转换、New Function()、 forEach()、DOM事件流等js开发基础小结
2017/08/10 Javascript
微信小程序的生命周期的详解
2017/10/19 Javascript
微信小程序开发问题之wx.previewImage
2018/12/25 Javascript
浅谈vue websocket nodeJS 进行实时通信踩到的坑
2020/09/22 NodeJs
[01:15:12]DOTA2上海特级锦标赛主赛事日 - 1 败者组第一轮#4Newbee VS CDEC
2016/03/03 DOTA
Python 条件判断的缩写方法
2008/09/06 Python
Python实战小程序利用matplotlib模块画图代码分享
2017/12/09 Python
python编写Logistic逻辑回归
2020/12/30 Python
Python中的TCP socket写法示例
2018/05/11 Python
Python读取mat文件,并转为csv文件的实例
2018/07/04 Python
Python在for循环中更改list值的方法【推荐】
2018/08/17 Python
Django框架文件上传与自定义图片上传路径、上传文件名操作分析
2019/05/10 Python
Python3视频转字符动画的实例代码
2019/08/29 Python
python super的使用方法及实例详解
2019/09/25 Python
python中threading开启关闭线程操作
2020/05/02 Python
Tensorflow加载Vgg预训练模型操作
2020/05/26 Python
Python实现壁纸下载与轮换
2020/10/19 Python
机械设计毕业生自荐信
2014/02/02 职场文书
爱国卫生月活动总结范文
2014/04/25 职场文书
电子商务专业求职信
2014/07/10 职场文书
预备党员期盼十八届四中全会召开思想汇报
2014/10/17 职场文书
2019大学毕业晚会主持词
2019/06/21 职场文书
SQL模糊查询报:ORA-00909:参数个数无效问题的解决
2021/06/21 Oracle
SQL Server表分区降低运维和维护成本
2022/04/08 SQL Server