diff --git a/plugins/ai_face_det/ai_face_det_node.cpp b/plugins/ai_face_det/ai_face_det_node.cpp index c90b891..6f89b1c 100644 --- a/plugins/ai_face_det/ai_face_det_node.cpp +++ b/plugins/ai_face_det/ai_face_det_node.cpp @@ -1,4 +1,5 @@ #include +#include #include #include #include @@ -210,7 +211,7 @@ bool ExtractNc(const Tensor& t, int c, NcTensor& out) { if (n <= 0) { if (elem_cnt % static_cast(c) != 0) return false; n = static_cast(elem_cnt / static_cast(c)); - transposed = false; + layout = Layout::FlatNc; } if (static_cast(n) * static_cast(c) != elem_cnt) {