在Windows下对Caffe源代码进行重新编译之后,新建x64工程,来构建深度神经网络,主要代码:
初始化一个网络
char *proto = "data\\test.prototxt";
Caffe::set_mode(Caffe::CPU);
// Caffe::set_mode(Caffe::GPU);
// Caffe::SetDevice(0);
shared_ptr<Net<float> > net;
net.reset(new Net<float>(proto, caffe::TEST));
加载已训练好的模型
char *model = "data\\*.caffemodel";
net->CopyTrainedLayersFrom(mod