python人物視頻背景替換實(shí)現(xiàn)虛擬空間穿梭
引言
近期網(wǎng)上這位賣蜂蜜的小伙鬼畜挺火的,大家質(zhì)疑背景造假,這里我就帶著大家實(shí)現(xiàn)“背景造假”(PS:原視頻小伙是在真實(shí)場(chǎng)景拍攝的)
準(zhǔn)備工作
在實(shí)現(xiàn)該功能之前,我們需要準(zhǔn)備好python==3.7 然后執(zhí)行:
pip install mediapipe
方案一: PC端可以選擇外界攝像頭或者連接網(wǎng)絡(luò)攝像頭,最好挑選一個(gè)純屬的背景板作為視頻畫面背景(這樣有利于任務(wù)分割);
方案二: 網(wǎng)上下載有人物活動(dòng)的視頻,然后用下載的視頻替代連接攝像頭。
到這里我們基礎(chǔ)物件已經(jīng)搭建好了,本實(shí)驗(yàn)代碼選擇方案二,對(duì)下載視頻分析,其shape=(740,640,3)。接下來(lái)我將帶領(lǐng)大家實(shí)現(xiàn)三個(gè)版本的背景替換
純RGB背景替換
基礎(chǔ)板:將人物提取出來(lái),替換成設(shè)定好的純色背景(可以通過(guò)RGB設(shè)定)。其基本流程如下:
1. 輸入視頻
2. 獲取視頻幀
3. 提取幀內(nèi)人物圖像
4. 設(shè)定背景RGB值
5. 創(chuàng)建同視頻幀同shape的RGB圖像
6. 結(jié)合背景圖與人物圖
7. 輸出結(jié)合圖
示例demo:
import cv2 import mediapipe as mp import numpy as np mp_drawing = mp.solutions.drawing_utils mp_selfie_segmentation = mp.solutions.selfie_segmentation BG_COLOR = (0, 0, 225) # red cap = cv2.VideoCapture('111.mp4') with mp_selfie_segmentation.SelfieSegmentation( model_selection=0) as selfie_segmentation: bg_image = None while cap.isOpened(): success, image = cap.read() if not success: break image = cv2.cvtColor(cv2.flip(image, 1), cv2.COLOR_BGR2RGB) image.flags.writeable = False results = selfie_segmentation.process(image) image.flags.writeable = True image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) condition = np.stack( (results.segmentation_mask,) * 3, axis=-1) > 0.8 if bg_image is None: bg_image = np.zeros(image.shape, dtype=np.uint8) bg_image[:] = BG_COLOR print(bg_image.shape) output_image = np.where(condition, image, bg_image) cv2.imshow('MediaPipe Selfie Segmentation', output_image) if cv2.waitKey(5) & 0xFF == 27: break cap.release()
原視頻節(jié)選幀&替換背景幀:
自定義圖像背景板替換
由于我們已知圖像的shape,我們可以在網(wǎng)上挑選自己喜歡的背景圖,然后調(diào)整背景圖的shape,使其與視頻圖像保持一致。整體流程圖下:
1. 下載選擇喜歡的背景圖像
2. 調(diào)整下載圖像的shape=(740,604,3)
3. 讀取視頻
4. 獲取視頻幀
5. 獲取視頻幀內(nèi)人物圖像
6. 結(jié)合背景圖和人物圖像
7. 輸出視頻畫面
修改圖像大小并進(jìn)行50%縮放后的背景圖:
背景圖結(jié)合人物圖:
這里我的背景圖選擇的有些單調(diào),也不是那么適合,望讀者見(jiàn)諒! 代碼部分修改:
bg_image = cv2.imread("BBB.jpg")
以上就是python實(shí)現(xiàn)人物視頻背景替換的詳細(xì)內(nèi)容,更多關(guān)于python視頻背景替換的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
Python?selenium實(shí)現(xiàn)大麥網(wǎng)自動(dòng)購(gòu)票過(guò)程解析
大麥網(wǎng)是中國(guó)綜合類現(xiàn)場(chǎng)娛樂(lè)票務(wù)營(yíng)銷平臺(tái),業(yè)務(wù)覆蓋演唱會(huì)、?話劇、音樂(lè)劇、體育賽事等領(lǐng)域,今天我們要用代碼來(lái)實(shí)現(xiàn)他的購(gòu)票過(guò)程,感興趣的朋友一起看看吧2022-05-05python使用xlrd和xlwt讀寫Excel文件的實(shí)例代碼
這篇文章主要介紹了python使用xlrd和xlwt讀寫Excel文件的實(shí)例代碼,非常不錯(cuò),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2018-09-09python實(shí)現(xiàn)批量修改圖片格式和尺寸
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)批量修改圖片格式和尺寸的方法,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2018-06-06python神經(jīng)網(wǎng)絡(luò)學(xué)習(xí)數(shù)據(jù)增強(qiáng)及預(yù)處理示例詳解
這篇文章主要為大家介紹了python神經(jīng)網(wǎng)絡(luò)學(xué)習(xí)數(shù)據(jù)增強(qiáng)及預(yù)處理示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-05-05Python實(shí)現(xiàn)GIF動(dòng)圖加載和降幀的方法詳解
這篇文章主要為大家詳細(xì)介紹了如何利用Python和Pygame實(shí)現(xiàn)GIF動(dòng)圖加載和降幀的效果,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2023-02-02python每次處理固定個(gè)數(shù)的字符的方法總結(jié)
使用python每次處理固定個(gè)數(shù)的字符,很多情況下都會(huì)遇到。本文對(duì)可能的方法做下總結(jié),供各位朋友學(xué)習(xí)參考2013-01-01