diff options
Diffstat (limited to 'runtime/contrib/style_transfer_app/CMakeLists.txt')
| -rw-r--r-- | runtime/contrib/style_transfer_app/CMakeLists.txt | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/runtime/contrib/style_transfer_app/CMakeLists.txt b/runtime/contrib/style_transfer_app/CMakeLists.txt new file mode 100644 index 000000000..4896e6710 --- /dev/null +++ b/runtime/contrib/style_transfer_app/CMakeLists.txt @@ -0,0 +1,40 @@ +if(NOT BUILD_STYLE_TRANSFER_APP) + return() +endif(NOT BUILD_STYLE_TRANSFER_APP) + +if(NOT BUILD_ONERT) + return() +endif(NOT BUILD_ONERT) + +find_package(JPEG) +if(JPEG_FOUND) + add_definitions(-DNNFW_ST_APP_JPEG_SUPPORTED) +else(JPEG_FOUND) + message(WARNING "JPEG not found.") +endif(JPEG_FOUND) + +list(APPEND STYLE_TRANSFER_APP_SRCS "src/style_transfer_app.cc") +list(APPEND STYLE_TRANSFER_APP_SRCS "src/args.cc") +list(APPEND STYLE_TRANSFER_APP_SRCS "src/bitmap_helper.cc") +if(JPEG_FOUND) + list(APPEND STYLE_TRANSFER_APP_SRCS "src/jpeg_helper.cc") +endif(JPEG_FOUND) + +nnas_find_package(Boost REQUIRED) + +add_executable(style_transfer_app ${STYLE_TRANSFER_APP_SRCS}) +target_include_directories(style_transfer_app PRIVATE src) +target_include_directories(style_transfer_app PRIVATE ${Boost_INCLUDE_DIRS}) +if(JPEG_FOUND) + target_include_directories(style_transfer_app PRIVATE ${JPEG_INCLUDE_DIRS}) +endif(JPEG_FOUND) + +target_link_libraries(style_transfer_app onert_core onert tflite_loader) +target_link_libraries(style_transfer_app tensorflow-lite ${LIB_PTHREAD} dl nnfw_lib_tflite) +target_link_libraries(style_transfer_app nnfw-dev) +target_link_libraries(style_transfer_app boost_program_options boost_system boost_filesystem) +if(JPEG_FOUND) + target_link_libraries(style_transfer_app ${JPEG_LIBRARIES}) +endif(JPEG_FOUND) + +install(TARGETS style_transfer_app DESTINATION bin) |