比特技术教程 -24小时在线 - 古典小说 - 文学小说 - 关于我们
您的当前位置:首页 > 开发语言 > Python >

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

来源:比特技术网 编辑:比特在线 时间:2018-12-06点击:
导读:Python读取YUV格式文件,并使用opencv显示的方法 opencv可以读取的图片类型比较多,但大多是比较常见的类型,比如.jpg和.png,但它不能直接读取YUV格式的文件,需要通过python读取YUV文件,并进行相应的转换后,才能被opencv读取,并进行后续相应的处理. 话不多说,

Python读取YUV格式文件,并使用opencv显示的方法

opencv可以读取的图片类型比较多,但大多是比较常见的类型,比如".jpg"和".png",但它不能直接读取YUV格式的文件,需要通过python读取YUV文件,并进行相应的转换后,才能被opencv读取,并进行后续相应的处理.

话不多说,直接上程序。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
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文件,并显示的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。


来源:比特技术在线(www.byte.online),欢迎分享本文,转载请保留出处!
http://www.byte.online/a/kaifayuyan/Python/2018/871469.html
责任编辑:比特在线

    打赏

    取消

    感谢您的支持,我会继续努力的!

    扫码支持
    扫码打赏,你说多少就多少

    打开支付宝扫一扫,即可进行扫码打赏哦

    网友评论:

    Copyright © www.byte.online 比特技术在线 版权所有 粤ICP备15038486号-3
    本站所有资料来源于网络,由个人收集整理,仅限于个人学习和研究,如果侵犯您的权益请联系我们删除!
    Top