有没有什么方法可以通过训练有素的VGG用火炬传递我系统中的现有图像?我使用的是Ubuntu 14.04,遗憾的是没有GPU。我搜索得非常广泛,但我发现的所有内容都需要GPU。还有其他方法可以在没有火炬的情况下使用VGG吗?我愿意接受建议,但该方法不需要GPU。
答案 0 :(得分:0)
使用GPU运行网络会使事情变得更快,您只能以CPU模式运行网络。
加载模型后,在GPU上进行预训练,您可以将其简单地转换为CPU,如下所示:
model = model:float()
您可以在image库的帮助下轻松地从计算机加载图像,然后进行正向传递
local img = image.load(imagefile,3,'byte')
local output = model:forward(img)