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

Posted 2021-04-25

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

Here is my code

答案

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

然后：

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

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