2017-08-02 6 views
0

私の仕事では、vgg-19 netのpool5レイヤからフィーチャを抽出する必要があります。これまではCaffeを使用していましたが、その後、Tensorflowを使用して抽出された機能をさらに練習しました。しかし今、私はトップにデコン層を追加し、ネット全体をエンドツーエンドのトレーニングのために微調整したいと考えています。今度は、天気予報を使ってグラジエントを取得していますが、vaff-19からcaffeを使ってフィーチャを取得していましたので、これらのグラジエントをCFPに逆伝播することは難しいでしょう。それで、テンソルフローのvgg-19モデルを探しています。これは、どのレイヤーからもフィーチャを抽出するために使用できます。私はいくつかのプロジェクトをオンラインで見つけましたが、Tensorflowによって公式に支持されていません。TensorFlowはvgg-19モデル/フィーチャ抽出ツールを承認しました

+0

Kerasにはkeras.applicationsのVGG-19モジュールがあります。 –

+0

マティアスありがとうございましたがテンソルフローで何かを探していました。テンソルフローでレイヤーと目的関数をすでにコーディングしていたので、そのコードをケラスに戻すのに多くの時間がかかりました。 –

答えて

0

tensorflowの場合は、TF-Slim libraryを使用し、herepretrained modelsを使用できます。それらの有用な方法を確認するには、slim walk through notebookを使用してください。

+0

お返事ありがとうございます!だからちょうどフォローアップの質問。あなたはcaffeと比較してそのパフォーマンスを認識していますか? –

+0

最後に私は(https://arxiv.org/pdf/1608.07249.pdf)tfのパフォーマンスはcaffeほど良くはありませんが、最近のバージョンについてはわかりません。 –

+0

この便利なリソースに感謝します! –

関連する問題