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的Django框架中使用SQLAlchemy操作数据库的教程
Jun 02 Python
python PyTorch预训练示例
Feb 11 Python
使用celery执行Django串行异步任务的方法步骤
Jun 06 Python
ubuntu 16.04下python版本切换的方法
Jun 14 Python
使用Python paramiko模块利用多线程实现ssh并发执行操作
Dec 05 Python
Pytorch技巧:DataLoader的collate_fn参数使用详解
Jan 08 Python
Python图像处理库PIL的ImageEnhance模块使用介绍
Feb 26 Python
python shapely.geometry.polygon任意两个四边形的IOU计算实例
Apr 12 Python
Python爬虫实现百度翻译功能过程详解
May 29 Python
python中doctest库实例用法
Dec 31 Python
python制作图形界面的2048游戏, 基于tkinter
Apr 06 Python
基于Python编写简易版的天天跑酷游戏的示例代码
Mar 23 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中防止直接访问或查看或下载config.php文件的方法
2012/07/07 PHP
PHP实现查询手机归属地的方法详解
2017/04/28 PHP
定位地理位置PHP判断员工打卡签到经纬度是否在打卡之内
2019/05/23 PHP
laravel 时间格式转时间戳的例子
2019/10/11 PHP
Javascript的IE和Firefox兼容性汇编
2006/07/01 Javascript
jQuery extend 的简单实例
2013/09/18 Javascript
js加入收藏以及使用Jquery更改透明度
2014/01/26 Javascript
使用js判断当前时区TimeZone是否是夏令时
2014/02/23 Javascript
一些老手都不一定知道的JavaScript技巧
2014/05/06 Javascript
javascript制作游戏开发碰撞检测的封装代码
2015/03/31 Javascript
localResizeIMG先压缩后使用ajax无刷新上传(移动端)
2015/08/11 Javascript
Javascript操作表单实例讲解(下)
2016/06/20 Javascript
Javascript中this绑定的3种方法与比较
2016/10/13 Javascript
微信小程序 表单Form实例详解(附源码)
2016/12/22 Javascript
nodeJS实现路由功能实例代码
2017/06/08 NodeJs
理解Angular的providers给Http添加默认headers
2017/07/04 Javascript
vue router学习之动态路由和嵌套路由详解
2017/09/21 Javascript
React.Js添加与删除onScroll事件的方法详解
2017/11/03 Javascript
微信小程序使用picker实现时间和日期选择框功能【附源码下载】
2017/12/11 Javascript
详解html-webpack-plugin用法全解
2018/01/22 Javascript
微信小程序页面间跳转传参方式总结
2019/06/13 Javascript
electron+vue实现div contenteditable截图功能
2020/01/07 Javascript
VUE和Antv G6实现在线拓扑图编辑操作
2020/10/28 Javascript
[原创]使用豆瓣提供的国内pypi源
2017/07/02 Python
django2 快速安装指南分享
2018/01/05 Python
手把手教你如何安装Pycharm(详细图文教程)
2018/11/28 Python
PIL图像处理模块paste方法简单使用详解
2019/07/17 Python
python matplotlib库绘制散点图例题解析
2019/08/10 Python
python文件读写代码实例
2019/10/21 Python
Python unittest如何生成HTMLTestRunner模块
2020/09/08 Python
python 装饰器重要在哪
2021/02/14 Python
浅谈CSS3鼠标移入图片动态提示效果(transform)
2017/11/06 HTML / CSS
百思买美国官网:Best Buy
2016/07/28 全球购物
写给汽车4S店的创业计划书,拿来即用!
2019/08/09 职场文书
处世之道:关于真诚相待的名言推荐
2019/12/02 职场文书
Win7/8.1用户可以免费升级到Windows 11系统吗?
2021/11/21 数码科技