opencv 读取yuv

OpenCV可以通过cv::VideoCapture类读取常见的视频文件格式，如.avi和.mp4等。然而，在某些应用场景中，我们需要读取yuv格式的原始视频数据，例如压缩算法的研究或图像处理的探索。在这种情况下，OpenCV提供了一个cv::VideoCapture类，可以读取yuv格式的视频数据。

使用cv::VideoCapture::open()打开yuv文件时，需要指定文件的路径，并设置四个参数：视频宽度、视频高度、帧率和视频格式。对于yuv文件，格式通常是YUV 4:2:0或YUV 4:2:2。例如：

cv::VideoCapture capture;
std::string filename = "path/to/yuvfile.yuv";
int width = 1920;
int height = 1080;
int fps = 30;
int format = CV_8UC1; // YUV格式通常为单通道
bool success = capture.open(filename, cv::CAP_ANY);
if (!success) {
std::cerr << "Error: Failed to open the video file." << std::endl;
return -1;
}

读取yuv文件时，每个帧通常由Y、U和V三个分量组成。因此，我们需要使用不同的函数读取不同的分量。例如：

cv::Mat yuv_image;
cv::Mat y_plane(height, width, format);
cv::Mat u_plane(height / 2, width / 2, format);
cv::Mat v_plane(height / 2, width / 2, format);
while (capture.read(yuv_image)) {
// 分离y、u、v分量
cv::extractChannel(yuv_image, y_plane, 0);
cv::extractChannel(yuv_image, u_plane, 1);
cv::extractChannel(yuv_image, v_plane, 2);

// 处理y分量
// ...

// 将分量合并以重构原始图像
cv::Mat bgr_image;
cv::merge(std::vector<cv::Mat>{y_plane, u_plane, v_plane}, bgr_image);
// 显示图像
cv::imshow("Video", bgr_image);
cv::waitKey(1);
}

需要注意的是，由于yuv文件的像素值通常不是8位，而是10位，12位等，因此需要根据实际情况对像素值进行缩放和偏移。此外，在将Y、U和V三个分量合并以重构原始图像之前，需要将U和V分量上采样到与Y分量相同的分辨率。