Actually call yolov5 inference
Signed-off-by: Filip Kujawa <filip.j.kujawa@gmail.com>
Change-Id: I693aa253d09e88f6000ee9ea5a8c68862ac47629
diff --git a/third_party/libedgetpu/libedgetpu.BUILD b/third_party/libedgetpu/libedgetpu.BUILD
index 0289452..181c93a 100644
--- a/third_party/libedgetpu/libedgetpu.BUILD
+++ b/third_party/libedgetpu/libedgetpu.BUILD
@@ -1,11 +1,27 @@
cc_library(
- visibility = ["//visibility:public"],
name = "libedgetpu-k8",
- srcs = ["k8/libedgetpu.so.1.0"]
+ srcs = ["k8/libedgetpu.so.1.0"],
+ hdrs = glob(["include/**/*.h"]),
+ strip_include_prefix = "include",
+ visibility = ["//visibility:public"],
)
cc_library(
- visibility = ["//visibility:public"],
name = "libedgetpu-arm",
- srcs = ["arm/libedgetpu.so.1.0"]
-)
\ No newline at end of file
+ srcs = ["arm/libedgetpu.so.1.0"],
+ hdrs = glob(["include/**/*.h"]),
+ strip_include_prefix = "include",
+ visibility = ["//visibility:public"],
+)
+
+genrule(
+ name = "renamed_libedgetpu-arm",
+ srcs = [
+ "arm/libedgetpu.so.1.0",
+ ],
+ outs = [
+ "arm/libedgetpu.so.1",
+ ],
+ cmd = "cp $< $@",
+ visibility = ["//visibility:public"],
+)
diff --git a/third_party/libedgetpu/libedgetpu_build_script.sh b/third_party/libedgetpu/libedgetpu_build_script.sh
old mode 100644
new mode 100755
index 0eafccf..76e20db
--- a/third_party/libedgetpu/libedgetpu_build_script.sh
+++ b/third_party/libedgetpu/libedgetpu_build_script.sh
@@ -1,10 +1,18 @@
+#!/bin/bash
+#This script creates a compressed tarball file named libedgetpu-${GIT_VERSION}.tar.gz,
+# which contains the header files, libraries, and binaries needed to use Edge TPU on both arm and x86 architectures.
+# This script assumes you have Docker installed.
+#
# Clone the correct version of libedgetpu
git clone https://github.com/google-coral/libedgetpu.git
cd libedgetpu
+GIT_VERSION=ddfa7bde33c23afd8c2892182faa3e5b4e6ad94e
+git checkout ${GIT_VERSION}
# Build libedgetpu.so.1.0 for both arm and x86
DOCKER_CPUS="k8" DOCKER_IMAGE="ubuntu:18.04" DOCKER_TARGETS=libedgetpu make docker-build
DOCKER_CPUS="aarch64" DOCKER_IMAGE="debian:stretch" DOCKER_TARGETS=libedgetpu make docker-build
-# Create the directory for the tarball and move the resulting files into it
+# Create the directory for the tarball and move the resulting files into it
+rm -rf libedgetpu-bazel
mkdir libedgetpu-bazel
mkdir libedgetpu-bazel/arm
mkdir libedgetpu-bazel/k8
@@ -12,5 +20,7 @@
cp out/direct/k8/libedgetpu.so.1.0 libedgetpu-bazel/k8
# Copy header files to the include directory
-mkdir libedgetpu-bazel/include
-cp -r include/* libedgetpu-bazel/include/
+mkdir -p libedgetpu-bazel/include/tflite/
+rsync -zarv --include="*/" --include='*.h' --exclude='*' tflite/ libedgetpu-bazel/include/tflite/
+tar zcvf libedgetpu-${GIT_VERSION}.tar.gz libedgetpu-bazel
+