torch - 如何使用经过训练的 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. 使用 forward 函数得到预测，例如：res = model:forward(img)