修复人脸画框1
This commit is contained in:
parent
e1f9cd0bef
commit
3cd9971996
@ -1,4 +1,5 @@
|
||||
#include <algorithm>
|
||||
#include <atomic>
|
||||
#include <array>
|
||||
#include <cctype>
|
||||
#include <cmath>
|
||||
@ -210,7 +211,7 @@ bool ExtractNc(const Tensor& t, int c, NcTensor& out) {
|
||||
if (n <= 0) {
|
||||
if (elem_cnt % static_cast<size_t>(c) != 0) return false;
|
||||
n = static_cast<int>(elem_cnt / static_cast<size_t>(c));
|
||||
transposed = false;
|
||||
layout = Layout::FlatNc;
|
||||
}
|
||||
|
||||
if (static_cast<size_t>(n) * static_cast<size_t>(c) != elem_cnt) {
|
||||
|
||||
Loading…
Reference in New Issue
Block a user