2017-12-05 8 views
0

私は、私のプロジェクトにVGG-Face記述子モデル(http://www.robots.ox.ac.uk/~vgg/software/vgg_face/)を使用しようとしています。私がしたいのは、単に入力画像からネットワークの出力を得ることだけです。PyTorchでVGG-Face記述子を設定する

私はMatConvNet、Caffe、PyTorchのいずれも使用していないので、PyTorchをランダムに選んでいます。 (torch.legacy.nn.Sequential.Sequentialクラスの)モデルが古いバージョンのPyTorchに保存されていたため、構文はPyTorchのドキュメントのものと若干異なりました。

私はそうのようなのlua .t7モデルをロードすることができました:入力画像の

vgg_net = load_lua('./vgg_face_torch/VGG_FACE.t7', unknown_classes=True) 

とロード:

# load image 
image = imread('./ak.png') 
# convert to tensor 
input = torch.from_numpy(image).float() 

嬉々として、私は多くをモデルに画像にロードされました予期:

# load into vgg_net 
output = vgg_net.forward(input) 

しかし、私が望むことは、コンパイルに失敗します。奇妙なエラーメッセージを残して:

Traceback (most recent call last): 
    File "~/Documents/python/vgg-face-test/vgg-pytorch.py", line 25, in <module> 
    output = vgg_net.forward(input) 
    File "~/.local/lib/python3.6/site-packages/torch/legacy/nn/Module.py", line 33, in forward 
    return self.updateOutput(input) 
    File "~/.local/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 235, in updateOutput_patch 
    return obj.updateOutput(*args) 
    File "~/.local/lib/python3.6/site-packages/torch/legacy/nn/Sequential.py", line 36, in updateOutput 
    currentOutput = module.updateOutput(currentOutput) 
TypeError: 'NoneType' object is not callable 

私は絶対に馬鹿げています。

これは私がStackoverflowのヘルプを求めている理由です。ここの誰かが私にモデルのセットアップに手を貸してくれることを願っています。実際にはTorchでもそうではありません。実際にはどの作業モデルでも、特定の画像の説明を得ることができます。

答えて

関連する問題