Python實現(xiàn)視頻畫質(zhì)增強的示例代碼
前言
前面通過文章 幾行代碼，實現(xiàn)Python捕獲、播放和保存攝像頭視頻！給大家介紹了如何讀取、播放和保存視頻，后面又通過文章 Python美圖技術(shù)也就幾行代碼！給大家介紹了如何對圖像的亮度、對比度、色度或者銳度進行調(diào)整，從而達到基本的圖像處理操作。
其實，那兩篇文章都是鋪墊，都是為了給大家介紹如何對視頻進行畫質(zhì)增強。本文將結(jié)合前面兩篇文章的內(nèi)容，來講講如何對視頻畫質(zhì)進行增強。
想要直接看效果的，可以拉到文末。
原理
不知道大家小時候有沒有玩過這個？
最早的動畫就是這么形成的，記得小時候還有這種小書賣。
其實視頻的原理也是這樣，一個視頻是由很多張圖片組成的，一個圖片是一幀。所以我們要對視頻進行畫質(zhì)增強，可以拆分成對每一幀的圖片進行操作，這個操作我們在前面介紹過。
因此，對視頻進行畫質(zhì)增強的方法可以分為三步：拆分->處理->合成。
實現(xiàn)步驟
拆分
我們在第一篇文章講到過如何捕獲攝像頭的視頻流，以及如何讀取視頻并播放。不管通過哪種方法，我們都是通過幀操作的。所以這里所謂的拆分就是獲取到我們捕獲到的視頻流或者讀取的視頻流的每一幀。
success,?img1?=?cap.read()
????#?如果正確讀取幀，success為True
????if?not?success:
????????break
????cv2.imshow('img1',?img1)就是這么簡單，我們就可以獲取到視頻每一幀了。
處理
獲取到視頻的一幀之后，我們就要把這一幀轉(zhuǎn)換成我們可以處理的格式的圖片。在前面我們介紹如何對圖片進行畫質(zhì)增強的時候，使用的是 ImageEnhance 這個函數(shù)的相關(guān)方法，這個函數(shù)是 PIL 圖像處理庫里面的，所以我們必須把我們每一幀的圖片讀取成 PIL 可以處理的格式：
image?=?Image.fromarray(np.uint8(img1))??#?轉(zhuǎn)換成PIL可以處理的格式
讀取到圖像之后，我們就可以對圖像進行畫質(zhì)增強處理了，這里還是用我們上篇文章中講到的代碼：
#?圖像處理 def?img_enhance(image,?brightness=1,?color=1,contrast=1,sharpness=1): ????#?亮度增強 ????enh_bri?=?ImageEnhance.Brightness(image) ????if?brightness: ????????image?=?enh_bri.enhance(brightness) ????#?色度增強 ????enh_col?=?ImageEnhance.Color(image) ????if?color: ????????image?=?enh_col.enhance(color) ????#?對比度增強 ????enh_con?=?ImageEnhance.Contrast(image) ????if?contrast: ????????image?=?enh_con.enhance(contrast) ????#?銳度增強 ????enh_sha?=?ImageEnhance.Sharpness(image) ????if?sharpness: ????????image?=?enh_sha.enhance(sharpness) ????return?image
合成
圖像處理完，我們需要每一幀圖像進行合成，從而得到我們最終的視頻：
cap?=?cv2.VideoCapture('你的視頻目錄/xxx.mp4')
success,?_?=?cap.read()
#?分辨率-寬度
width?=?int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
#?分辨率-高度
height?=?int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
#?總幀數(shù)
frame_counter?=?int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
video_writer?=?cv2.VideoWriter('輸出.mp4',?cv2.VideoWriter_fourcc('M',?'P',?'4',?'V'),?15,?(width,?height),?True)
while?success:
????success,?img1?=?cap.read()
????try:
????????image?=?Image.fromarray(np.uint8(img1))??#?轉(zhuǎn)換成PIL可以處理的格式
????????img_enhanced?=?img_enhance(image,?2,?2,?2,?3)
????????video_writer.write(np.asarray(img_enhanced))
????????if?cv2.waitKey(1)?&?0xFF?==?ord('q'):
????????????break
????except:
????????break
cap.release()
video_writer.release()
cv2.destroyAllWindows()
我這里讀取的是 mp4 格式的視頻，所以在合成寫視頻文件的時候，我們需要用
cv2.VideoWriter_fourcc('M', 'P', '4', 'V') 這個格式。
我這里沒有對圖片的分辨率進行修改，只是分別獲取原始視頻的分辨率，然后寫入視頻文件的時候，將原始分辨率傳入作為參數(shù)。
如果你需要修改視頻的分辨率的話，可以使用下面的方式：
cv2.resize(src, dsize[, dst[, fx[, fy[, interpolation]]]])
使用 resize 方法就可以了：
resized?=?cv2.resize(img,?(width,?height),?interpolation?=?cv2.INTER_AREA)
效果
我們先來看看處理前的視頻：
處理之后（我這里處理比較隨意，參數(shù)都是隨意寫的）的視頻是這樣子的：
總結(jié)
到此為止，我們的視頻畫質(zhì)增強的功能算是基本實現(xiàn)了，代碼也不復雜，加起來也就這么點。但是，如果要處理成自己滿意的效果，還是需要下一番功夫去調(diào)參數(shù)，去優(yōu)化。甚至針對每一幀可能傳入的參數(shù)都不一樣，這就需要各位自己去慢慢研究了。
以上就是Python實現(xiàn)視頻畫質(zhì)增強的示例代碼的詳細內(nèi)容，更多關(guān)于Python視頻畫質(zhì)增強的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
本篇博客為大家介紹一款新的自動化測試工具，效果類似?selenium，但是這個模塊年輕。模塊名稱為?playwright-python，微軟開源的，是針對?Python?語言的純自動化工具，感興趣的可以了解一下2022-10-10
在本篇文章里小編給大家整理的是一篇關(guān)于python常見的占位符總結(jié)及用法，有興趣的朋友們可以跟著學習參考下。2021-07-07
我們都知道tensorflow框架可以使用tensorboard這一高級的可視化的工具，這篇文章主要介紹了pytorch下tensorboard的使用,需要的朋友可以參考下2021-10-10
在使用pandas進行數(shù)據(jù)統(tǒng)計分析時遇到了問題,找了很久才找到解決辦法,所以下面這篇文章主要給大家介紹了關(guān)于pandas中df.groupby()方法的相關(guān)資料,需要的朋友可以參考下2022-12-12
這篇文章主要給大家介紹了關(guān)于OpenCV-Python給圖像去除水印的多種方法,文中介紹的每種方法都有其適用場景和實現(xiàn)步驟,具體實現(xiàn)需要根據(jù)實際情況選擇和調(diào)整,需要的朋友可以參考下2024-12-12
Python設(shè)計模式結(jié)構(gòu)型代理模式
這篇文章主要介紹了Python設(shè)計模式結(jié)構(gòu)型代理模式,代理模式即Proxy?Pattern，為其他對象提供一種代理以控制對這個對象的訪問，下文內(nèi)容詳細介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-02-02
如何在mac環(huán)境中用python處理protobuf
這篇文章主要介紹了如何在mac環(huán)境中用python處理protobuf,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-12-12
Python內(nèi)存池是Python解釋器為了提高內(nèi)存分配效率而設(shè)計的一種內(nèi)存管理機制,本文主要介紹了Python內(nèi)存池機制的實現(xiàn),具有一定的參考價值,感興趣的可以了解一下2025-04-04
OpenCV（Open Source Computer Vision Library）作為一個強大的計算機視覺庫,提供了豐富的圖像處理和計算機視覺功能,本文將帶領(lǐng)讀者使用Python編程語言,通過簡單的代碼示例,初步掌握OpenCV的圖像處理技術(shù),需要的朋友可以參考下2024-09-09
python web應用程序之Django數(shù)據(jù)庫詳解
這篇文章主要介紹了python web應用程序之Django數(shù)據(jù)庫,本文通過實例代碼給大家介紹的非常詳細,需要的朋友可以參考下2024-06-06