git clone or download frome Caffe-MobileNetV2-ReLU6 : (1).
在 Caffe-MobileNetV2-ReLU6/目錄下:
relu6_layer.cpp、
relu6_layer.cu
拷貝到 $ YOUR_CAFFE_ROOT_PATH\src\caffe\layers 目錄裡
(2).
在 Caffe-MobileNetV2-ReLU6/目錄下
relu6_layer.hpp
拷貝到 $ YOUR_CAFFE_ROOT_PATH\include\caffe\layers 目錄裡
caffe.proto改寫:
(1).
到$ YOUR_CAFFE_ROOT_PATH\src\caffe\proto 目錄裡打開"caffe.proto"
在 message LayerParameter {
...
}
裡面加入:
ReLU6Parameter relu6_param = 100000
(2).
到$ YOUR_CAFFE_ROOT_PATH\src\caffe\proto 目錄裡打開"caffe.proto"
在最底下加入:
message ReLU6Parameter {
optional float negative_slope = 1 [default = 0];
}
重新編譯caffe.pb.cc caffe.pb.h:
$ cd YOUR_CAFFE_ROOT_PATH/src/caffe/proto
$ protoc --cpp_out=/YOUR_CAFFE_ROOT_PATH/src/caffe/proto caffe.proto
$ cp /YOUR_CAFFE_ROOT_PATH/src/proto/caffe.pb.h /YOUR_CAFFE_ROOT_PATH/include/caffe/proto/caffe.ph.h
重新編譯caffe:
$ cd YOUR_CAFFE_ROOT_PATH
$ mkdir build
$ make
$ make runtest