在Caffe中创建新图层

bat*_*man 3 machine-learning neural-network gradient-descent deep-learning caffe

我用caffe来深入学习.要为caffe创建一个新图层,我需要

(1)将图层添加到proto/caffe.proto以更新下一个可用ID

(2)通过将图层添加到layer_factory.cpp来创建图层

我的查询是caffe.proto中新目的的用途和layer_factory.cpp中createable的功能.

在layer_factory.cpp中有一些不包含为创建的图层.包含和不包含在layer_factory.cpp中的这些层之间有什么区别.

Sha*_*hai 6

如果要为caffe添加新图层,请按照wiki/Development中的说明进行操作.
你会看到那里,你不需要改变layer_factory.cpp,而是添加两个宏INSTANTIATE_CLASS,并REGISTER_LAYER_CLASS在你的CPP文件.

关于caffe.proto:如果您希望图层具有特定my_layer_param消息,则只需要更改.如果您的图层重复使用现有图层的参数,则根本不需要更改caffe.proto.