C++代码cuda版本的opencv实现图片BGR转RGB-HWC转CHW

前几天聊天的时候,有人疑问无法实现这种转换,因为原先实践过的缘故,就把解决方案写下

int Detect::dataProcess(unsigned char *img_ptr, int h, int w){
    img = cv::Mat(h, w, CV_8UC3, img_ptr);
    cuda::GpuMat matScr(img)
    matSrc.upload(img); 
    cuda::resise(matSrc, matSrc, Size(320, 320), 0, 0, cv::INTER_LINEAR);
    matSrc.convertTo(matResizeCvt, CV_32FC3, 1, 0);
    cuda::cvtColor(matResizeCvt, matSrc, 4, 0); //BGR TO RGB
    cuda::split(matSrc, matChannel); //HWC TO CHW
}

 

posted @ 2023-12-13 15:33  海_纳百川  阅读(309)  评论(0编辑  收藏  举报
本站总访问量