opencv中图像色彩空间变换

opencv颜色空间转换

1、 rgb转yuv

win下使用bgr通道顺序。 目前仅3种转换。要求图像宽、高是2的倍数。

cv::Mat yuv, yuvI420, yuvYV12;
cv::cvtColor(img, yuv, cv::COLOR_BGR2YUV);
cv::cvtColor(img, yuvI420, cv::COLOR_BGR2YUV_I420);
cv::cvtColor(img, yuvYV12, cv::COLOR_BGR2YUV_YV12); //yuv420p

2、 yuv420p转rgb

YUV_YV12实际就是 YUV_420p。

cv::Mat rgb;
cv::cvtColor(yuvI420, rgb, cv::COLOR_YUV2BGR_I420);
cv::cvtColor(yuvYV12, rgb, cv::COLOR_YUV2BGR_YV12);
//cv::cvtColor(yuvYV12, rgb, cv::COLOR_YUV420p2BGR); // 效果同上

3、 yuv420sp转rgb

先使用yuv420p转换成420sp，再使用转换函数。

cv::Mat yuv420sp = yuvYV12.clone();
uchar* uv = yuv420sp.data + img.rows*img.cols;
const uchar* u = yuvYV12.data + img.rows*img.cols;
const uchar* v = u + img.rows*img.cols / 4;
for(int i = 0; i < img.rows*img.cols / 4; i++) {
    uv[2*i]   = u[i];
    uv[2*i+1] = v[i];
}

cv::Mat rgb;
cv::cvtColor(yuv420sp, rgb, cv::COLOR_YUV420sp2BGR);