生成caffe.pb.cc和caffe.pb.h文件

本文针对windows-caffe添加新层过程中,需要用到extract-proto批命令生成caffe.pb.cc和caffe.pb.h文件,在.\src\caffe\proto\里没有extract_proto.bat文件的情况。
我之前走了一些弯路,在网上找了一些方法,不完全适用我的问题,可能我这里的解决方案也只是适用于某些同仁,如果我的方法不能解决,多尝试几种网上的方案,总有一款适合你。
如果是在caffe下添加自己的新层,相信你已经修改好caffe.proto文件。我们直入主题
step1:写一个批处理文件(后缀名是.bat),在文本上写,保存的时候后缀名为.bat。文本内容:

protoc caffe.proto --cpp_out=./
protoc caffe.proto --python_out=./
md ..\..\..\python\caffe\proto\
copy /y .\caffe_pb2.py ..\..\..\python\caffe\proto\
copy nul ..\..\..\python\caffe\proto\__init__.py
pause

文件的名字自己随便取。推荐 : extract_proto.bat

当你双击这个批命令时,可能提示protoc不是内部命令。

step2: 你应该下载一个protoc.exe文件,放到C:\Windows\System32目录下。
之前我再网上找到一种方法,说放在caffe-master\windows\scripts目录下,我并没有成功,但是看到有些成功了,可能这种方法不适合我,同样的,也可能我的方法不适合某些人,反而另外一种方法会适合。一个原则:多试试就出来了。
protoc.exe文件下载链接,点这里

step3: protoc.exe放到指定路径下后,再双击extract_proto.bat就会在.\src\caffe\proto下从新生成caffe.pb.cc和caffe.pb.h两个文件。

猜你喜欢

转载自blog.csdn.net/lqp888888/article/details/77943837
今日推荐