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获取脚本所在目录的正确方法
Apr 15 Python
Python文档生成工具pydoc使用介绍
Jun 02 Python
使用python实现tcp自动重连
Jul 02 Python
Python最火、R极具潜力 2017机器学习调查报告
Dec 11 Python
Python编程pygame模块实现移动的小车示例代码
Jan 03 Python
Python对CSV、Excel、txt、dat文件的处理
Sep 18 Python
Python最小二乘法矩阵
Jan 02 Python
python设定并获取socket超时时间的方法
Jan 12 Python
Python实现栈和队列的简单操作方法示例
Nov 29 Python
关于sys.stdout和print的区别详解
Dec 05 Python
Pandas 解决dataframe的一列进行向下顺移问题
Dec 27 Python
python3列表删除大量重复元素remove()方法的问题详解
Jan 04 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版)
2012/08/21 PHP
php调整gif动画图片尺寸示例代码分享
2013/12/05 PHP
php获取根域名方法汇总
2014/10/28 PHP
33道php常见面试题及答案
2015/07/06 PHP
Zend Studio使用技巧两则
2016/04/01 PHP
浅谈PHP链表数据结构(单链表)
2016/06/08 PHP
JavaScript 继承详解(三)
2009/07/13 Javascript
兼容最新firefox、chrome和IE的javascript图片预览实现代码
2014/08/08 Javascript
javascript最基本的函数汇总
2015/06/25 Javascript
jQuery实现动态添加和删除一个div
2015/08/12 Javascript
js实现简洁大方的二级下拉菜单效果代码
2015/09/01 Javascript
javascript实现五星评分功能
2015/11/10 Javascript
理解javascript模块化
2016/03/28 Javascript
jQ处理xml文件和xml字符串的方法(详解)
2016/11/22 Javascript
js实现复选框的全选和取消全选效果
2017/01/03 Javascript
详谈js中数组(array)和对象(object)的区别
2017/02/27 Javascript
js实现倒计时关键代码
2017/05/05 Javascript
关于meta viewport中target-densitydpi属性详解(推荐)
2017/08/18 Javascript
vue将单页面改造成多页面应用的方法
2018/11/25 Javascript
Vue中this.$nextTick的作用及用法
2020/02/04 Javascript
微信小程序实现点赞业务
2021/02/10 Javascript
python统计cpu利用率的方法
2015/06/02 Python
python3.6使用pymysql连接Mysql数据库
2018/05/25 Python
python读取并写入mat文件的方法
2019/07/12 Python
pandas的相关系数与协方差实例
2019/12/27 Python
PyCharm最新激活码(2020/10/27全网最新)
2020/10/27 Python
TripAdvisor越南:全球领先的旅游网站
2017/09/21 全球购物
自我评价范文点评
2013/12/04 职场文书
学校七一活动方案
2014/01/19 职场文书
2014年教师批评与自我批评思想汇报
2014/09/20 职场文书
医院领导班子整改方案
2014/10/01 职场文书
新员工考核评语
2014/12/31 职场文书
英语辞职信范文
2015/02/28 职场文书
慈善献爱心倡议书
2015/04/27 职场文书
关于环保的广播稿
2015/12/17 职场文书
详解前端任务构建利器Gulp.js使用指南
2021/04/30 Javascript