C++工程中使用caffe

来源:互联网 发布:淘宝助理5.7 编辑:程序博客网 时间:2024/05/16 02:53

1.首先,按照官方文档的方式,准备好LMDB或LEVELDB数据库,也可以使用HDF5和图片格式,具体图片格式的使用方式,可以参考ImageNet的例子Caffe | ImageNet tutorial

2.之后,用protobuf写好你的网络模型和Solver,可以参考mnist的写法。也可使用./model/下已有的model。

3.使用命令行caffe进行训练。
以上几个步骤在denny的博客中有详细介绍:
http://blog.csdn.net/lengconglin/article/details/76732312

4.新建一个C++ PROJECT,然后把caffe的include文件夹作为include path加进去,把./build/lib作为library bath加进去,把./build/bin作为链接库加进去。

5、文件开始#include<caffe/caffe.hpp>,然后参考caffe/examples/cpp_classification/classification.cpp的例子来写

注意:程序可能出现编译问题,最简单有效的方法是,把自己的cpp文件放到caffe的examples目录下,然后在caffe工程中make

原创粉丝点击