提交 2f36f8cb 编写于 作者: S sjtubinlong

c++ inference: improve performance of mat flatten

上级 9682025c
...@@ -108,10 +108,11 @@ namespace utils { ...@@ -108,10 +108,11 @@ namespace utils {
int rh = im.rows; int rh = im.rows;
int rw = im.cols; int rw = im.cols;
int rc = im.channels(); int rc = im.channels();
int top_index = 0; #pragma omp parallel for
for (int h = 0; h < rh; ++h) { for (int h = 0; h < rh; ++h) {
const uchar* ptr = im.ptr<uchar>(h); const uchar* ptr = im.ptr<uchar>(h);
int im_index = 0; int im_index = 0;
int top_index = h * rw * rc;
for (int w = 0; w < rw; ++w) { for (int w = 0; w < rw; ++w) {
for (int c = 0; c < rc; ++c) { for (int c = 0; c < rc; ++c) {
float pixel = static_cast<float>(ptr[im_index++]); float pixel = static_cast<float>(ptr[im_index++]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册