我怎样才能将自定义图像输入到这个模型中？
How could I feed a custom image into this model?
我一直在学习在线课程，其中一个练习是创建一个简单的图像检测模型（使用 MNIST 数据）来检测书写数字。我一直在尝试加载我绘制的自定义图像 (128x128 jpg)，但我似乎无法弄明白。我真的很接近，但我想我只是对模型采用的参数感到困惑。任何帮助将不胜感激！！
只需将您的图像转换为 128x128 值介于 0 和 1 之间的 numpy 数组。
然后：
image = Variable(torch.from_numpy(image))[None, :, :]
classification = model(image)
classification 是一个 pytorch 变量，包含属于每个 class 的概率。
我一直在学习在线课程，其中一个练习是创建一个简单的图像检测模型（使用 MNIST 数据）来检测书写数字。我一直在尝试加载我绘制的自定义图像 (128x128 jpg)，但我似乎无法弄明白。我真的很接近，但我想我只是对模型采用的参数感到困惑。任何帮助将不胜感激！！
只需将您的图像转换为 128x128 值介于 0 和 1 之间的 numpy 数组。
然后：
image = Variable(torch.from_numpy(image))[None, :, :]
classification = model(image)
classification 是一个 pytorch 变量，包含属于每个 class 的概率。