pytorch中的inference使用实例


Posted in Python onFebruary 20, 2020

这里inference两个程序的连接,如目标检测,可以利用一个程序提取候选框,然后把候选框输入到分类cnn网络中。

这里常需要进行一定的连接。

#加载训练好的分类CNN网络
model=torch.load('model.pkl')

#假设proposal_img是我们提取的候选框,是需要输入到CNN网络的数据

#先定义transforms对输入cnn的网络数据进行处理,常包括resize、totensor等操作
data_transforms=transforms.Compose([transforms.RandomSizedCrop(224),
transforms.ToTensor()])

#由于transforms是对PIL格式数据操作,所以必要时转化格式

def tensor_to_PIL(tensor):
 image = tensor.cpu().clone()
 image = image.squeeze(0)
 image = unloader(image)
 return image
 
#unqueeze(0)是加多一维,对应原来batchsiaze
data=data_transforms(proposal_img).unqueeze(0)

#新版本pytorch已经不用variable,可以省略这句
data=Variable(data)
#貌似这句也是多余的
torch.no_grad()

predict=F.softmax(model(data.cuda()).cuda())

以上这篇pytorch中的inference使用实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python笔记(2)
Oct 24 Python
Python中for循环详解
Jan 17 Python
Python调用C语言开发的共享库方法实例
Mar 18 Python
举例讲解如何在Python编程中进行迭代和遍历
Jan 19 Python
对python中raw_input()和input()的用法详解
Apr 22 Python
Python嵌套列表转一维的方法(压平嵌套列表)
Jul 03 Python
Python 获取项目根路径的代码
Sep 27 Python
详解Django将秒转换为xx天xx时xx分
Sep 27 Python
Django与pyecharts结合的实例代码
May 13 Python
Django返回HTML文件的实现方法
Sep 17 Python
Ubuntu20.04环境安装tensorflow2的方法步骤
Jan 29 Python
提取视频中的音频 Python只需要三行代码!
May 10 Python
python encrypt 实现AES加密的实例详解
Feb 20 #Python
Python关于反射的实例代码分享
Feb 20 #Python
Python3监控疫情的完整代码
Feb 20 #Python
浅谈Pytorch torch.optim优化器个性化的使用
Feb 20 #Python
关于torch.optim的灵活使用详解(包括重写SGD,加上L1正则)
Feb 20 #Python
Python sys模块常用方法解析
Feb 20 #Python
pytorch 实现在一个优化器中设置多个网络参数的例子
Feb 20 #Python
You might like
显示程序执行时间php函数代码
2013/08/29 PHP
thinkphp实现多语言功能(语言包)
2014/03/04 PHP
PHP+swoole实现简单多人在线聊天群发
2016/01/19 PHP
php数组函数array_walk用法示例
2016/05/26 PHP
extjs 学习笔记(三) 最基本的grid
2009/10/15 Javascript
DLL+ ActiveX控件+WEB页面调用例子
2010/08/07 Javascript
C#中TrimStart,TrimEnd,Trim在javascript上的实现
2011/01/17 Javascript
深入理解js函数的作用域与this指向
2016/05/28 Javascript
详解bootstrap的modal-remote两种加载方式【强化】
2017/01/27 Javascript
js实现图片旋转 js滚动鼠标中间对图片放大缩小
2017/07/05 Javascript
javascript定时器取消定时器及优化方法
2017/07/08 Javascript
JavaScript严格模式下关于this的几种指向详解
2017/07/12 Javascript
微信小程序实现的picker多级联动功能示例
2019/05/23 Javascript
vue element upload组件 file-list的动态绑定实现
2019/10/11 Javascript
vue路由切换之淡入淡出的简单实现
2019/10/31 Javascript
Python生成不重复随机值的方法
2015/05/11 Python
Eclipse中Python开发环境搭建简单教程
2016/03/23 Python
Python内置函数 next的具体使用方法
2017/11/24 Python
python 按照固定长度分割字符串的方法小结
2018/04/30 Python
python 用正则表达式筛选文本信息的实例
2018/06/05 Python
对Tensorflow中的变量初始化函数详解
2018/07/27 Python
python实现的汉诺塔算法示例
2019/10/23 Python
使用Python实现分别输出每个数组
2019/12/06 Python
python 解决cv2绘制中文乱码问题
2019/12/23 Python
基于MSELoss()与CrossEntropyLoss()的区别详解
2020/01/02 Python
使用IPython或Spyder将省略号表示的内容完整输出
2020/04/20 Python
解决Jupyter Notebook使用parser.parse_args出现错误问题
2020/04/20 Python
pymysql之cur.fetchall() 和cur.fetchone()用法详解
2020/05/15 Python
2019年Java面试必问之经典试题
2012/09/12 面试题
Linux如何为某个操作添加别名
2013/03/01 面试题
劳动实践课感言
2014/02/01 职场文书
致跳远运动员广播稿
2014/02/11 职场文书
低碳日宣传活动总结
2014/07/09 职场文书
委托书怎么写
2014/07/31 职场文书
感谢信怎么写
2015/01/21 职场文书
2015年预算员工作总结
2015/05/14 职场文书