0

有什么方法可以通过训练有素的 VGG 和火炬传递系统中的现有图像?我使用的是 Ubuntu 14.04,不幸的是没有 GPU。我已经进行了相当广泛的搜索,但我发现的所有内容都需要 GPU。有没有其他方法可以在没有 Torch 的情况下使用 VGG?我愿意接受建议,但该方法不需要 GPU。

4

1 回答 1

0

虽然使用 GPU 运行网络会使事情变得更快,但您只能在 CPU 模式下运行网络。

加载模型并在 GPU 上进行预训练后,您可以简单地将其转换为 CPU,如下所示:

model = model:float()

您可以借助图像库轻松地从计算机加载图像,然后进行前向传递

local img = image.load(imagefile,3,'byte') local output = model:forward(img)

于 2016-07-06T03:11:17.423 回答