FindCaffe.cmake 529 字节
Newer Older
B
Bikramjot Hanzra 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
# Copied from 
# https://raw.githubusercontent.com/opencv/opencv_contrib/master/modules/cnn_3dobj/FindCaffe.cmake

unset(Caffe_FOUND)
unset(Caffe_INCLUDE_DIRS)
unset(Caffe_LIBS)

find_path(Caffe_INCLUDE_DIRS NAMES 
    caffe/caffe.hpp 
    caffe/common.hpp 
    caffe/net.hpp 
    caffe/proto/caffe.pb.h 
    caffe/util/io.hpp
    HINTS
    /usr/local/include)

find_library(Caffe_LIBS NAMES caffe
    HINTS
    /usr/local/lib)

if(CAFFE_LIBS AND Caffe_INCLUDE_DIRS)
  set(Caffe_FOUND 1)
endif(CAFFE_LIBS AND Caffe_INCLUDE_DIRS)