15 lines
345 B
CMake
15 lines
345 B
CMake
# Caffe package for CNN Triplet training
|
|
unset(Caffe_FOUND)
|
|
|
|
find_path(Caffe_INCLUDE_DIR 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_DIR)
|
|
set(Caffe_FOUND 1)
|
|
endif()
|