由于Caffe是通过数据库(lmdb或levelddb)读取输入数据,因此不可避免的需要更改附带的convert_imageset代码,将训练数据按照自己的需求写入数据库。比如多任务学习中,需要整合分类学习和相似度学习两种策略,输入数据就需要更改,不再是 <img,label> <script type="math/tex" id="MathJax-Element-1"> </script>的形式,而是 <img,imgpair,labelclass,labelsimilarity> <script type="math/tex" id="MathJax-Element-2"> </script>这种形式。
Codeblocks配置环境的位置在:Project->Build options :
1.头文件的搜索位置
在Search directories ->Compile中添加:
$$(CAFFE_ROOT)/include
$$(CAFFE_ROOT)/build/src (caffe.pb.h 的位置)
2.链接库的搜索位置
在Search directories ->Linker中添加:
$$(CAFFE_ROOT)/build/lib (libcaffe 的位置)
3.添加需要的链接库名
在Linker settings->Other linker options中添加:
-pthread -lcaffe -lglog -lgflags -lprotobuf -lboost_system -lboost_filesystem -lm -lhdf5_hl -lhdf5 -lleveldb -lsnappy -llmdb -lopencv_core -lopencv_highgui -lopencv_imgproc -lboost_thread -lstdc++ -lcblas -latlas
4.添加相关宏定义
在Compile settings->#defines 中添加:
CPU_ONLY
USE_OPENCV
USE_LEVELDB
USE_LMDB
5.添加相关caffe源码
这里需要注意,不能将caffe源码全部导入项目中,否则会编译出错。
右键工程->add files recursively,找到需要的源码。一般的有:
caffe.pb.cc (./build_release/src/caffe/proto中)
src/caffe/util/*.cpp
src/caffe/*.cpp(这里去掉一些会使编译报错的cpp文件,比如net.cpp)
在我配置过程中出现了malloc(): smallbin double linked list corrupted错误,原因是编译缺少对应的cpp文件,比如caffe.pb.cc, common.cpp等。然而编译时不会报错,但运行会出错。