如何使用经过训练的 CNN 模型（torch7 格式）进行单张图像推理？
Posted
技术标签:
【中文标题】如何使用经过训练的 CNN 模型（torch7 格式）进行单张图像推理？【英文标题】：How to do a single image inference using a trained CNN model (torch7 format)? 【发布时间】：2018-09-28 11:17:01 【问题描述】：我已经获得了一个经过训练的 torch7 格式的 CNN 模型。
如何使用已经训练好的模型对单个图像进行推理并返回预测，类似于this？
【问题讨论】：
【参考方案1】：-
您需要通过给定的文件路径使用 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)
【讨论】：
以上是关于如何使用经过训练的 CNN 模型（torch7 格式）进行单张图像推理？的主要内容，如果未能解决你的问题，请参考以下文章