pytorch中的inference使用实例
作者:Star_ACE 时间:2023-07-24 16:22:56
这里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())
来源:https://blog.csdn.net/qq_34886403/article/details/87833000
标签:pytorch,inference
0
投稿
猜你喜欢
Python anaconda安装库命令详解
2023-01-31 16:36:36
使用MySQL的geometry类型处理经纬度距离问题的方法
2024-01-21 15:44:06
Python根据当前日期取去年同星期日期
2021-09-14 15:01:48
Jupyter notebook如何修改平台字体
2022-03-21 13:37:59
python使用rpc框架gRPC的方法
2021-02-28 06:28:37
Git原理和常用操作
2023-02-01 23:55:55
Discuz!NT 论坛整合ASP程序论坛
2011-03-31 10:40:00
python实现集中式的病毒扫描功能详解
2022-03-04 03:16:00
给Linux定时备份数据库的实现脚本
2024-01-15 09:15:22
Django单元测试工具test client使用详解
2021-04-11 22:25:58
MySQL定期分析检查与优化表的方法小结
2024-01-24 15:04:14
python重复值处理得方法
2023-07-03 20:57:43
油猴脚本编写教程详解
2023-05-26 12:29:51
用JS实现一个页面两份样式表
2008-07-23 12:37:00
python爬取梨视频生活板块最热视频
2023-12-30 09:38:56
解决Mac node版本升级失败的问题
2024-05-13 09:34:52
python防止栈溢出的实例讲解
2022-07-31 17:17:15
教你精确编写高质量高性能的MySQL语法
2009-01-14 12:57:00
go按行读取文件的三种实现方式汇总
2024-04-25 15:08:15
python使用ctypes调用扩展模块的实例方法
2021-11-01 22:22:58