c++实现NHWC2NCHW的几种方法

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);
    }
}

2.采用memcpy函数操作

inline static int convertMat_bgr_bbbgggrrr(const cv::Mat& input_img, cv::Mat& bbbgggrrr)
{
    if (!inp