Python读取YUV文件,并显示的方法_Python

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函数中,进行稍微的处理。

1 2	`YY=data[0][0]` `im=Image.frombytes('L',(720,1280),YY.tostring())`

另外,如果PIL的image想转换为opencv的mat格式,只需将PIL转换为矩阵的形式。

1 2	`im_array` `=` `np.array(im)` `# 也可以用 np.asarray(im) 区别是 np.array() 是深拷贝，np.asarray() 是浅拷贝`

以上这篇Python读取YUV文件,并显示的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持服务器之家。

原文链接：https://blog.csdn.net/lovelyaiq/article/details/76067085

Python读取YUV文件,并显示的方法

相关文章

热门资讯