MobileNet 在caffe训练加速方法:convolution层改成depthwiseconvolution

1、将cpp 和cu文件复制到caffe/src/caffe/layers目录下,将hpp文件复制到caffe/include/caffe/layers/目录下

文件地址:https://github.com/yonghenglh6/DepthwiseConvolution

2、编译caffe 
cd caffe_ROOT 
make all -j8 


3、训练.prototxt中替换相应的层的名字

将group > 1的 Convolution 换成 DepthwiseConvolution

if layer.type == "Convolution":

    if layer.convolution_param.group !=1:

         layer.type = "DepthwiseConvolution"
 

4、效果

同样的配置,替换前1个batch平均时间4s,替换后1个batch平均时间0.5。训练时间提升8倍。

参考:https://blog.csdn.net/fengzhongluoleidehua/article/details/80797660

猜你喜欢

转载自blog.csdn.net/weixin_41770169/article/details/88289788