使用Torch训练的VGG脸部

时间:2016-07-03 11:07:01

标签: machine-learning face-recognition torch vgg-net

有没有什么方法可以通过训练有素的VGG用火炬传递我系统中的现有图像?我使用的是Ubuntu 14.04,遗憾的是没有GPU。我搜索得非常广泛,但我发现的所有内容都需要GPU。还有其他方法可以在没有火炬的情况下使用VGG吗?我愿意接受建议,但该方法不需要GPU。

1 个答案:

答案 0 :(得分:0)

使用GPU运行网络会使事情变得更快,您只能以CPU模式运行网络。

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

model = model:float()

您可以在image库的帮助下轻松地从计算机加载图像,然后进行正向传递

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