Upload 6 files

cut-onnx.py

@@ -0,0 +1,14 @@
+import onnx
+
+input_path = "yolo11s-seg.onnx"
+output_path = "yolo11s-seg-cut.onnx"
+input_names = ["images"]
+output_names = ["/model.23/Concat_1_output_0", 
+                "/model.23/Concat_2_output_0", 
+                "/model.23/Concat_3_output_0",
+                "/model.23/cv4.0/cv4.0.2/Conv_output_0",
+                "/model.23/cv4.1/cv4.1.2/Conv_output_0",
+                "/model.23/cv4.2/cv4.2.2/Conv_output_0",
+                "output1"]
+
+onnx.utils.extract_model(input_path, output_path, input_names, output_names)

yolo11-seg-test.py

@@ -0,0 +1,16 @@
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO("yolo11s-seg.pt")
+model.info()
+
+# Use the model
+results = model("./images/ssd_horse.jpg")
+
+# Save the results
+results[0].save("yolo11s-seg-result.jpg")
+
+# Export to onnx with simplify
+model.export(format='onnx', simplify=True)
+
+

yolo11_seg_config.json

@@ -0,0 +1,52 @@
+{
+  "model_type": "ONNX",
+  "npu_mode": "NPU1",
+  "quant": {
+    "input_configs": [
+      {
+        "tensor_name": "images",
+        "calibration_dataset": "./dataset/coco_1000.tar",
+        "calibration_size": 32,
+        "calibration_mean": [0, 0, 0],
+        "calibration_std": [255.0, 255.0, 255.0]
+      }
+    ],
+    "calibration_method": "MinMax",
+    "precision_analysis": true,
+    "precision_analysis_method":"EndToEnd"
+  },
+  "input_processors": [
+    {
+      "tensor_name": "images",
+      "tensor_format": "BGR",
+      "src_format": "BGR",
+      "src_dtype": "U8",
+      "src_layout": "NHWC"
+    }
+  ],
+  "output_processors": [
+    {
+      "tensor_name": "/model.23/Concat_1_output_0",
+      "dst_perm": [0, 2, 3, 1]
+    },    {
+      "tensor_name": "/model.23/Concat_2_output_0",
+      "dst_perm": [0, 2, 3, 1]
+    },    {
+      "tensor_name": "/model.23/Concat_3_output_0",
+      "dst_perm": [0, 2, 3, 1]
+    },
+    {
+      "tensor_name": "/model.23/cv4.0/cv4.0.2/Conv_output_0",
+      "dst_perm": [0, 2, 3, 1]
+    },    {
+      "tensor_name": "/model.23/cv4.1/cv4.1.2/Conv_output_0",
+      "dst_perm": [0, 2, 3, 1]
+    },    {
+      "tensor_name": "/model.23/cv4.2/cv4.2.2/Conv_output_0",
+      "dst_perm": [0, 2, 3, 1]
+    }
+  ],
+  "compiler": {
+    "check": 0
+  }
+}

yolo11s-seg-cut.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:17c958a92905a17f085056fd678259b754942db803210d7cae38ff8978a51307
+size 40482328

yolo11s-seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8f1d774af7212af31a43c6a658660c7beeee89b04bc669bf7583680f97ffc694
+size 40657019