用MNIST的训练集训练的LeNet,这里就不展示代码了
直接加载保存好的模型
lenet = torch.load('resourses/trained_model/LeNet_trained.pkl')
测试代码附上
print("Testing")
#定义转换操作
#将测试图片读入并transform后传入模型
test_images = Image.open('resourses/LeNet_test/0.png')
img_to_tensor = transforms.Compose([
transforms.Resize(