FindCaffe.cmake 345 B

1234567891011121314
  1. # Caffe package for CNN Triplet training
  2. unset(Caffe_FOUND)
  3. find_path(Caffe_INCLUDE_DIR NAMES caffe/caffe.hpp caffe/common.hpp caffe/net.hpp caffe/proto/caffe.pb.h caffe/util/io.hpp
  4. HINTS
  5. /usr/local/include)
  6. find_library(Caffe_LIBS NAMES caffe
  7. HINTS
  8. /usr/local/lib)
  9. if(Caffe_LIBS AND Caffe_INCLUDE_DIR)
  10. set(Caffe_FOUND 1)
  11. endif()