14 lines
345 B
14 lines
345 B
# 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()
|
|
|