登录 注册
当前位置:主页 > 资源下载 > 9 > Caffe中leaky层的源代码实现

Caffe中leaky层的源代码实现

  • 更新:2024-08-25 10:46:14
  • 大小:1KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:深度学习 - 人工智能
  • 格式:ZIP

资源介绍

在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文件