* Build the demos with tidl offload option for AM5 * Include mobilenet v1 model for demonstrating tidl offload * Remove the demo scripts under the files directory. The scripts are now maintained in the source repo. * Make adjustments with the updated Makefile
Signed-off-by: Hongmei Gou <[email protected]> --- .../files/run_classification.sh | 7 ----- .../tensorflow-lite/files/run_segmentation.sh | 8 ------ .../tensorflow-lite-demo_git.bb | 27 +++++++++++-------- 3 files changed, 16 insertions(+), 26 deletions(-) delete mode 100644 meta-arago-extras/recipes-support/tensorflow-lite/files/run_classification.sh delete mode 100644 meta-arago-extras/recipes-support/tensorflow-lite/files/run_segmentation.sh diff --git a/meta-arago-extras/recipes-support/tensorflow-lite/files/run_classification.sh b/meta-arago-extras/recipes-support/tensorflow-lite/files/run_classification.sh deleted file mode 100644 index b2ad7fc4..00000000 --- a/meta-arago-extras/recipes-support/tensorflow-lite/files/run_classification.sh +++ /dev/null @@ -1,7 +0,0 @@ -#!/bin/sh - -cd /usr/share/tensorflow-lite/demos - -# Run tflite_classification binary with the model, the input image, and the label file specified. -# Classification label is overlayed with the input image. Right click the image display window to exit. -./tflite_classification -m ../examples/mobilenet_v1_1.0_224_quant.tflite -i ../examples/grace_hopper.bmp -l ../examples/labels.txt diff --git a/meta-arago-extras/recipes-support/tensorflow-lite/files/run_segmentation.sh b/meta-arago-extras/recipes-support/tensorflow-lite/files/run_segmentation.sh deleted file mode 100644 index 5caea127..00000000 --- a/meta-arago-extras/recipes-support/tensorflow-lite/files/run_segmentation.sh +++ /dev/null @@ -1,8 +0,0 @@ -#!/bin/sh - -cd /usr/share/tensorflow-lite/demos - -# Run tflite_segmentation binary with the model and the input image specified. -# The display window shows the input(resized) image, segmentation map, overlay -# of the former two. Right click the image display window to exit. -./tflite_segmentation -m ./deeplabv3_257_mv_gpu.tflite -i ./bird_segmentation.bmp diff --git a/meta-arago-extras/recipes-support/tensorflow-lite/tensorflow-lite-demo_git.bb b/meta-arago-extras/recipes-support/tensorflow-lite/tensorflow-lite-demo_git.bb index 9b99b63c..57d6ef20 100644 --- a/meta-arago-extras/recipes-support/tensorflow-lite/tensorflow-lite-demo_git.bb +++ b/meta-arago-extras/recipes-support/tensorflow-lite/tensorflow-lite-demo_git.bb @@ -1,33 +1,33 @@ -DESCRIPTION = "Tensorflow Lite Demo with input and display via OpenCV and TIDL acceleration for AM5" +SUMMARY = "Tensorflow Lite Demo with input and display via OpenCV and TIDL acceleration for AM5" LICENSE = "Apache-2.0" LIC_FILES_CHKSUM = "file://LICENSE;md5=d8927f3331d2b3e321b7dd1925166d25" DEPENDS = "tensorflow-lite flatbuffers opencv" +DEPENDS_append_dra7xx = " tidl-api" COMPATIBLE_MACHINE = "ti33x|ti43x|omap-a15|am65xx" +PACKAGE_ARCH = "${MACHINE_ARCH}" PV = "01.00.00" -PR = "r0" +PR = "r1" BRANCH = "master" SRC_URI = "git://git.ti.com/apps/tensorflow-lite-examples.git;protocol=git;branch=${BRANCH} \ http://storage.googleapis.com/download.tensorflow.org/models/tflite/gpu/deeplabv3_257_mv_gpu.tflite;name=deeplabv3;subdir=${WORKDIR}/model;destsuffix=model \ - http://raw.githubusercontent.com/google-coral/edgetpu/master/test_data/bird_segmentation.bmp;name=bird_segmentation;subdir=${WORKDIR}/test_data;destsuffix=test_data \ - file://run_classification.sh \ - file://run_segmentation.sh" + http://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_224.tgz;name=mobilenetv1;subdir=${WORKDIR}/model;destsuffix=model \ + http://raw.githubusercontent.com/google-coral/edgetpu/master/test_data/bird_segmentation.bmp;name=bird_segmentation;subdir=${WORKDIR}/test_data;destsuffix=test_data"; SRC_URI[deeplabv3.md5sum] = "4a24db5a5fb05c47586a1197765e8548" SRC_URI[deeplabv3.sha256sum] = "68a539782c2c6a72f8aac3724600124a85ed977162b44e84cbae5db717c933c6" +SRC_URI[mobilenetv1.md5sum] = "d5f69cef81ad8afb335d9727a17c462a" +SRC_URI[mobilenetv1.sha256sum] = "1ccb74dbd9c5f7aea879120614e91617db9534bdfaa53dfea54b7c14162e126b" SRC_URI[bird_segmentation.md5sum] = "fc3202ac4e3033b36df3043de6dc32c9" SRC_URI[bird_segmentation.sha256sum] = "e43cedd29ab385db787531c8a6cdcb31c5e4292af34f242877ed40611e502670" -SRCREV = "d84f1d43585c0db2fa6a1c15db9145ab53f859ce" +SRCREV = "bd6a429c21f96d5112e0b6dea44a13b748eefd74" S = "${WORKDIR}/git" -CXXFLAGS_append = " -O3 -DNDEBUG -fPIC -I${STAGING_INCDIR}" -LDFLAGS_append = " -L${STAGING_LIBDIR}" - EXTRA_OEMAKE = "\ 'CXX=${CXX}' \ 'CXXFLAGS=${CXXFLAGS}' \ @@ -37,8 +37,12 @@ EXTRA_OEMAKE = "\ 'TARGET=${TARGET_OS}' \ 'TARGET_ARCH=${TARGET_ARCH}'" +EXTRA_OEMAKE += "SYSROOT_INCDIR="${STAGING_INCDIR}" SYSROOT_LIBDIR="${STAGING_LIBDIR}" \ + TIDL_API_DIR="${STAGING_DATADIR}/ti/tidl"" + +TIDL_dra7xx = "yes" do_compile() { - oe_runmake -f Makefile + oe_runmake -f Makefile TIDL_ACC=${TIDL} } do_install() { @@ -46,8 +50,9 @@ do_install() { install -m 0755 ${S}/tflite_classification ${D}${datadir}/tensorflow-lite/demos install -m 0755 ${S}/tflite_segmentation ${D}${datadir}/tensorflow-lite/demos install -m 0644 ${WORKDIR}/model/deeplabv3_257_mv_gpu.tflite ${D}${datadir}/tensorflow-lite/demos + install -m 0644 ${WORKDIR}/model/mobilenet_v1_1.0_224.tflite ${D}${datadir}/tensorflow-lite/demos install -m 0644 ${WORKDIR}/test_data/bird_segmentation.bmp ${D}${datadir}/tensorflow-lite/demos - install -m 0755 ${WORKDIR}/run*.sh ${D}${datadir}/tensorflow-lite/demos + install -m 0755 ${S}/scripts/run*.sh ${D}${datadir}/tensorflow-lite/demos } FILES_${PN} = "${datadir}/tensorflow-lite/demos" -- 2.17.1 _______________________________________________ meta-arago mailing list [email protected] http://arago-project.org/cgi-bin/mailman/listinfo/meta-arago
![http://raw.githubusercontent.com/google-coral/edgetpu/master/test_data/bird_segmentation.bmp;name=bird_segmentation;subdir=${WORKDIR}/test_data;destsuffix=test_data"](http://raw.githubusercontent.com/google-coral/edgetpu/master/test_data/bird_segmentation.bmp;name=bird_segmentation;subdir=${WORKDIR}/test_data;destsuffix=test_data"){kind=link}