原版的caffe1.0并不支持上采样层,此处教大家如何添加自定义上采样层
首先在此处下载upsample的代码: https://github.com/SeanQ88/caffe_upsample
caffe源码:https://github.com/BVLC/caffe
1. upsample_layer.hpp放入include/caffe/layers文件夹里
2. upsample_layer.cpp和upsample_layer.cu放入src/caffe/layers文件夹里。
3. 打开src/caffe/proto/caffe.proto,在message LayerParameter层添加upsample的ID。
message LayerParameter大约在426行,在{}最后添加如下:
message LayerParameter {
.....
optional UpsampleParameter upsample_param =149;
}
然后再caffe.proto中添加upsample层的参数:
message UpsampleParameter{
optional int32 scale =1[default=1];
}
直接添加到文件最后即可。
重新编译caffe,即完成!