2016-07-03 49 views
0

有沒有什麼辦法可以通過訓練有素的VGG與火炬我的系統中的現有圖像?我使用的是Ubuntu 14.04,不幸的是沒有GPU。我已經搜索了相當廣泛的,但我發現所有需要GPU的。有沒有其他方式使用VGG沒有火炬?我接受建議,但該方法不應該需要GPU。使用火炬訓練VGG臉

回答

0

使用GPU運行網絡會使事情變得更快,您只能在CPU模式下運行網絡。

一旦你加載模型,預訓練在GPU上,你可以簡單的將其轉換爲CPU如下:

model = model:float() 

您可以從您的計算機提供image庫的幫助下輕鬆地將一個圖像,然後做正向傳球

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

+0

非常感謝!你在談論哪種VGG模型?我在哪裏可以找到它? – Sibi

+0

@Sibi你可以在這裏找到可用的預訓練網絡列表:https://github.com/torch/torch7/wiki/ModelZoo(它們是caffe模型),你可以使用loadcaffe模塊作爲火炬模型加載它們作爲:在上面的鏈接中描述。 –