Python讀取YUV文件,并顯示的方法

更新時間：2018年12月04日 15:04:37 作者：TiRan_Yang

今天小編就為大家分享一篇Python讀取YUV文件,并顯示的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

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

opencv可以讀取的圖片類型比較多,但大多是比較常見的類型,比如".jpg"和".png",但它不能直接讀取YUV格式的文件,需要通過python讀取YUV文件,并進(jìn)行相應(yīng)的轉(zhuǎn)換后,才能被opencv讀取,并進(jìn)行后續(xù)相應(yīng)的處理.

話不多說，直接上程序。

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文件需要轉(zhuǎn)換為PIL的image格式,那么只需在main函數(shù)中,進(jìn)行稍微的處理。

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

另外,如果PIL的image想轉(zhuǎn)換為opencv的mat格式,只需將PIL轉(zhuǎn)換為矩陣的形式。

im_array = np.array(im)
# 也可以用 np.asarray(im) 區(qū)別是 np.array() 是深拷貝，np.asarray() 是淺拷貝

以上這篇Python讀取YUV文件,并顯示的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

Python
YUV

相關(guān)文章

python繪制橫豎條形圖的方法
這篇文章主要為大家詳細(xì)介紹了python繪制橫豎條形圖的方法，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2022-04-04
python標(biāo)準(zhǔn)庫OS模塊詳解
這篇文章主要介紹了python標(biāo)準(zhǔn)庫OS模塊詳細(xì)介紹,需要的朋友可以參考下
2020-03-03
python 統(tǒng)計list中各個元素出現(xiàn)的次數(shù)的幾種方法
這篇文章主要介紹了python 統(tǒng)計list中各個元素出現(xiàn)的次數(shù)的幾種方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2021-02-02
python dict 相同key 合并value的實(shí)例
今天小編就為大家分享一篇python dict 相同key 合并value的實(shí)例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-01-01
使用遺傳算法求二元函數(shù)的最小值
今天小編就為大家分享一篇使用遺傳算法求二元函數(shù)的最小值，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-02-02
Pytorch中關(guān)于nn.Conv2d()參數(shù)的使用
這篇文章主要介紹了Pytorch中關(guān)于nn.Conv2d()參數(shù)的使用，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2023-06-06
python全棧開發(fā)語法總結(jié)
在本篇文章里小編給大家整理的是一篇關(guān)于python全棧開發(fā)語法的相關(guān)總結(jié)內(nèi)容，有興趣的朋友們可以跟著學(xué)習(xí)參考下。
2020-11-11
python文件與路徑管理方法
這篇文章主要介紹了python文件與路徑管理方法，文章屬于python使用技巧的分享，下文圍繞文件與路徑管理相關(guān)內(nèi)容，需要的小伙伴可以參考一下，希望對你有所幫助
2022-02-02
python中閉包Closure函數(shù)作為返回值的方法示例
閉包(closure)是函數(shù)式編程的重要的語法結(jié)構(gòu)，Python也支持這一特性，下面這篇文章主要給大家介紹了關(guān)于python中閉包Closure函數(shù)作為返回值的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，需要的朋友可以參考下。
2017-12-12
Python實(shí)現(xiàn)判斷字符串中包含某個字符的判斷函數(shù)示例
這篇文章主要介紹了Python實(shí)現(xiàn)判斷字符串中包含某個字符的判斷函數(shù),涉及Python自定義函數(shù)中使用find方法針對字符串的簡單判斷查找相關(guān)操作技巧,需要的朋友可以參考下
2018-01-01