2017-09-06 12 views
1

私のチームはCaffeモデルを数ヶ月トレーニングしましたが、現在はTensorflowへの変換を検討しており、数ヶ月の再トレーニングをしたくありません。Caffe CaffemodelウェイトファイルをTensorFlowウェイトファイルに変換する

カフェモデルウェイトファイルをTensorFlowウェイトファイルに変換するにはどうすればよいですか?

私はcaffe-tensorflowを試しましたが、複数のレイヤーを持つレイヤーを持つCaffeモデルでは機能しません。

答えて

1

私はに慣れていませんが、this python codeを使用して、訓練されたカフェウェイトをPythonで読み取り可能なnumpy配列に「ダンプ」できます。私は、TFが数量的な配列からその重みを初期化する機能を持つことを期待しています - この方法は自動でも簡単でもないかもしれませんが、一度だけ行う必要があります。

0

caffe-tensorflowから得られるエラーは何ですか?これが妥当性チェックである場合は、無効にすることができます。 load_using_pb機能を使用して重みをロードするツールを実行することもできます。 (transformers.py/DataInjector/load関数)

関連する問題