一、很多时候模型输入都是NCHW,opecv加载图片的时候需要转换一下,代码如下

    cv::Mat frame = cv::imread(filePath);
    cv::resize(frame, frame, cv::Size(640, 640));
    cv::Mat rgb_image;
    cv::cvtColor(frame, rgb_image, cv::COLOR_BGR2RGB);

    // 转换为 NCHW 格式
    cv::Mat nchw_image;
    {
        int height = rgb_image.rows;
        int width = rgb_image.cols;
        int channels = rgb_image.channels();

        // 将输入图像转换为 CHW 格式
        cv::Mat chw(height * width * channels, 1, CV_8UC1);
        int index = 0;

        for (int c = 0; c < channels; ++c) {
            for (int h = 0; h < height; ++h) {
                for (int w = 0; w < width; ++w) {
                    chw.at<uchar>(index++) = rgb_image.at<cv::Vec3b>(h, w)[c];
                }
            }
        }

        // 创建 NCHW 格式的输出图像
        nchw_image.create(1, channels * height * width, CV_8UC1);
        std::memcpy(nchw_image.data, chw.data, chw.total() * chw.elemSize());
    }

mark

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐