Doni/safesight
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复人脸检测，o4.5 4
Some checks are pending
CI / host-build (push) Waiting to run
Details
CI / rk3588-cross-build (push) Waiting to run
Details

Browse Source
This commit is contained in:
sladro 2026-01-07 19:42:29 +08:00
parent 40b929d9d5
commit a37a5b056b
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
plugins/ai_face_det/ai_face_det_node.cpp
View File

@ -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;
}