caffe | caffe源码中添加upsample层

本文主要是介绍caffe | caffe源码中添加upsample层，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

原版的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，即完成！

这篇关于caffe | caffe源码中添加upsample层的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---