• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2# TensorFlow Lite Example Label Image.
3
4load("//tensorflow/lite:build_def.bzl", "tflite_linkopts")
5
6package(
7    default_visibility = ["//visibility:public"],
8    licenses = ["notice"],
9)
10
11exports_files(glob([
12    "testdata/*.bmp",
13]))
14
15cc_binary(
16    name = "label_image",
17    srcs = [
18        "get_top_n.h",
19        "get_top_n_impl.h",
20        "label_image.cc",
21    ],
22    linkopts = tflite_linkopts() + select({
23        "//tensorflow:android": [
24            "-pie",  # Android 5.0 and later supports only PIE
25            "-lm",  # some builtin ops, e.g., tanh, need -lm
26            "-Wl,-rpath=/data/local/tmp",  # for hexagon delegate
27        ],
28        "//conditions:default": [],
29    }),
30    deps = [
31        ":bitmap_helpers",
32        "//tensorflow/lite:framework",
33        "//tensorflow/lite:string_util",
34        "//tensorflow/lite/c:common",
35        "//tensorflow/lite/kernels:builtin_ops",
36        "//tensorflow/lite/profiling:profiler",
37        "//tensorflow/lite/tools:command_line_flags",
38        "//tensorflow/lite/tools:tool_params",
39        "//tensorflow/lite/tools/delegates:delegate_provider_hdr",
40        "//tensorflow/lite/tools/delegates:tflite_execution_providers",
41        "@com_google_absl//absl/memory",
42        "@com_google_absl//absl/strings",
43    ],
44)
45
46cc_library(
47    name = "bitmap_helpers",
48    srcs = ["bitmap_helpers.cc"],
49    hdrs = [
50        "bitmap_helpers.h",
51        "bitmap_helpers_impl.h",
52        "label_image.h",
53        "log.h",
54    ],
55    deps = [
56        "//tensorflow/lite:builtin_op_data",
57        "//tensorflow/lite:framework",
58        "//tensorflow/lite:string",
59        "//tensorflow/lite:string_util",
60        "//tensorflow/lite/kernels:builtin_ops",
61        "//tensorflow/lite/schema:schema_fbs",
62    ] + select({
63        "//tensorflow:android": [
64            "//tensorflow/lite/delegates/gpu:delegate",
65            "//tensorflow/lite/delegates/hexagon:hexagon_delegate",
66        ],
67        "//tensorflow:android_arm64": [
68            "//tensorflow/lite/delegates/gpu:delegate",
69            "//tensorflow/lite/delegates/hexagon:hexagon_delegate",
70        ],
71        "//conditions:default": [],
72    }),
73)
74
75cc_test(
76    name = "label_image_test",
77    srcs = [
78        "get_top_n.h",
79        "get_top_n_impl.h",
80        "label_image_test.cc",
81    ],
82    data = [
83        "testdata/grace_hopper.bmp",
84    ],
85    deps = [
86        ":bitmap_helpers",
87        "//tensorflow/lite/c:common",
88        "@com_google_googletest//:gtest_main",
89    ],
90)
91