如何使用经过训练的CNN模型（torch7格式）进行单个图像推断？

Question

我以torch7格式获得了经过训练的CNN模型。

如何使用已经过训练的模型对单个图像进行推理并返回预测，类似于this？

Answer 1

1. 您需要使用给定文件路径的torch.load函数来加载模型，例如：model = torch.load("/tmp/youmodel.t7")
2. 然后加载您的图片，例如：img = image.load("/tmp/yourimage.png")
3. 检查img大小是否与模型的输入相同，如果它们不同，您可以使用图像大小调整功能使img达到您想要的大小，详见https://github.com/torch/image
4. 使用转发函数来获取预测，例如：res = model:forward(img)