背景:
论文入口:
https://arxiv.org/abs/1704.04861
配置文件下载入口:
https://github.com/farmingyard/caffe-mobilenet
在caffe版本的MobileNet 和ShuufleNet中,我们会用到ConvolutionDepthwise
层,即深度可分离卷积。
Caffe原本的框架里是没有ConvolutionDepthwise
层的定义。
我们需要自己添加进Caffe里,并重新编译,否则就可能会出现下面的错误:
Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: ConvolutionDepthwise
配置方法:
先下载配置文件
下载后文件如下:
主要我们用到框框里的3个文件进行配置
因为这些都是作者自己定义的layer,是Caffe框架本身没有的,需要我们添加到Caffe框架下