Python结合OpenCV读取与显示YUV文件的方法
发布时间：2026-10-13 |
3人浏览过
本文介绍如何使用Python读取YUV格式文件并通过OpenCV进行显示。由于OpenCV不支持直接读取YUV，需先通过Python解析Y、U、V分量数据，再转换为PIL Image对象或NumPy矩阵，最终利用OpenCV实现图像可视化。
Python读取YUV格式文件,并使用opencv显示的方法
OpenCV支持读取多种常见图片格式，例如.jpg和.png，但无法直接处理YUV格式文件。因此，必须借助Python读取YUV数据并完成相应转换后，才能由OpenCV读取并执行后续处理操作。
话不多说，直接上程序。
import cv2
from numpy import *
import Image
screenLevels = 255.0
def yuv_import(filename,dims,numfrm,startfrm):
fp=open(filename,'rb')
blk_size = prod(dims) *3/2
fp.seek(blk_size*startfrm,0)
Y=[]
U=[]
V=[]
print dims[0]
print dims[1]
d00=dims[0]//2
d01=dims[1]//2
print d00
print d01
Yt=zeros((dims[0],dims[1]),uint8,'C')
Ut=zeros((d00,d01),uint8,'C')
Vt=zeros((d00,d01),uint8,'C')
for i in range(numfrm):
for m in range(dims[0]):
for n in range(dims[1]):
#print m,n
Yt[m,n]=ord(fp.read(1))
for m in range(d00):
for n in range(d01):
Ut[m,n]=ord(fp.read(1))
for m in range(d00):
for n in range(d01):
Vt[m,n]=ord(fp.read(1))
Y=Y+[Yt]
U=U+[Ut]
V=V+[Vt]
fp.close()
return (Y,U,V)
if __name__ == '__main__':
width=1280
height=720
data=yuv_import('test.yuv',(height,width),1,0)
YY=data[0][0]
cv2.imshow("sohow",YY)
cv2.waitKey(0)
若需将YUV文件转换为PIL的Image格式，只需在main函数中执行少量处理代码。
YY=data[0][0]
im=Image.frombytes('L',(720,1280),YY.tostring())
此外，若要将PIL的Image转换为OpenCV的Mat格式，只需将PIL对象转换为矩阵形式即可。
im_array = np.array(im) # 也可以用 np.asarray(im) 区别是 np.array() 是深拷贝，np.asarray() 是浅拷贝
以上这篇Python读取YUV文件,并显示的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
