commit | 8381f479cd611dc744cc253b9c29c9841120f4fb | [log] [tgz] |
---|---|---|
author | Julian Mullings-Black <julianmb@google.com> | Tue Apr 09 19:47:04 2024 +0000 |
committer | Julian Mullings-Black <julianmb@google.com> | Fri Apr 12 19:11:58 2024 +0000 |
tree | 0d1ae756070d01a4e1e8bf8b85713de9f4be6fe2 | |
parent | f7d52957f418d162cdb4231d90abcf773fa5d145 [diff] |
Match target names to the patterns in ml/ml-models Change-Id: I4d112800f965c88d893c3cfd21a64ad4912d9611
This is the model zoo for public models used in Shodan vector core examples.
The person presence detection quantized model from https://github.com/tensorflow/tflite-micro/blob/main/tensorflow/lite/micro/models/person_detect.tflite
Mobilenet V1 float-point model from https://tfhub.dev/tensorflow/lite-model/mobilenet_v1_0.25_224/1/default/1?lite-format=tflite
Mobilenet V1 quantized model from https://tfhub.dev/tensorflow/lite-model/mobilenet_v1_0.25_224_quantized/1/default/1?lite-format=tflite
Mobilenet V2 quantized model from https://tfhub.dev/tensorflow/lite-model/mobilenet_v2_1.0_224_quantized/1/default/1?lite-format=tflite
HPS (Human Presence Sensor) non-tiled quantized model from https://chromium.googlesource.com/chromiumos/platform/hps-firmware/+/6cdea6d1158a8cd3238b8ae4f744fdb494779c80/models/shared.tflite
For tflite models, use the web-based visualization tool to inspect the file. The tool supports drag and drop or file GUI.
Test data for sample model inputs are in the “test_data” directory. Currently, only test data for the Human Presence Sensor model are included.
Test image files for Human Presence Sensor model, from ChromeOS codebase.
For each model, the corresponding model executable under iree_exec
can be built with the .c/.h files using sw/vec_iree
project as library. They can be run as unit tests via lit framework using Renode or QEMU.