| commit | 3af6ebc3cb1df4de0a1b2dc10efe26c39df53cf5 | [log] [tgz] |
|---|---|---|
| author | Cindy Liu <hcindyl@google.com> | Fri May 19 17:21:30 2023 -0700 |
| committer | Cindy Liu <hcindyl@google.com> | Fri May 19 17:21:30 2023 -0700 |
| tree | d6c71d673a3b094ffff5d9232175fbf4bcc36182 | |
| parent | 264407ec8183ed115f893bb59053e71b0391094d [diff] |
Update the model in cantrip-builtin to fit in 4MB memory ML core only has 4MB in HW. For the models running on Shodan, we need to make sure it fits in 4MB. Bug: 258850990 Change-Id: I8a5428b775538664313bf2835089ba5b1e1bd22f
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.