| commit | 0c517adbde435c2cdc9d3b0855ddb0fb42f40006 | [log] [tgz] |
|---|---|---|
| author | Cindy Liu <hcindyl@google.com> | Fri May 03 14:14:53 2024 -0700 |
| committer | Cindy Liu <hcindyl@google.com> | Fri May 03 14:14:53 2024 -0700 |
| tree | a9f8f77d879dae150755e97fbabc560ecb1cb50e | |
| parent | 120a662d9b37ed645c5489440af389835bfeda9d [diff] |
Update lit test to use python filecheck Change-Id: I288c6f05c6c4944f22fa911c5ed2eb122c889144
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.