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出现segfault错误解决方法
Apr 16 Python
Python读取一个目录下所有目录和文件的方法
Jul 15 Python
python制作爬虫爬取京东商品评论教程
Dec 16 Python
Python安装Numpy和matplotlib的方法(推荐)
Nov 02 Python
Python实现的求解最大公约数算法示例
May 03 Python
Python RabbitMQ消息队列实现rpc
May 30 Python
python 按不同维度求和,最值,均值的实例
Jun 28 Python
Win10下python3.5和python2.7环境变量配置教程
Sep 18 Python
很酷的python表白工具 你喜欢我吗
Apr 11 Python
python selenium操作cookie的实现
Mar 18 Python
Python requests上传文件实现步骤
Sep 15 Python
python四个坐标点对图片区域最小外接矩形进行裁剪
Jun 04 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中的foreach函数
2013/08/31 PHP
php递归删除目录下的文件但保留的实例分享
2014/05/10 PHP
PHP中模拟处理HTTP PUT请求的例子
2014/07/22 PHP
JavaScript 格式字符串的应用
2010/03/29 Javascript
用JQuery模仿淘宝的图片放大镜显示效果
2011/09/15 Javascript
网站接入QQ登录的两种方法
2014/07/22 Javascript
在JavaScript中操作数组之map()方法的使用
2015/06/09 Javascript
js操作table元素实现表格行列新增、删除技巧总结
2015/11/18 Javascript
基于javascript实现随机颜色变化效果
2016/01/14 Javascript
jquery分页插件jquery.pagination.js实现无刷新分页
2016/04/01 Javascript
JavaScript中的prototype原型学习指南
2016/05/09 Javascript
简单实现轮播图效果的实例
2016/07/15 Javascript
KnockoutJS 3.X API 第四章之表单value绑定
2016/10/10 Javascript
bootstrap datetimepicker日期插件超详细使用方法介绍
2017/02/23 Javascript
vue2.0结合DataTable插件实现表格动态刷新的方法详解
2017/03/17 Javascript
浅谈Angular2 ng-content 指令在组件中嵌入内容
2017/08/18 Javascript
vue-router项目实战总结篇
2018/02/11 Javascript
JavaScript设计模式之单例模式简单实例教程
2018/07/02 Javascript
JS使用new操作符创建对象的方法分析
2019/05/30 Javascript
elementui之el-tebs浏览器卡死的问题和使用报错未注册问题
2019/07/06 Javascript
vue中filters 传入两个参数 / 使用两个filters的实现方法
2019/07/15 Javascript
Python中每次处理一个字符的5种方法
2015/05/21 Python
python 中的divmod数字处理函数浅析
2017/10/17 Python
Python深度优先算法生成迷宫
2018/01/22 Python
python3+PyQt5重新实现QT事件处理程序
2018/04/19 Python
python3.4爬虫demo
2019/01/22 Python
win10安装python3.6的常见问题
2020/07/01 Python
HTML5+CSS3实例 :canvas 模拟实现电子彩票刮刮乐代码
2016/12/30 HTML / CSS
使用HTML5做个画图板的方法介绍
2013/05/03 HTML / CSS
国际书籍零售商:Wordery
2017/11/01 全球购物
党员查摆剖析材料
2014/10/10 职场文书
个人务虚会发言材料
2014/10/20 职场文书
党的群众路线教育实践活动个人对照检查材料(企业)
2014/11/05 职场文书
2016党校学习心得体会
2016/01/07 职场文书
SQL实现LeetCode(180.连续的数字)
2021/08/04 MySQL
浅谈MySql整型索引和字符串索引失效或隐式转换问题
2021/11/20 MySQL