用GPU计算cvtColor ,需要注意加上下面两个头文件
#include <opencv2/cudaarithm.hpp>
#include <opencv2/cudaimgproc.hpp>
否则编译会报错显示
报错显示 error: ‘cvtColor’ is not a member of ‘cv::cuda’; did you mean ‘cv::cvtColor’?
下面是chatgpt给的示例代码
#include <opencv2/opencv.hpp>
#include <opencv2/cudaarithm.hpp>
#include <opencv2/cudaimgproc.hpp>
int main() {
// 1. 加载你的 YUV 图像
cv::Mat yuvImage; // 假设已加载 YUV 图像到 yuvImage
cv::Mat bgrImage; // 用于存放转换后的 BGR 图像
// 2. 创建 CUDA 版本的图像
cv::cuda::GpuMat d_yuvImage;
cv::cuda::GpuMat d_bgrImage;
// 3. 将 YUV 图像传输到 GPU
d_yuvImage.upload(yuvImage);
// 4. 使用 CUDA 加速的 cvtColor 进行颜色空间转换
cv::cuda::cvtColor(d_yuvImage, d_bgrImage, cv::COLOR_YUV2BGR_UYVY);
// 5. 从 GPU 中取回 BGR 图像
d_bgrImage.download(bgrImage);
// 6. 现在 bgrImage 中包含转换后的图像,可以进行后续处理或显示
cv::imshow("BGR Image", bgrImage);
cv::waitKey(0);
return 0;
}