如何保存预训练caffe模型的子集?

时间:2016-08-15 11:20:57

标签: caffe pre-trained-model

我正在研究一个有3层的预训练caffe模型(在python中)。我想分解这个caffe模型并创建一个与该模型的第一层相同的新模型。例如:

原始Caffe模型 数据 - > conv1_1 - > conv1_2 - > conv2_1 - > conv2_2 - > conv3_1 - > conv3_2

新Caffe模型 数据 - > conv1_1 - > conv1_2

有人能帮助我吗?

1 个答案:

答案 0 :(得分:0)

Python公开.caffemodel文件中的数据。它可以作为数组访问。例如,

windowManager.removeViewImmediate(view);

您可以将此数据复制到新文件中,并将其另存为.caffemodel文件。请查看thisthis