feat: add densenet_onnx model

Signed-off-by: Gaius <[email protected]>

.github/workflows/check-size.yml

@@ -6,9 +6,6 @@ on:
   pull_request:
     branches: [ main, release-* ]
 
-env:
-  JAVA_VERSION: 8
-
 jobs:
   check_size:
     name: Check Size
@@ -25,7 +22,6 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         with:
-          filesizelimit: "524288"
+          filesizelimit: "1048576"
           exclusionPatterns: |
-            deploy/helm-charts
-            manager/console
+            examples/model_repository/densenet_onnx

.gitignore

@@ -1,6 +1,7 @@
 # Prerequisites
 *.d
 .idea
+.DS_Store
 
 # Compiled Object files
 *.slo

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",
+    }
+  ]
+}