尝试将cvtColor与cv2.COLOR_YUV2BGR_Y422一起使用时出错 - 错误：（-215）scn == 2 && depth == 0 in function cv :: cvtColor

Question

我想用openCV阅读YUV视频。 YUV视频是1920 * 1080并使用YUV422格式（我已经尝试用yuviewer阅读此视频并且它有效）

我正在使用带有OpenCV 3.3.1的Anaconda环境的Python3.6.4

首先，我尝试了这个脚本https://stackoverflow.com/a/47815385/8236556但这一行

cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_Y422)

给我这个错误。

  File "read-yuv-video.py", line 29, in <module>
    bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_UYVY)
cv2.error: C:\ci\opencv_1512688052760\work\modules\imgproc\src\color.cpp:11228: error: (-215) scn == 2 && depth == 0 in function cv::cvtColor

尝试从Y444或Y420转换为RGB或RGBA或灰色时，我也有此错误。我没有在google或stackoverflow上找到完全相同的错误。我试图重新安装opencv包但它没有用。

在我的脚本下面，每个cvtColor都可以工作，除了最后一个。通过工作我的意思是它显示图像，但错误的颜色。

import numpy as np
import cv2

# filename = r'C:\myPath\voile.yuv'
filename = r'C:\myPath\credits.yuv'
yuv_file = open(filename, 'rb')
frame_len = 1920 * 1080 * 3/2
shape = (int(1080 * 1.5), 1920)
raw = yuv_file.read(int(frame_len))
yuv = np.frombuffer(raw, dtype=np.uint8)
yuv = yuv.reshape(shape)
yuv = cv2.resize(yuv, (0, 0), fx=0.5, fy=0.5)

cv2.imshow("raw yuv", yuv)
cv2.waitKey(0)

bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_NV21)
cv2.imshow("YUV2BGR_NV21", bgr)
cv2.waitKey(0)

bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_NV12)
cv2.imshow("YUV2BGR_NV12", bgr)
cv2.waitKey(0)

bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_I420)
cv2.imshow("YUV2BGR_I420", bgr)
cv2.waitKey(0)

bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_Y422)
cv2.imshow("YUV2BGR_Y422", bgr)
cv2.waitKey(0)

编辑

scn == 2表示该功能需要2通道图像。我改变了这些路线：

frame_len = 1920 * 1080 * 2 # 4 Bytes for 2 pixels
shape = (height, width, 2) # To have a 2 channels image (not really working obviously)

和cv2.cvtColor不再抛出错误但仍无法正常工作。我认为我的重塑是完全错误的。

我的结果：

• Original frame
• Raw frame, reshape(height*2, width)
• YUV2BGR_Y422, reshape(height, width, 2)

我在numpy.array手动查看，并与YUView比较像素的值和形状对应于我所期望的（Y值然后U然后V）但这不是opencv显示的。

为什么在原始帧中Y矩阵下面有4个图像？

更准确地说，在我的情况下，cv2.cvtColor函数输入中的数据形状应该如何呢？

Answer 1

由于YUV 4：2：2有一些变体，如果你能确切地找到你的YUV格式会更容易。我使用罗技Brio网络摄像头用ffmpeg拍摄YUV视频，这给了我在(YUY2 / 0x32595559), yuyv422的YUV。这也是YUV 4：2：2，但在字节顺序上有一些差异。

由于YUV 4：2：2压缩比为2/3，因此对于1个像素使用2个字节是正确的。这对我有用。

frame_len = int(width * height * 2)
shape = (height, width, 2)

并将YUV转换为BGR（opencv格式）：

bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_YUY2)