在Caffe中,如何修改特定的conv层的行为?

时间:2018-04-17 08:58:30

标签: c++ machine-learning neural-network deep-learning caffe

在Caffe中,如何修改特定的转化层行为?

我想我可以修改转换层类C ++代码,但我不知道如何在转换层类中获取转换层的名称,所以我不能使用if()来使我的代码仅适用于特定的转换层。

任何人都可以帮助我,非常感谢〜

1 个答案:

答案 0 :(得分:0)

图层名称是其LayerParameter的一部分。要访问它,只需

const LayerParameter& param;
std::cout << "My layer's name is: " << param.name();