feat: add densenet_onnx model

Signed-off-by: Gaius <[email protected]>
dragonflyoss · Mar 29, 2024 · 0728632 · 0728632
1 parent 443779f
commit 0728632
Show file tree

Hide file tree

Showing 4 changed files with 1,030 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,6 +1,7 @@
 # Prerequisites
 *.d
 .idea
+.DS_Store
 
 # Compiled Object files
 *.slo

diff --git a/examples/model_repository/densenet_onnx/1/model.onnx b/examples/model_repository/densenet_onnx/1/model.onnx
diff --git a/examples/model_repository/densenet_onnx/config.pbtxt b/examples/model_repository/densenet_onnx/config.pbtxt
@@ -0,0 +1,29 @@
+name: "densenet_onnx"
+platform: "onnxruntime_onnx"
+max_batch_size : 0
+input [
+  {
+    name: "data_0"
+    data_type: TYPE_FP32
+    format: FORMAT_NCHW
+    dims: [ 3, 224, 224 ]
+    reshape { shape: [ 1, 3, 224, 224 ] }
+  }
+]
+output [
+  {
+    name: "fc6_1"
+    data_type: TYPE_FP32
+    dims: [ 1000 ]
+    reshape { shape: [ 1, 1000, 1, 1 ] }
+    label_filename: "densenet_labels.txt"
+  }
+]
+model_repository_agents
+{
+  agents [
+    {
+      name: "dragonfly",
+    }
+  ]
+}