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

Create build target for image recognition test.

Browse Source 
PiperOrigin-RevId: 325094476
Change-Id: I6e0ed20c868dd80f2f3d8735ecd126108bf20a02
This commit is contained in:
Nat Jeffries 2020-08-05 13:54:48 -07:00 committed by TensorFlower Gardener
parent a6339f25b9
commit ff61ec799e
2 changed files with 37 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tensorflow/lite/micro/examples/image_recognition_experimental/BUILD Normal file
View File

@ -0,0 +1,36 @@
# Description:
# TensorFlow Lite for Microcontrollers image recognition example.
load(
"//tensorflow/lite/micro/testing:micro_test.bzl",
"tflite_micro_cc_test",
)
licenses(["notice"]) # Apache 2.0
cc_library(
name = "image_model_data",
srcs = [
"first_10_cifar_images.cc",
"image_recognition_model.cc",
],
hdrs = [
"first_10_cifar_images.h",
"image_recognition_model.h",
"util.h",
],
)
tflite_micro_cc_test(
name = "image_recognition_test",
srcs = ["image_recognition_test.cc"],
deps = [
":image_model_data",
"//tensorflow/lite:schema_fbs_version",
"//tensorflow/lite/micro:micro_error_reporter",
"//tensorflow/lite/micro:micro_framework",
"//tensorflow/lite/micro:op_resolvers",
"//tensorflow/lite/micro/testing:micro_test",
"//tensorflow/lite/schema:schema_fbs",
],
)

1
tensorflow/lite/micro/examples/image_recognition_experimental/Makefile.inc
View File

@ -20,6 +20,7 @@ tensorflow/lite/micro/examples/image_recognition_experimental/image_recognition_
$(MAKEFILE_DIR)/downloads/image_recognition_model/image_recognition_model.cc
IMAGE_RECOGNITION_TEST_HDRS := \
tensorflow/lite/micro/examples/image_recognition_experimental/first_10_cifar_images.h \
tensorflow/lite/micro/examples/image_recognition_experimental/image_recognition_model.h \
tensorflow/lite/micro/examples/image_recognition_experimental/util.h