【发布时间】：2018-09-28 11:17:01
【问题描述】：
我已经获得了一个经过训练的 torch7 格式的 CNN 模型。
如何使用已经训练好的模型对单个图像进行推理并返回预测，类似于this？
【问题讨论】：
标签： torch convolutional-neural-network
【发布时间】：2018-09-28 11:17:01
【问题描述】：
- 您需要通过给定的文件路径使用 torch.load 函数来加载您的模型，例如：
model = torch.load("/tmp/youmodel.t7") - 然后加载你的图片，例如：
img = image.load("/tmp/yourimage.png") - 检查 img 大小是否与你的模型输入相同，如果它们不同，你可以使用 image resize 功能将 img 调整为你想要的大小，查看更多来自https://github.com/torch/image
- 使用forward函数得到预测，例如：
res = model:forward(img)
【讨论】：