caffe的src目标下有:proto和layers文件夹,则其关系为:
proto:定义各种层的参数;
layers:定义这种层的操作;每个层中有xx.cpp和xx.cu;其中xx.cpu用作层的cpu计算,xx.cu用作层的GPU计算;
1.将proto/caffe.proto编译成caffe.pb.cc和caffe.pb.h
2.在layers/xx.cpp和xx.cu中调用caffe.pb.h中定义的类;
3.将layers/xx.cpp和xx.cu封装成python可以调用的库,则python通过import即可调用。
caffe编译
最新推荐文章于 2024-05-11 23:32:26 发布