caffe的leaky层源码实现
在caffe下实现yolo需要添加一些特殊的层,其中yolo使用的激活函数为leaky,即需要在caffe特定目录下添加该层的源码实现。 (1)leaky_layer.hpp放在caffe/include/caffe/layers下; (2)leaky_layer.cpp放在caffe/src/caffe/layers下; (3)修改caffe/src/caffe/proto下的caffe.proto文件
文件列表
caffe的leaky层实现.zip
(预估有个2文件)
caffe的leaky层实现
leaky_layer.hpp
1KB
leaky_layer.cpp
1KB
暂无评论