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实现Windows上气泡提醒效果的方法
Jun 03 Python
python的多重继承的理解
Aug 06 Python
微信跳一跳python辅助软件思路及图像识别源码解析
Jan 04 Python
python pyinstaller 加载ui路径方法
Jun 10 Python
Flask-WTF表单的使用方法
Jul 12 Python
python基于socket进行端口转发实现后门隐藏的示例
Jul 25 Python
解析pip安装第三方库但PyCharm中却无法识别的问题及PyCharm安装第三方库的方法教程
Mar 10 Python
500行python代码实现飞机大战
Apr 24 Python
Python字符串split及rsplit方法原理详解
Jun 29 Python
Python如何爬取b站热门视频并导入Excel
Aug 10 Python
Python实现自动签到脚本的示例代码
Aug 19 Python
Python爬取梨视频的示例
Jan 29 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的session cookie错误
2009/08/09 PHP
fleaphp crud操作之find函数的使用方法
2011/04/23 PHP
php设计模式之适配器模式原理、用法及注意事项详解
2019/09/24 PHP
jquery实现手风琴效果实例代码
2013/11/15 Javascript
禁止空格提交表单的js代码
2013/11/17 Javascript
jQuery 中DOM 操作详解
2015/01/13 Javascript
CSS+JS实现点击文字弹出定时自动关闭DIV层菜单的方法
2015/05/12 Javascript
JavaScript中函数(Function)的apply与call理解
2015/07/08 Javascript
详解JavaScript数组的操作大全
2015/10/19 Javascript
使用vue.js开发时一些注意事项
2016/04/27 Javascript
Vue.js每天必学之数据双向绑定
2016/09/05 Javascript
Vue.js 父子组件通讯开发实例
2016/09/06 Javascript
微信小程序--组件(swiper)详细介绍
2017/06/13 Javascript
详解vue-cli官方脚手架配置
2018/07/20 Javascript
vue和better-scroll实现列表左右联动效果详解
2019/04/29 Javascript
如何实现echarts markline标签名显示自己想要的
2020/07/20 Javascript
[01:38]完美世界高校联赛决赛花絮
2018/12/02 DOTA
Python列表list内建函数用法实例分析【insert、remove、index、pop等】
2017/07/24 Python
为什么选择python编程语言入门黑客攻防 给你几个理由!
2018/02/02 Python
widows下安装pycurl并利用pycurl请求https地址的方法
2018/10/15 Python
Python可迭代对象操作示例
2019/05/07 Python
用python3 返回鼠标位置的实现方法(带界面)
2019/07/05 Python
关于PyTorch 自动求导机制详解
2019/08/18 Python
通过python3实现投票功能代码实例
2019/09/26 Python
在OpenCV里使用Camshift算法的实现
2019/11/22 Python
python 判断txt每行内容中是否包含子串并重新写入保存的实例
2020/03/12 Python
Java的五个基础面试题
2016/02/26 面试题
南京软件公司的.net程序员笔试题
2014/08/31 面试题
医院护士的求职信范文
2013/12/26 职场文书
优秀毕业生事迹材料
2014/02/12 职场文书
工作汇报开头与结尾怎么写
2014/11/08 职场文书
保护环境的宣传语
2015/07/13 职场文书
2015年音乐教学工作总结
2015/07/22 职场文书
医德医风学习心得体会
2016/01/25 职场文书
python开发制作好看的时钟效果
2022/05/02 Python
Linux磁盘管理方法介绍
2022/06/01 Servers