| # Copyright 2020 The Chromium OS Authors. All rights reserved. |
| # Distributed under the terms of the GNU General Public License v2 |
| |
| EAPI=7 |
| |
| DESCRIPTION="TFLite models and supporting assets used for testing ML & NNAPI." |
| HOMEPAGE="https://chromium.googlesource.com/aosp/platform/test/mlts/models/" |
| |
| LICENSE="Apache-2.0" |
| SLOT="0" |
| KEYWORDS="*" |
| IUSE="label_image benchmark_model" |
| |
| # Basic test model (mobilenet.tflite, grace_hopper.bmp, labels.txt) |
| LABEL_IMAGE_TARBALL="ml-test-assets-label-image-assets_20200721.tar.xz" |
| |
| # Models from https://android.googlesource.com/platform/test/mlts/models |
| # tarball is from commit 73313692ee90e6d768cd086fbb7c94298ecaf1ac 2020-07-03 |
| AOSP_MLTS_MODELS_TARBALL="aosp-mlts-models-7331369.tar.xz" |
| |
| SRC_URI=" |
| label_image? ( gs://chromeos-localmirror/distfiles/${LABEL_IMAGE_TARBALL} ) |
| benchmark_model? ( gs://chromeos-localmirror/distfiles/${AOSP_MLTS_MODELS_TARBALL} ) |
| " |
| |
| S="${WORKDIR}" |
| |
| src_install() { |
| if use label_image || use benchmark_model; then |
| insinto /usr/share/ml-test-assets |
| doins -r ./* |
| fi |
| } |