c++实现NHWC2NCHW的几种方法

cv2016_DL

已于 2024-11-12 16:28:25 修改

阅读量465

点赞数 4

文章标签： opencv 人工智能计算机视觉

于 2024-06-11 12:04:50 首次发布

本文链接：https://blog.csdn.net/u012374012/article/details/139595451

版权

这篇博客介绍了两种在C++中将NHWC格式的图像数据转换为NCHW格式的方法。第一种使用了OpenCV的extractChannel函数，结合imread和cvtColor进行操作。第二种方法则是通过memcpy直接处理像素数据。这两种方法都涉及到图像通道的重新排列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.利用opencv中extractChannel函数

void NHWC2NCHW(float * input_buf)
{
cv::Mat image = cv::imread("../data/bus.jpg");
if (image.empty()) {
std::cerr << "Failed to read image." << std::endl;
}
cv::cvtColor(image, image, cv::COLOR_BGR2RGB);
// 将像素值归一化到 [0, 1] 范围
cv::Mat input_array;
image.convertTo(input_array, CV_32FC3, 1.0 / 255.0);

int rh = input_array.rows;
int rw = input_array.cols;
int rc = input_array.channels();
for (int i = 0; i < rc; ++i) {
cv::extractChannel(input_array, cv::Mat(rh, rw, CV_32FC1, input_buf + i * rh * rw), i);
}
}