526互联

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

发布时间 2023-12-13 15:37:02作者: 海_纳百川

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

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
}