pytorch框架--简单模型预测

模型预测示例

使用训练好的模型进行预测

import torchvisionfrom model import Tudui
import torch
from PIL import Image# 读取图像
img = Image.open("./data/train/Dog/9.jpg")
# 数据预处理# 缩放
transform = torchvision.transforms.Compose([torchvision.transforms.Resize((32, 32)),torchvision.transforms.ToTensor()])
image = transform(img)
print(image.shape)# 根据保存方式加载
model = torch.load("tudui_99.pth", map_location=torch.device('cpu'))# 注意维度转换，单张图片
image1 = torch.reshape(image, (1, 3, 32, 32))# 测试开关
model.eval()
# 节约性能
with torch.no_grad():output = model(image1)
print(output)
# print(output.argmax(1))
# 定义类别对应字典
dist = {0: "飞机", 1: "汽车", 2: "鸟", 3: "猫", 4: "鹿", 5: "狗", 6: "青蛙", 7: "马", 8: "船", 9: "卡车"}
# 转numpy格式,列表内取第一个
a = dist[output.argmax(1).numpy()[0]]
img.show()
print(a)

pytorch框架--简单模型预测相关推荐

使用Pytorch自带模型预测图片
这里要用到pytorch,所以得在前面博客基础上安装下,安装方法见博主的博客 ultralytics/yolov3训练预测自己数据集的配置过程_竹叶青lvye的博客-CSDN博客_ultralytic ...
Pytorch框架TorchScript模型转换方法
为什么要使用TorchScript对模型进行转换? a).TorchScript代码可以在它自己的解释器中调用,它本质上是一个受限的Python解释器.这个解释器不获取全局解释器锁,因此可以在同一个实 ...
【CANN训练营第三季】基于昇腾PyTorch框架的模型训练调优
文章目录性能分析工具PyTorch Profiling 性能分析工具CANN Profiling 结业考核 1.使用Pytorch实现LeNet网络的minist手写数字识别. 2.采用课程中学习到 ...
pytorch-softmax解决分类问题，用fashion-mnist为例子，再走一遍数据获取到模型预测的流程。深度了解分类指标的递进关系
softmax回归线性回归模型适用于输出为连续值的情景.在另一类情景中,模型输出可以是一个像图像类别这样的离散值.对于这样的离散值预测问题,我们可以使用诸如softmax回归在内的分类模型.和线性回 ...
如何从零开始设计与开发一款通用模型预测调度系统 | 干货分享
机器学习模型从训练导出到生成环境部署,这个过程中涉及大量工作,会面临着各种问题与挑战,比如不断丰富的业务场景,系统需要负载种类繁多的神经网络,预测任务不均匀等,然而企业拥有的计算资源(如GPU个数)是 ...
Facebook 万字长文：AI 模型全部迁移至 PyTorch 框架
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达转自 | 新智元来源 | Facebook AI Blog 编辑 ...
Facebook万字长文：AI模型全部迁移至PyTorch框架
来源:新智元本文约3800字,建议阅读8分钟PyTorch 占领 Facebook. [ 导读 ] 2017年,PyTorch诞生,成为当下最流行的深度学习框架.近日,Facebook宣布让PyTor ...
DL之Keras：基于Keras框架建立模型实现【预测】功能的简介、设计思路、案例分析、代码实现之详细攻略(经典，建议收藏)
DL之Keras:基于Keras框架建立模型实现[预测]功能的简介.设计思路.案例分析.代码实现之详细攻略(经典,建议收藏) 目录 Keras框架使用分析 Keras框架设计思路案例分析代码实现 ...
净迁移人口预测程序python_高质量深度学习模型，一键模型预测，迁移学习很简单...
飞桨(PaddlePaddle)核心框架Paddle Fluid v1.5已经发布,而作为其关键工具,用来迁移学习的PaddleHub也进行了全面更新,正式发布了1.0版本. 全新的PaddleHub ...

pytorch框架--简单模型预测

模型预测示例

pytorch框架--简单模型预测相关推荐

最新文章

热门文章