| commit | ea117024e28c49ade30d1b3d62fa5fa3fb13fee2 | [log] [tgz] |
|---|---|---|
| author | Lun Dong <lundong@google.com> | Fri Jun 02 13:20:11 2023 -0700 |
| committer | Lun Dong <lundong@google.com> | Fri Jun 02 13:20:11 2023 -0700 |
| tree | 38931a2ece416c52449eca6af46f039ad163016a | |
| parent | 3af6ebc3cb1df4de0a1b2dc10efe26c39df53cf5 [diff] |
ml-model-public: Fix upstream iree breakage renaming mhlo to mhlo-legacy due to https://github.com/openxla/iree/pull/13870. Will further renaming to stablehlo once mnist.mlir is updated. Change-Id: I077a21979da4712f1c04f93072f1ae541cd319c2
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.