diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-04-19 09:55:06 +0100 |
---|---|---|
committer | Les Bell <les.bell@arm.com> | 2019-04-23 08:38:47 +0000 |
commit | 8271f8144db825960699fffd190ab3e546ee65fc (patch) | |
tree | e8f5bf7ef7c1aecaa74a8a02736674866e2ec3ae /tests/CMakeLists.txt | |
parent | 861985ff2964720a0165e109c3fc568cb245bbe9 (diff) | |
download | armnn-8271f8144db825960699fffd190ab3e546ee65fc.tar.gz |
IVGCVSW-2899 Create a tool to preprocess the images, generating the RAW tensor data from the image files
* ImageTensorGenerator tool generates .raw file contains tensor of the image
* ImageCSVFileGenerator tool generates .csv file contains list of .raw files
Change-Id: Ic7e148857b9f885044bd69da1077b60104cd6509
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to 'tests/CMakeLists.txt')
-rw-r--r-- | tests/CMakeLists.txt | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index cc103c98b6..b7a6d747e0 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -284,3 +284,35 @@ if (BUILD_ARMNN_SERIALIZER OR BUILD_CAFFE_PARSER OR BUILD_TF_PARSER OR BUILD_TF_ ${Boost_PROGRAM_OPTIONS_LIBRARY}) addDllCopyCommands(ExecuteNetwork) endif() + +if(BUILD_ARMNN_QUANTIZER) + macro(ImageTensorExecutor executorName) + target_link_libraries(${executorName} ${CMAKE_THREAD_LIBS_INIT}) + if(OPENCL_LIBRARIES) + target_link_libraries(${executorName} ${OPENCL_LIBRARIES}) + endif() + target_link_libraries(${executorName} + ${Boost_SYSTEM_LIBRARY} + ${Boost_FILESYSTEM_LIBRARY} + ${Boost_PROGRAM_OPTIONS_LIBRARY}) + addDllCopyCommands(${executorName}) + endmacro() + + set(ImageTensorGenerator_sources + InferenceTestImage.hpp + InferenceTestImage.cpp + ImageTensorGenerator/ImageTensorGenerator.cpp) + + add_executable_ex(ImageTensorGenerator ${ImageTensorGenerator_sources}) + target_include_directories(ImageTensorGenerator PRIVATE ../src/armnn) + target_include_directories(ImageTensorGenerator PRIVATE ../src/armnnUtils) + + target_link_libraries(ImageTensorGenerator armnn) + ImageTensorExecutor(ImageTensorGenerator) + + set(ImageCSVFileGenerator_sources + ImageCSVFileGenerator/ImageCSVFileGenerator.cpp) + + add_executable_ex(ImageCSVFileGenerator ${ImageCSVFileGenerator_sources}) + ImageTensorExecutor(ImageCSVFileGenerator) +endif() |