本文介绍了如何在Caffe中配置ShuffleNet v1，包括添加自定义的shuffle_channel_layer。首先，从指定链接下载配置文件，然后将cpp和cu文件放入相应目录，接着将头文件添加到include目录，并在caffe.proto中声明新层。最后，进行编译，解决可能遇到的文本编辑器导致的编译问题。
背景：
论文入口：
https://arxiv.org/pdf/1707.01083.pdf
配置文件下载入口：
https://github.com/farmingyard/ShuffleNet
下载后文件如下：
主要我们用到框框里的3个文件进行配置
因为这些都是作者自己定义的layer，是Caffe框架本身没有的，需要我们添加到Caffe框架下
配置方法：
同一般的添加层的方法类似：
（1）
将
shuffle_channel_layer.cpp
shuffle_channel_layer.cu
这2个文件放置到 caffe/src/caffe/layers 目录下
（2）
将
shuffle_channel_layer.hpp
最低0.47元/天 解锁文章
扫一扫
678
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
