1. 首页
  2. 人工智能
  3. 深度学习
  4. caffe的leaky层源码实现

caffe的leaky层源码实现

上传者: 2018-12-27 03:15:18上传 ZIP文件 1KB 热度 39次
在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文件
下载地址
用户评论