Actually call yolov5 inference
Signed-off-by: Filip Kujawa <filip.j.kujawa@gmail.com>
Change-Id: I693aa253d09e88f6000ee9ea5a8c68862ac47629
diff --git a/third_party/libtensorflowlite/libtensorflowlite.BUILD b/third_party/libtensorflowlite/libtensorflowlite.BUILD
index a6b837d..761ca76 100644
--- a/third_party/libtensorflowlite/libtensorflowlite.BUILD
+++ b/third_party/libtensorflowlite/libtensorflowlite.BUILD
@@ -1,16 +1,19 @@
cc_library(
- visibility = ["//visibility:public"],
name = "tensorflow-k8",
+ srcs = ["k8/libtensorflowlite.so"],
hdrs = glob(["include/**/*.h"]),
+ copts = ["-Wno-unused-parameter"],
strip_include_prefix = "include",
- srcs = ["k8/libtensorflowlite.so"]
+ visibility = ["//visibility:public"],
)
cc_library(
- visibility = ["//visibility:public"],
name = "tensorflow-arm",
+ srcs = ["arm/libtensorflowlite.so"],
hdrs = glob(["include/**/*.h"]),
+ copts = ["-Wno-unused-parameter"],
strip_include_prefix = "include",
- srcs = ["arm/libtensorflowlite.so"]
+ visibility = ["//visibility:public"],
)
+exports_files(["arm/libtensorflowlite.so"])