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

修复人脸识别问题,更换facerknn，修复崩溃2
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-08 15:27:18 +08:00
parent 6d9b22517f
commit f61b9d4c99
3 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
models/face_gallery.db
View File

Binary file not shown.

2
plugins/ai_face_recog/ai_face_recog_node.cpp
View File

@ -897,9 +897,7 @@ private:
in.type = RKNN_TENSOR_UINT8; in.type = RKNN_TENSOR_UINT8;
} }
std::cerr << "[ai_face_recog] DEBUG: before InferBorrowed, in.size=" << in.size << " in.type=" << in.type << "\n";
auto r = AiScheduler::Instance().InferBorrowed(model_handle_, in); auto r = AiScheduler::Instance().InferBorrowed(model_handle_, in);
std::cerr << "[ai_face_recog] DEBUG: after InferBorrowed, success=" << r.success << "\n";
if (!r.success || r.outputs.empty()) { if (!r.success || r.outputs.empty()) {
std::cerr << "[ai_face_recog] inference failed: " << (r.error.empty() ? "unknown" : r.error) << "\n"; std::cerr << "[ai_face_recog] inference failed: " << (r.error.empty() ? "unknown" : r.error) << "\n";
continue; continue;

11
src/ai_scheduler.cpp
View File

@ -401,17 +401,6 @@ AiScheduler::BorrowedInferResult AiScheduler::InferBorrowed(ModelHandle handle,
return result; return result;
} }
// DEBUG: 检查输出大小是否超出预分配 buffer
for (uint32_t i = 0; i < ctx->n_output; ++i) {
size_t prealloc_sz = ctx->output_buffers[i].size();
size_t actual_sz = outputs[i].size;
std::cerr << "[AiScheduler] output[" << i << "] prealloc=" << prealloc_sz
<< " actual=" << actual_sz << " type=" << ctx->output_attrs[i].type << "\n";
if (actual_sz > prealloc_sz && prealloc_sz > 0) {
std::cerr << "[AiScheduler] ERROR: output size exceeds preallocated buffer!\n";
}
}
result.outputs.resize(ctx->n_output); result.outputs.resize(ctx->n_output);
for (uint32_t i = 0; i < ctx->n_output; ++i) { for (uint32_t i = 0; i < ctx->n_output; ++i) {
auto& out = result.outputs[i]; auto& out = result.outputs[i];