Caffe项目外使用classification.cpp

caffe项目中自带了一个classification.cpp文件,能够用训练好的模型进行直接分类,不过须要在caffe项目外使用的话就须要本身写编译代码。html

能够参考一下这篇文章,找了很久才找到http://blog.sina.com.cn/s/blog_534497fd0102wf2t.html 里面写的很全。而后根据这篇简单总结了一下cmakelists.txt的简单写法python

cmake_minimum_required(VERSION 2.8)
project(classification)

find_package(OpenCV REQUIRED)
find_package(Caffe REQUIRED)

include_directories( ${Caffe_INCLUDE_DIRS} )
add_definitions(${Caffe_DEFINITIONS}) # ex. -DCPU_ONLY
add_executable(caffeClassify classification.cpp)
target_link_libraries(caffeClassify ${OpenCV_LIBS} ${Caffe_LIBRARIES})

注意caffe必定要make install。 用make的方法或者cmake方法编译的都要install一下,注意!

若是要在qt中使用的话推荐一下这个https://github.com/withwsf/Image_detection/blob/master/Image_detection.pro,里面有用qt管理的caffe项目,我把他的.pro复制出来。c++

QT       += core gui
CONFIG +=c++11

greaterThan(QT_MAJOR_VERSION, 4): QT += widgets

TARGET = Image_detection
TEMPLATE = app
SOURCES += main.cpp\
        mainwindow.cpp \
    image_detection.cpp \
    nms.cpp \
    dockwidget.cpp \
    detector_warpper.cpp

HEADERS  += mainwindow.h \
        image_detection.h \
    nms.h \
    dockwidget.h \
    detector_warpper.h

FORMS    += mainwindow.ui \
    dockwidget.ui

INCLUDEPATH += /home/vcc/caffe_depen/caffe-fast-rcnn/include \
               /usr/include/opencv /usr/include/opencv2 \



LIBS += -L/home/vcc/caffe_depen/lib -lcaffe -lcblas -latlas

LIBS+= -L/usr/local/lib  -lglog -lgflags -lprotobuf -lleveldb -lsnappy -llmdb -lboost_system -lhdf5_hl -lhdf5 -lm -lopencv_core -lopencv_highgui -lopencv_imgproc -lboost_thread -lstdc++  -lprotobuf
INCLUDEPATH +=/usr/include/python2.7/
INCLUDEPATH +=/usr/include/
LIBS += -lboost_python -lpython2.7 -lboost_system

这是这个做者的工程,能够借鉴着改。git


想要在windows中使用caffe项目,推荐这个githubhttps://github.com/happynear/caffe-windows/tree/ms到这个目录中https://github.com/happynear/caffe-windows/tree/ms/windows/caffe.binding,你们也能够借鉴着改。github

相关文章
相关标签/搜索