pytorch获取vgg16-feature层输出的例子


Posted in Python onAugust 20, 2019

实际应用时可能比较想获取VGG中间层的输出,

那么就可以如下操作:

import numpy as np
import torch
from torchvision import models
from torch.autograd import Variable
import torchvision.transforms as transforms
 
 
class CNNShow():
  def __init__(self, model):
    self.model = model
    self.model.eval()
 
    self.created_image = self.image_for_pytorch(np.uint8(np.random.uniform(150, 180, (224, 224, 3))))
 
 
  def show(self):
    x = self.created_image
    for index, layer in enumerate(self.model):
      print(index,layer)
      x = layer(x)
 
  def image_for_pytorch(self,Data):
    transform = transforms.Compose([
      transforms.ToTensor(), # range [0, 255] -> [0.0,1.0]
      transforms.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225))
    ]
    )
    imData = transform(Data)
    imData = Variable(torch.unsqueeze(imData, dim=0), requires_grad=True)
    return imData
 
if __name__ == '__main__':
 
  pretrained_model = models.vgg16(pretrained=True).features
  CNN = CNNShow(pretrained_model)
  CNN.show()

以上这篇pytorch获取vgg16-feature层输出的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python中sleep函数用法实例分析
Apr 29 Python
在Python的Flask框架中构建Web表单的教程
Jun 04 Python
python实现应用程序在右键菜单中添加打开方式功能
Jan 09 Python
python实现图书管理系统
Mar 12 Python
Python数据分析库pandas基本操作方法
Apr 08 Python
查看django版本的方法分享
May 14 Python
在scrapy中使用phantomJS实现异步爬取的方法
Dec 17 Python
在Python中os.fork()产生子进程的例子
Aug 08 Python
python定时任务 sched模块用法实例
Nov 04 Python
解决pycharm 安装numpy失败的问题
Dec 05 Python
Pycharm中安装wordcloud等库失败问题及终端通过pip安装的Python库如何添加到Pycharm解释器中(推荐)
May 10 Python
requests在python中发送请求的实例讲解
Feb 17 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
python爬虫 urllib模块url编码处理详解
Aug 20 #Python
pytorch实现用Resnet提取特征并保存为txt文件的方法
Aug 20 #Python
You might like
修改PHP的memory_limit限制的方法分享
2012/02/21 PHP
php最简单的删除目录与文件实现方法
2014/11/28 PHP
PHP文件上传、客户端和服务器端加限制、抓取错误信息、完整步骤解析
2017/01/12 PHP
php封装的mongodb操作类代码
2017/08/06 PHP
开发跨浏览器javascript常见注意事项
2009/01/01 Javascript
jQuery 翻牌或百叶窗效果(内容三秒自动切换)
2012/06/14 Javascript
jQuery链式操作如何实现以及为什么要用链式操作
2013/01/17 Javascript
将JavaScript的jQuery库中表单转化为JSON对象的方法
2015/11/17 Javascript
javascript中eval解析JSON字符串
2016/02/27 Javascript
在AngularJS框架中处理数据建模的方式解析
2016/03/05 Javascript
jQuery实现最简单实用的分秒倒计时
2017/02/05 Javascript
基于angular2 的 http服务封装的实例代码
2017/06/29 Javascript
Angular 2父子组件数据传递之@Input和@Output详解 (上)
2017/07/05 Javascript
JS加密插件CryptoJS实现的DES加密示例
2018/08/16 Javascript
jQuery动态操作表单示例【基于table表格】
2018/12/06 jQuery
JS+canvas画布实现炫酷的旋转星空效果示例
2019/02/13 Javascript
JS实现鼠标移动拖尾
2020/12/27 Javascript
[00:35]可解锁地面特效
2018/12/20 DOTA
[05:15]2018年度CS GO社区贡献奖-完美盛典
2018/12/16 DOTA
Python内置数据类型详解
2014/08/18 Python
Python简单实现安全开关文件的两种方式
2016/09/19 Python
http请求 request失败自动重新尝试代码示例
2018/01/25 Python
python实现图像拼接
2020/03/05 Python
解决Python中导入自己写的类,被划红线,但不影响执行的问题
2020/07/13 Python
分享CSS3中必须要知道的10个顶级命令
2012/04/26 HTML / CSS
中国网上药店领导者:1药网
2017/02/16 全球购物
Tenstickers法国:墙贴和装饰贴纸
2019/08/26 全球购物
全球烹饪课程的领先预订平台:Cookly
2020/01/28 全球购物
20岁生日感言
2014/01/13 职场文书
2014自荐信的写作技巧
2014/01/28 职场文书
亲子读书活动方案
2014/02/22 职场文书
出纳会计岗位职责
2014/03/12 职场文书
关于法制教育的宣传语
2015/07/13 职场文书
2015年卫生局工作总结
2015/07/24 职场文书
生活委员竞选稿
2015/11/21 职场文书
Vue实现导入Excel功能步骤详解
2021/07/03 Vue.js