gpt4 book ai didi

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

转载 作者:qq735679552 更新时间:2022-09-27 22:32:09 30 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章Python读取YUV文件,并显示的方法由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

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

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

最后此篇关于Python读取YUV文件,并显示的方法的文章就讲到这里了,如果你想了解更多关于Python读取YUV文件,并显示的方法的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com