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抓取京东图书评论数据
Aug 31 Python
在Django框架中运行Python应用全攻略
Jul 17 Python
Windows下Python使用Pandas模块操作Excel文件的教程
May 31 Python
在Python程序和Flask框架中使用SQLAlchemy的教程
Jun 06 Python
django开发之settings.py中变量的全局引用详解
Mar 29 Python
git使用.gitignore设置不生效或不起作用问题的解决方法
Jun 01 Python
使用tensorflow实现AlexNet
Nov 20 Python
Python 经典面试题 21 道【不可错过】
Sep 21 Python
利用python读取YUV文件 转RGB 8bit/10bit通用
Dec 09 Python
Python如何在main中调用函数内的函数方式
Jun 01 Python
python else语句在循环中的运用详解
Jul 06 Python
Python 多进程原理及实现
Dec 21 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 Mysql类 可以参考学习熟悉下
2009/06/21 PHP
php fckeditor 调用的函数
2009/06/21 PHP
php中addslashes函数与sql防注入
2014/11/17 PHP
ThinkPHP添加更新标签的方法
2014/12/05 PHP
PHP实现登录搜狐广告获取广告联盟数据的方法【附demo源码】
2016/10/14 PHP
php+mysql+jquery实现简易的检索自动补全提示功能
2017/04/15 PHP
PHP数组常用函数实例小结
2018/08/20 PHP
js不是基础的基础
2006/12/24 Javascript
Jquery 滑入滑出效果实现代码
2010/03/27 Javascript
jquery选择器原理介绍($()使用方法)
2014/03/25 Javascript
JavaScript基本的输出和嵌入式写法教程
2015/10/20 Javascript
继续学习javascript闭包
2015/12/03 Javascript
js表单提交和submit提交的区别实例分析
2015/12/10 Javascript
Ext JS框架程序中阻止键盘触发回退或者刷新页面的代码分享
2016/06/07 Javascript
JS+HTML5实现的前端购物车功能插件实例【附demo源码下载】
2016/10/17 Javascript
浅谈mint-ui 填坑之路
2017/11/06 Javascript
浅谈React和Redux的连接react-redux
2017/12/04 Javascript
javascript实现摄像头拍照预览
2019/09/30 Javascript
python中文件变化监控示例(watchdog)
2017/10/16 Python
Python实现矩阵加法和乘法的方法分析
2017/12/19 Python
python+matplotlib演示电偶极子实例代码
2018/01/12 Python
学习Python3 Dlib19.7进行人脸面部识别
2018/01/24 Python
Python自定义一个类实现字典dict功能的方法
2019/01/19 Python
Python3.5局部变量与全局变量作用域实例分析
2019/04/30 Python
Pandas之read_csv()读取文件跳过报错行的解决
2020/04/21 Python
Speedo速比涛中国官方网站:全球领先泳装运动品牌
2018/04/24 全球购物
一篇.NET面试题
2014/09/29 面试题
4s店机修工岗位职责
2013/12/20 职场文书
校三好学生主要事迹
2014/01/11 职场文书
关于孝道的演讲稿
2014/05/21 职场文书
护士优质服务演讲稿
2014/08/26 职场文书
信贷客户经理岗位职责
2015/04/09 职场文书
史上最牛辞职信
2015/05/13 职场文书
导游词之铁岭象牙山
2019/12/06 职场文书
详解分布式系统中如何用python实现Paxos
2021/05/18 Python
MySQL 计算连续登录天数
2022/05/11 MySQL