欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-03-21 14:04:48
python读取yuv格式文件,并使用opencv显示的方法 opencv可以读取的图片类型比较多,但大多是比较常见的类型,比如".jpg"和".png",但它不能直接读...

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文件,并显示的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。