machine-learning - Torch でトレーニングされた VGG 顔を使用する

Question

トーチを使用してトレーニング済みの VGG を介して、システム内の既存のイメージを渡す方法はありますか? Ubuntu 14.04 を使用していますが、残念ながら GPU がありません。かなり広範囲に検索しましたが、見つけたものはすべて GPU を必要とします。トーチなしで VGG を使用する他の方法はありますか? 私は提案を受け入れますが、この方法は GPU を必要としません。

Answer 1

GPU を使用してネットワークを実行すると処理が大幅に高速化されますが、ネットワークは CPU モードでのみ実行できます。

GPU で事前トレーニングされたモデルをロードしたら、次のように簡単に CPU に変換できます。

model = model:float()

イメージライブラリを使用してコンピュータからイメージを簡単にロードし、フォワード パスを実行できます。

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