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的Flask框架的简介和安装方法
Nov 13 Python
django model去掉unique_together报错的解决方案
Oct 18 Python
Python连接phoenix的方法示例
Sep 29 Python
用Python登录好友QQ空间点赞的示例代码
Nov 04 Python
VSCode下配置python调试运行环境的方法
Apr 06 Python
Python3实现的简单验证码识别功能示例
May 02 Python
Python3使用SMTP发送带附件邮件
Jun 16 Python
python五子棋游戏的设计与实现
Jun 18 Python
python list的index()和find()的实现
Nov 16 Python
cookies应对python反爬虫知识点详解
Nov 25 Python
Python利用zhdate模块实现农历日期处理
Mar 31 Python
Python中的 No Module named ***问题及解决
Jul 23 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中将网址转换为超链接的函数
2011/09/02 PHP
采集邮箱的php代码(抓取网页中的邮箱地址)
2012/07/17 PHP
XAMPP安装与使用方法详细解析
2013/11/27 PHP
PHP5各个版本的新功能和新特性总结
2014/03/16 PHP
php使用mysqli和pdo扩展,测试对比连接mysql数据库的效率完整示例
2019/05/09 PHP
监控 url fragment变化的js代码
2010/04/19 Javascript
hover的用法及live的用法介绍(鼠标悬停效果)
2013/03/29 Javascript
jquery的map与get方法详解
2013/11/04 Javascript
JS保留两位小数 四舍五入函数的小例子
2013/11/20 Javascript
简单的ajax连接库分享(不用jquery的ajax)
2014/01/19 Javascript
DOM基础教程之事件对象
2015/01/20 Javascript
使用JavaScript实现弹出层效果的简单实例
2016/05/31 Javascript
jQuery基础_入门必看知识点
2016/07/04 Javascript
jquery 实现回车登录详解及实例代码
2016/10/23 Javascript
用director.js实现前端路由使用实例
2017/01/27 Javascript
JavaWeb表单及时验证功能在输入后立即验证(含用户类型,性别,爱好...的验证)
2017/06/09 Javascript
Three.js基础学习之场景对象
2017/09/27 Javascript
bootstrap日期插件daterangepicker使用详解
2017/10/19 Javascript
Angular @HostBinding()和@HostListener()用法
2018/03/05 Javascript
vue.js删除列表中的一行
2018/06/30 Javascript
vue使用websocket的方法实例分析
2019/06/22 Javascript
JS中call()和apply()的功能及用法实例分析
2019/06/28 Javascript
layui中select,radio设置不生效的解决方法
2019/09/05 Javascript
[02:53]DOTA2英雄昆卡基础教程
2013/11/25 DOTA
Python中优化NumPy包使用性能的教程
2015/04/23 Python
python使用reportlab实现图片转换成pdf的方法
2015/05/22 Python
PyQt5多线程刷新界面防假死示例
2019/12/13 Python
python中os包的用法
2020/06/01 Python
Python生成pdf目录书签的实例方法
2020/10/29 Python
整理的15个非常有用的 HTML5 开发教程和速查手册
2011/10/18 HTML / CSS
伦敦鲜花递送:Flower Station
2021/02/03 全球购物
鲜花方阵解说词
2014/02/13 职场文书
小小商店教学反思
2014/04/27 职场文书
2014年后备干部工作总结
2014/12/08 职场文书
班主任工作总结范文
2015/08/13 职场文书
Spring Cloud 中@FeignClient注解中的contextId属性详解
2021/09/25 Java/Android