【caffe Net】使用举例和代码中文注释

首页 > 代码库 > 【caffe Net】使用举例和代码中文注释

【caffe Net】使用举例和代码中文注释

2024-10-14 15:16:02 217人阅读

首先是Net使用的小例子：

#include <vector>
#include <iostream>
#include <caffe/net.hpp>
using namespace std;
using namespace caffe;
int main()
{
    std::string proto("./bambootry/deploy.prototxt");
    Net<float> nn(proto,caffe::TEST);
    vector<string> bn=nn.blob_names();//获取Net中所有Blob对象名
    for(int i=0;i<bn.size();i++)
    {
        cout<<"Blob #"<<i<<" : "<<bn[i]<<endl;
    }
    return 0;
}

linux下编译（bambootry为自己创建的文件夹）

g++ -o ./bambootry/netapp ./bambootry/net.cpp -I ./include -D CPU_ONLY -I ./.build_release/src/ -L ./build/lib -lcaffe -lglog -lboost_system -lprotobuf

结果：

技术分享