caffe_windows1.0扩展upsample层

报错缺少upsample层:

google.protobuf.text_format.ParseError: 2622:5 : Message type "caffe.LayerParameter" has no field named "upsample_param".

扩展方式如下,基于caffe_windows,版本1.0:

1、修改D:\caffe-windows\src\caffe\proto\caffe.proto文件:
caffe_windows1.0扩展upsample层_第1张图片

optional UpsampleParameter upsample_param = 149;

caffe_windows1.0扩展upsample层_第2张图片

// upsample
message UpsampleParameter{
  optional int32 scale = 1 [default = 1];
}

2、将hpp文件放到:

D:\caffe-windows\include\caffe\layers

3、将CPP文件放到:

D:\caffe-windows\src\caffe\layers

4、在visual studio中分别导入:
caffe_windows1.0扩展upsample层_第3张图片
5、重新编译caffe和pycaffe
caffe_windows1.0扩展upsample层_第4张图片
6、可以在编译后D:\caffe-windows\python\caffe\proto\caffe_pb2.py中检查是否有upsample定义:
caffe_windows1.0扩展upsample层_第5张图片
7、将编译后的D:\caffe-windows\python\caffe复制到python的D:\Anaconda3\envs\py35\Lib\site-packages即可。

8、文件下载:https://download.csdn.net/download/samsung12345678/11649956

你可能感兴趣的:(caffe,python,C/C++)