如何使用经过训练的 CNN 模型(torch7 格式)进行单图像推理

Question

我得到了一个经过训练的Torch7格式的CNN模型。

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

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)