python - 尝试在cv2.COLOR_YUV2BGR_Y422中使用cvtColor时出错-错误:(-215)scn == 2 && depth == 0在函数cv::cvtColor中

我想用openCV阅读YUV视频。 YUV视频为1920 * 1080，并使用YUV422格式(我已经尝试使用yuviewer读取此视频，并且可以正常工作)

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

首先，我尝试了此脚本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或GREY时，我也遇到此错误。
我没有在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 channel 图像。
我更改了这些行:

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函数输入中的数据形状应该如何？

最佳答案

由于YUV 4:2:2具有某些变体，因此，如果您确切地找到您的YUV格式，将会更容易。我使用Logitech Brio网络摄像头通过ffmpeg拍摄YUV视频，这使我在(YUY2 / 0x32595559), yuyv422中获得了YUV。这也是YUV 4:2:2，但是在字节顺序上有一些差异。

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

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

并将YUV转换为BGR(opencv格式):

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

关于python - 尝试在cv2.COLOR_YUV2BGR_Y422中使用cvtColor时出错-错误:(-215)scn == 2 && depth == 0在函数cv::cvtColor中，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/49361591/