Assets/Models: Add example configuration files for bundled models

This patch adds the example configuration files for the bundled model files, YoLo_v8s and MobileNet. Signed-off-by: Wook Song <[email protected]>
nnstreamer · Sep 6, 2024 · aecfd32 · aecfd32
1 parent 04ad083
commit aecfd32
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/ml_inference_offloading/src/main/assets/models/mobilenet_v1_1.0_224_quant.conf b/ml_inference_offloading/src/main/assets/models/mobilenet_v1_1.0_224_quant.conf
@@ -0,0 +1,24 @@
+{
+    "single" :
+    {
+        "framework" : "tensorflow-lite",
+        "model" : ["../tests/test_models/models/mobilenet_v1_1.0_224_quant.tflite"],
+        "input_info" : [
+          {
+            "type" : "uint8",
+            "dimension" : "3:224:224:1"
+          }
+        ],
+        "output_info" : [
+          {
+            "type" : "uint8",
+            "dimension" : "1001:1"
+          }
+        ]
+    },
+    "information" :
+    {
+        "threshold" : "0.5",
+        "description" : "Config file to run unittest for ml-extension."
+    }
+}
diff --git a/ml_inference_offloading/src/main/assets/models/yolov8s_float32.conf b/ml_inference_offloading/src/main/assets/models/yolov8s_float32.conf
@@ -0,0 +1,19 @@
+{
+    "single" :
+    {
+        "framework" : "tensorflow-lite",
+        "model" : "yolov8s_float32.tflite",
+        "input_info" : [
+          {
+            "type" : "float32",
+            "dimension" : "3:224:224:1"
+          }
+        ],
+        "output_info" : [
+          {
+            "type" : "float32",
+            "dimension" : "1029:84:1"
+          }
+        ]
+    }
+}