Add object detection inferencing code
Add a cc_library for inferencing using tflite and edgetpu.
Signed-off-by: Filip Kujawa <filip.j.kujawa@gmail.com>
Change-Id: Ie4cfa1e960a3d7461a75df074ebf12e4d47727e5
diff --git a/y2023/vision/BUILD b/y2023/vision/BUILD
index 0f75aba..0221aad 100644
--- a/y2023/vision/BUILD
+++ b/y2023/vision/BUILD
@@ -1,5 +1,6 @@
load("@com_github_google_flatbuffers//:build_defs.bzl", "flatbuffer_cc_library")
load("@com_github_google_flatbuffers//:typescript.bzl", "flatbuffer_ts_library")
+load("//tools/build_rules:select.bzl", "cpu_select")
cc_binary(
name = "camera_reader",
@@ -267,3 +268,23 @@
"@com_github_google_glog//:glog",
],
)
+
+cc_library(
+ name = "yolov5_lib",
+ srcs = ["yolov5.cc"],
+ hdrs = ["yolov5.h"],
+ deps = [
+ "//third_party:opencv",
+ "@com_github_gflags_gflags//:gflags",
+ "@com_github_google_glog//:glog",
+ ] + cpu_select({
+ "amd64": [
+ "@libtensorflowlite//:tensorflow-k8",
+ "@libedgetpu//:libedgetpu-k8",
+ ],
+ "arm": [
+ "@libtensorflowlite//:tensorflow-arm",
+ "@libedgetpu//:libedgetpu-arm",
+ ],
+ }),
+)