Caffe |添加新图层后使用训练模型(微调)后未知的底部斑点

时间:2016-06-20 12:34:12

标签: machine-learning neural-network deep-learning caffe conv-neural-network

我想在我的数据集上微调bvlc_reference_caffenet模型。 我成功地完善了模型。但现在我正在尝试添加2个卷积层以检查它将如何执行。我尝试在"conv6_new"图层之后添加名称为"conv7_new""conv5"的图层。我复制了"conv3"图层,并将其重命名为"conv6_new",将"conv5"重命名为"conv7_new"。我修改了相应的"ReLU""Pooling"图层的名称。 但是我得到了以下错误 任何人都可以建议我在哪里犯错误?

F0620 17:29:49.967382 14194 insert_splits.cpp:29] Unknown bottom blob 'conv7_new' (layer 'conv7_new', bottom index 0)

1 个答案:

答案 0 :(得分:1)

图层"conv7_new"也不能将自己的输出(“顶部”)作为输入。确保不仅重命名图层的名称,还要以有意义的方式重命名"top""bottom"
如果您有"conv5" - > "conv6_new" - > "conv7_new" - > "fc"
那么"conv6_new"的底部应该是"top"的{​​{1}},而"conv5"的底部是"conv7_new"的{​​{1}}。最后,"top"的底部是新"conv6_new"图层的"fc" 请注意,我忽略了可以“就地”计算的"top",方法是将"conv7_new"名称设置为与"ReLU"相同。我也没有包含汇集层。