Google Git
Sign in
opensecura / ml-models-public / 811f55d90be8e81772ee30f9a1042bf9c1a11394
commit811f55d90be8e81772ee30f9a1042bf9c1a11394[log] [tgz]
authorCindy Liu <hcindyl@google.com>Mon Jul 11 19:57:21 2022 -0700
committerCindy Liu <hcindyl@google.com>Thu Jul 21 17:22:59 2022 -0700
treea3912e0de4bc5b6e9a44e00d98488a5dd33f52a6
parent2ab50f0c18e483bf4b0d503156c9eb02c0a19941 [diff]
ml-models-public: Move model-related code away from sw/vec_iree

Build the workload in the model project

Change-Id: I92f3442d78071113ea4fed86beefd32b3657059c
24 files changed
tree: a3912e0de4bc5b6e9a44e00d98488a5dd33f52a6
  1. float_models/
  2. quant_models/
  3. CMakeLists.txt
  4. README.md
README.md

Public ML model zoo

This is the model zoo for public models used in Shodan vector core examples.

Model discription

person_detection.tflite

The person presence detection quantized model from https://github.com/tensorflow/tflite-micro/blob/main/tensorflow/lite/micro/models/person_detect.tflite

mobilenet_v1_0.25_224_float.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_0.25_224_quant.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_1.0_224_quant.tflite

Mobilenet V2 quantized model from https://tfhub.dev/tensorflow/lite-model/mobilenet_v2_1.0_224_quantized/1/default/1?lite-format=tflite

Visualize the model

For tflite models, use the web-based visualization tool to inspect the file. The tool supports drag and drop or file GUI.

Model executables

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.