2017-03-15 7 views
0

私は訓練されたバイナリファイルをロードしてテキストファイルに出力するのが好きです。CaffeライブラリのC++またはPythonのH5トレーニングモデルからNetParameterを読み取る方法は?

訓練されたパラメータはC++ from caffemodel normally in binary protobuf formatにあります。私はどのように

は、

net_.reset(new Net<float>(model_file, TEST)); 
    NetParameter param; 

ReadNetParamsFromBinaryFileOrDie(trained_file_protobuf, &param); 
WriteProtoToTextFile(param, "model_protobuf.txt"); 

だった。しかし、いくつかの訓練を受けたモデルは、H5形式とどのようにファイルをロードし、C++またはPythonでテキストファイルに印刷するとして来ます。

net.cppでは、h5形式の場合、ReadNetParamsFromBinaryFileOrDieが見つかりません。 ReadNetParamsFromBinaryFileOrDiebinary protobuf formatです。

訓練されたH5モデルからNetParameterを読み込んでテキストファイルに印刷するにはどうすればよいですか?

答えて

0

CopyTrainedLayersFromHDF5(方法はNet)を見ましたか?

net_.reset(new Net<float>(model_file, TEST)); 
net_.CopyTrainedLayersFromHDF5(trained_file_h5); 
関連する問題