如何保存预先训练好的咖啡因模型的子集？
如何保存预先训练好的咖啡因模型的子集？
提问于 2016-08-15 19:20:57
我正在研究一个预先训练好的咖啡因模型(在python中)，它有3层。我想要分解这个caffe模型，并创建一个新的模型，与这个模型的第一层相同。例如：
原始咖啡因模型数据-> conv1_1 -> conv1_2 -> conv2_1 -> conv2_2 -> conv3_1 -> conv3_2
New Caffe model data -> conv1_1 -> conv1_2
有人能帮我吗？
回答 1
Stack Overflow用户
回答已采纳
发布于 2016-08-16 11:50:17
Python公开.caffemodel文件中的数据。它可以作为一个数组来访问。例如,
net = caffe.Net('path/to/conv.prototxt', 'path/to/conv.caffemodel', caffe.TEST)
W = net.params['con_1'][0].data[...]
b = net.params['con_1'][1].data[...]您可以将此数据复制到新文件中，并将其另存为.caffemodel文件。看看this和this吧。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/38954117
复制相关文章
相似问题
腾讯云开发者
