You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Merge the caffe folder in the repo with your own caffe.
$ cp -r $REPO/caffe/* $YOURCAFFE/
Then make.
$ cd $YOURCAFFE && make
Usage
Replacing the type of mobile convolution layer with "DepthwiseConvolution" is all.
Please refer to the example/Withdw_MN_train_128_1_train.prototxt, which is altered from
I write a script [transfer2Mobilenet.py] to convert normal net to mobilenet format. You may try too.
Usage:
python ./transfer2Mobilenet.py sourceprototxt targetprototxt [--midbn nobn --weight_filler msra --activation ReLU] ["--origin_type" means the depthwise convolution layer's type will be "Convolution" instead of "DepthwiseConvolution"]
The "transferTypeToDepthwiseConvolution.py" will be used for changing the depthwise convolution layer's type from "Convolution" to "DepthwiseConvolution".
Footnotes
When turn on cudnn, the memory consuming of mobilenet would increase to unbelievable level. You may try. ↩
About
A personal depthwise convolution layer implementation on caffe by liuhao.(only GPU)