41 lines
1.5 KiB
CMake
41 lines
1.5 KiB
CMake
if(NOT HAVE_PYTHON)
|
|
message(STATUS "Python interface is disabled or not all required dependencies found. Building without it...")
|
|
return()
|
|
endif()
|
|
|
|
file(GLOB_RECURSE python_srcs ${PROJECT_SOURCE_DIR}/python/*.cpp)
|
|
|
|
add_library(pycaffe SHARED ${python_srcs})
|
|
caffe_default_properties(pycaffe)
|
|
set_target_properties(pycaffe PROPERTIES PREFIX "" OUTPUT_NAME "_caffe")
|
|
target_include_directories(pycaffe PUBLIC ${PYTHON_INCLUDE_DIRS} ${NUMPY_INCLUDE_DIR})
|
|
target_link_libraries(pycaffe PUBLIC ${Caffe_LINK} ${PYTHON_LIBRARIES})
|
|
|
|
if(UNIX OR APPLE)
|
|
set(__linkname "${PROJECT_SOURCE_DIR}/python/caffe/_caffe.so")
|
|
add_custom_command(TARGET pycaffe POST_BUILD
|
|
COMMAND ln -sf $<TARGET_LINKER_FILE:pycaffe> "${__linkname}"
|
|
COMMAND ${CMAKE_COMMAND} -E make_directory ${PROJECT_SOURCE_DIR}/python/caffe/proto
|
|
COMMAND touch ${PROJECT_SOURCE_DIR}/python/caffe/proto/__init__.py
|
|
COMMAND cp ${proto_gen_folder}/*.py ${PROJECT_SOURCE_DIR}/python/caffe/proto/
|
|
COMMENT "Creating symlink ${__linkname} -> ${PROJECT_BINARY_DIR}/lib/_caffe${Caffe_POSTFIX}.so")
|
|
endif()
|
|
|
|
# ---[ Install
|
|
# scripts
|
|
file(GLOB python_files *.py requirements.txt)
|
|
install(FILES ${python_files} DESTINATION python)
|
|
|
|
# module
|
|
install(DIRECTORY caffe
|
|
DESTINATION python
|
|
FILES_MATCHING
|
|
PATTERN "*.py"
|
|
PATTERN "ilsvrc_2012_mean.npy"
|
|
PATTERN "test" EXCLUDE
|
|
)
|
|
|
|
# _caffe.so
|
|
install(TARGETS pycaffe DESTINATION python/caffe)
|
|
|