experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add tflite label_image example cmake build setting.

Browse Source
This commit is contained in:
Jerry Shih 2021-02-22 14:25:48 +08:00
parent dcfdade562
commit 8dfa06523a
2 changed files with 71 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/CMakeLists.txt
View File

@ -424,6 +424,9 @@ add_library(tensorflow::tensorflowlite ALIAS tensorflow-lite)
# The benchmark tool.
add_subdirectory(${TFLITE_SOURCE_DIR}/tools/benchmark)
# The label_image example.
add_subdirectory(${TFLITE_SOURCE_DIR}/examples/label_image)
# Python interpreter wrapper.
add_library(_pywrap_tensorflow_interpreter_wrapper SHARED EXCLUDE_FROM_ALL
${TFLITE_SOURCE_DIR}/python/interpreter_wrapper/interpreter_wrapper.cc

68
tensorflow/lite/examples/label_image/CMakeLists.txt Normal file
View File

@ -0,0 +1,68 @@
#
# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# The label_image example for Tensorflow Lite.
populate_source_vars("${TFLITE_SOURCE_DIR}/examples/label_image"
TFLITE_LABEL_IMAGE_SRCS
FILTER "_test\\.cc$"
)
list(APPEND TFLITE_LABEL_IMAGE_SRCS
${TF_SOURCE_DIR}/core/util/stats_calculator.cc
${TFLITE_SOURCE_DIR}/profiling/memory_info.cc
${TFLITE_SOURCE_DIR}/profiling/profile_summarizer.cc
${TFLITE_SOURCE_DIR}/profiling/profile_summary_formatter.cc
${TFLITE_SOURCE_DIR}/profiling/time.cc
${TFLITE_SOURCE_DIR}/tools/command_line_flags.cc
${TFLITE_SOURCE_DIR}/tools/delegates/default_execution_provider.cc
${TFLITE_SOURCE_DIR}/tools/evaluation/utils.cc
${TFLITE_SOURCE_DIR}/tools/tool_params.cc
)
if(TFLITE_ENABLE_XNNPACK)
list(APPEND TFLITE_LABEL_IMAGE_SRCS
${TFLITE_SOURCE_DIR}/tools/delegates/xnnpack_delegate_provider.cc
)
else()
set(TFLITE_LABEL_IMAGE_CC_OPTIONS "-DTFLITE_WITHOUT_XNNPACK")
endif() # TFLITE_ENABLE_XNNPACK
if(CMAKE_SYSTEM_NAME MATCHES "Android")
if(_TFLITE_ENABLE_NNAPI)
list(APPEND TFLITE_LABEL_IMAGE_SRCS
${TFLITE_SOURCE_DIR}/tools/delegates/nnapi_delegate_provider.cc
)
endif() # _TFLITE_ENABLE_NNAPI
endif() # Android
if(TFLITE_ENABLE_GPU)
list(APPEND TFLITE_LABEL_IMAGE_SRCS
${TFLITE_SOURCE_DIR}/tools/delegates/gpu_delegate_provider.cc
)
endif() # TFLITE_ENABLE_GPU
add_executable(label_image
EXCLUDE_FROM_ALL
${TFLITE_LABEL_IMAGE_SRCS}
)
target_compile_options(label_image
PRIVATE
${TFLITE_LABEL_IMAGE_CC_OPTIONS}
)
target_link_libraries(label_image
tensorflow-lite
${CMAKE_DL_LIBS}
)