From a37a5b056ba4f0bcde09d1ec4af271943786e2c9 Mon Sep 17 00:00:00 2001
From: sladro <sladro@163.com>
Date: Wed, 7 Jan 2026 19:42:29 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BA=BA=E8=84=B8=E6=A3=80?=
 =?UTF-8?q?=E6=B5=8B=EF=BC=8Co4.5=204?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plugins/ai_face_det/ai_face_det_node.cpp | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/plugins/ai_face_det/ai_face_det_node.cpp b/plugins/ai_face_det/ai_face_det_node.cpp
index 1cfc80e..f8fe325 100644
--- a/plugins/ai_face_det/ai_face_det_node.cpp
+++ b/plugins/ai_face_det/ai_face_det_node.cpp
@@ -162,15 +162,18 @@ bool ExtractNc(const Tensor& t, int c, NcTensor& out) {
     int n = 0;
     bool transposed = false;
     if (t.dims.size() == 3) {
-        // Common: [1, C, N] or [1, N, C]
+        // Common: [1, N, C] or [1, C, N]
         const uint32_t d1 = t.dims[1];
         const uint32_t d2 = t.dims[2];
-        if (static_cast<int>(d1) == c) {
-            n = static_cast<int>(d2);
-            transposed = true;  // CxN
-        } else if (static_cast<int>(d2) == c) {
+        if (static_cast<int>(d2) == c) {
             n = static_cast<int>(d1);
-            transposed = false;  // NxC
+            transposed = false;  // [1, N, C]
+        } else if (static_cast<int>(d1) == c) {
+            n = static_cast<int>(d2);
+            transposed = true;   // [1, C, N]
+        } else {
+            // dims don't match expected C, reject immediately (no fallback)
+            return false;
         }
     } else if (t.dims.size() == 2) {
         // [N, C] or [C, N]
@@ -182,15 +185,18 @@ bool ExtractNc(const Tensor& t, int c, NcTensor& out) {
         } else if (static_cast<int>(d0) == c) {
             n = static_cast<int>(d1);
             transposed = true;
+        } else {
+            return false;
         }
-    }
-
-    if (n <= 0) {
+    } else {
+        // Unknown dims layout, try fallback only if no explicit dims
         if (elem_cnt % static_cast<size_t>(c) != 0) return false;
         n = static_cast<int>(elem_cnt / static_cast<size_t>(c));
         transposed = false;
     }
 
+    if (n <= 0) return false;
+
     if (static_cast<size_t>(n) * static_cast<size_t>(c) != elem_cnt) {
         return false;
     }