如何将自定义图像输入此模型？

Question

我一直在线学习课程，其中一个练习是创建一个简单的图像检测模型（使用MNIST数据）来检测书面数字。我一直试图加载我画的自定义图像（128x128 jpg），但我似乎无法弄明白。我非常接近，但我认为我对模型所采用的参数感到困惑。任何帮助都会受到赞赏!!

Here is my code

Answer 1

只需将图像转换为128x128 numpy数组，其值介于0和1之间。

然后：

image = Variable(torch.from_numpy(image))[None, :, :]
classification = model(image)

classification是一个包含属于每个类的概率的pytorch变量。