Resnet for Fashion_Mnist（四）

最新推荐文章于 2024-01-23 16:34:13 发布

lai199508

最新推荐文章于 2024-01-23 16:34:13 发布

阅读量440

点赞数

分类专栏：深度学习 pytorch 计算机视觉 Resnet 文章标签：计算机视觉 pytorch FashionMnist

本文链接：https://blog.csdn.net/lai199508/article/details/85039967

版权

Resnet for Fashion_Mnist（四）

模型加载及预测数据

下面给出模型加载及预测数据的代码，预测结果保存到csv文件中。

# -*- coding: utf-8 -*-
import torch
import MyDataSet
from torchvision import transforms
from torch.autograd import Variable
import csv
import model

# 重新加载模型
path = 'E:/Fashion_Mnist/saved-models/resnet18-run-26.pth.tar'

# Set up the device
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print('Testing on {}'.format(device))

# Set seeds. If using numpy this must be seeded too.
torch.man