FindCaffe.cmake 345 Bytes
Newer Older
1 2 3
# Caffe package for CNN Triplet training
unset(Caffe_FOUND)

4
find_path(Caffe_INCLUDE_DIR NAMES caffe/caffe.hpp caffe/common.hpp caffe/net.hpp caffe/proto/caffe.pb.h caffe/util/io.hpp
5 6 7 8 9 10 11 12 13 14
  HINTS
  /usr/local/include)

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

if(Caffe_LIBS AND Caffe_INCLUDE_DIR)
    set(Caffe_FOUND 1)
endif()