文章/答案/技术大牛

发布

社区首页 >问答首页 >如何从咖啡馆的H5训练模型中读取C++或Python中的NetParameter？

问如何从咖啡馆的H5训练模型中读取C++或Python中的NetParameter？
EN

Stack Overflow用户

提问于 2017-03-15 10:52:11

回答 1查看 239关注 0票数 0

我喜欢加载经过训练的二进制文件并打印到文本文件。

我们可以在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。ReadNetParamsFromBinaryFileOrDie是针对binary protobuf format的。

如何从经过训练的H5模型中读取NetParameter并打印到文本文件？

neural-network

deep-learning

caffe

pycaffe

回答 1

Stack Overflow用户

发布于 2017-03-15 14:12:16

你看过CopyTrainedLayersFromHDF5 (Net的一种方法)吗？

应该是这样的：

net_.reset(new Net<float>(model_file, TEST));
net_.CopyTrainedLayersFromHDF5(trained_file_h5);

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/42800279

复制

相似问题

问如何从咖啡馆的H5训练模型中读取C++或Python中的NetParameter？
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问如何从咖啡馆的H5训练模型中读取C++或Python中的NetParameter？EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问如何从咖啡馆的H5训练模型中读取C++或Python中的NetParameter？
EN