Python實(shí)現(xiàn)視頻自動(dòng)打碼的示例代碼
序言
我們?cè)谟^(guān)看視頻的時(shí)候,有時(shí)候會(huì)出現(xiàn)一些奇怪的馬賽克,影響我們的觀(guān)影體驗(yàn),那么這些馬賽克是如何精確的加上去的呢?

本次我們就來(lái)用Python實(shí)現(xiàn)對(duì)視頻自動(dòng)打碼!
準(zhǔn)備工作
環(huán)境咱們還是使用 Python3.8 和 pycharm2021 即可
實(shí)現(xiàn)原理
將視頻分為音頻和畫(huà)面;
畫(huà)面中出現(xiàn)人臉和目標(biāo)比對(duì),相應(yīng)人臉進(jìn)行打碼;
處理后的視頻添加聲音;
模塊
手動(dòng)安裝一下 cv2 模塊 ,pip install opencv-python 安裝
素材工具
我們需要安裝一下 ffmpeg 音視頻轉(zhuǎn)碼工具

代碼解析
導(dǎo)入需要使用的模塊
import cv2 import face_recognition # 人臉識(shí)別庫(kù) 99.7% cmake dlib face_recognition import subprocess
將視頻轉(zhuǎn)為音頻
def video2mp3(file_name):
"""
:param file_name: 視頻文件路徑
:return:
"""
outfile_name = file_name.split('.')[0] + '.mp3'
cmd = 'ffmpeg -i ' + file_name + ' -f mp3 ' + outfile_name
print(cmd)
subprocess.call(cmd, shell=False)
打碼
def mask_video(input_video, output_video, mask_path='mask.jpg'):
"""
:param input_video: 需打碼的視頻
:param output_video: 打碼后的視頻
:param mask_path: 打碼圖片
:return:
"""
# 讀取圖片
mask = cv2.imread(mask_path)
# 讀取視頻
cap = cv2.VideoCapture(input_video)
# 視頻 fps width height
v_fps = cap.get(5)
v_width = cap.get(3)
v_height = cap.get(4)
# 設(shè)置寫(xiě)入視頻參數(shù) 格式MP4
# 畫(huà)面大小
size = (int(v_width), int(v_height))
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
# 輸出視頻
out = cv2.VideoWriter(output_video, fourcc, v_fps, size)
# 已知人臉
known_image = face_recognition.load_image_file('tmr.jpg')
biden_encoding = face_recognition.face_encodings(known_image)[0]
cap = cv2.VideoCapture(input_video)
while (cap.isOpened()):
ret, frame = cap.read()
if ret:
# 檢測(cè)人臉
# 人臉區(qū)域
face_locations = face_recognition.face_locations(frame)
for (top_right_y, top_right_x, left_bottom_y, left_bottom_x) in face_locations:
print((top_right_y, top_right_x, left_bottom_y, left_bottom_x))
unknown_image = frame[top_right_y - 50:left_bottom_y + 50, left_bottom_x - 50:top_right_x + 50]
if face_recognition.face_encodings(unknown_image) != []:
unknown_encoding = face_recognition.face_encodings(unknown_image)[0]
# 對(duì)比人臉
results = face_recognition.compare_faces([biden_encoding], unknown_encoding)
# [True]
# 貼圖
if results == [True]:
mask = cv2.resize(mask, (top_right_x - left_bottom_x, left_bottom_y - top_right_y))
frame[top_right_y:left_bottom_y, left_bottom_x:top_right_x] = mask
out.write(frame)
else:
break
音頻添加到畫(huà)面
def video_add_mp3(file_name, mp3_file):
"""
:param file_name: 視頻畫(huà)面文件
:param mp3_file: 視頻音頻文件
:return:
"""
outfile_name = file_name.split('.')[0] + '-f.mp4'
subprocess.call('ffmpeg -i ' + file_name + ' -i ' + mp3_file + ' -strict -2 -f mp4 ' + outfile_name, shell=False)
完整代碼
import cv2
import face_recognition # 人臉識(shí)別庫(kù) 99.7% cmake dlib face_recognition
import subprocess
def video2mp3(file_name):
outfile_name = file_name.split('.')[0] + '.mp3'
cmd = 'ffmpeg -i ' + file_name + ' -f mp3 ' + outfile_name
print(cmd)
subprocess.call(cmd, shell=False)
def mask_video(input_video, output_video, mask_path='mask.jpg'):
# 讀取圖片
mask = cv2.imread(mask_path)
# 讀取視頻
cap = cv2.VideoCapture(input_video)
# 視頻 fps width height
v_fps = cap.get(5)
v_width = cap.get(3)
v_height = cap.get(4)
# 設(shè)置寫(xiě)入視頻參數(shù) 格式MP4
# 畫(huà)面大小
size = (int(v_width), int(v_height))
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
# 輸出視頻
out = cv2.VideoWriter(output_video, fourcc, v_fps, size)
# 已知人臉
known_image = face_recognition.load_image_file('tmr.jpg')
biden_encoding = face_recognition.face_encodings(known_image)[0]
cap = cv2.VideoCapture(input_video)
while (cap.isOpened()):
ret, frame = cap.read()
if ret:
# 檢測(cè)人臉
# 人臉區(qū)域
face_locations = face_recognition.face_locations(frame)
for (top_right_y, top_right_x, left_bottom_y, left_bottom_x) in face_locations:
print((top_right_y, top_right_x, left_bottom_y, left_bottom_x))
unknown_image = frame[top_right_y - 50:left_bottom_y + 50, left_bottom_x - 50:top_right_x + 50]
if face_recognition.face_encodings(unknown_image) != []:
unknown_encoding = face_recognition.face_encodings(unknown_image)[0]
# 對(duì)比人臉
results = face_recognition.compare_faces([biden_encoding], unknown_encoding)
# [True]
# 貼圖
if results == [True]:
mask = cv2.resize(mask, (top_right_x - left_bottom_x, left_bottom_y - top_right_y))
frame[top_right_y:left_bottom_y, left_bottom_x:top_right_x] = mask
out.write(frame)
else:
break
def video_add_mp3(file_name, mp3_file):
outfile_name = file_name.split('.')[0] + '-f.mp4'
subprocess.call('ffmpeg -i ' + file_name + ' -i ' + mp3_file + ' -strict -2 -f mp4 ' + outfile_name, shell=False)
if __name__ == '__main__':
# 1.
video2mp3('cut.mp4')
# 2.
mask_video(input_video='cut.mp4',output_video='output.mp4')
# 3.
video_add_mp3(file_name='output.mp4',mp3_file='cut.mp3')
兄弟們,快去試試吧!
到此這篇關(guān)于Python實(shí)現(xiàn)視頻自動(dòng)打碼的示例代碼的文章就介紹到這了,更多相關(guān)Python視頻打碼內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python 彈窗提示警告框MessageBox的實(shí)例
今天小編就為大家分享一篇python 彈窗提示警告框MessageBox的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06
在Python的Django框架中用流響應(yīng)生成CSV文件的教程
這篇文章主要介紹了在Python的Django框架中用流響應(yīng)生成CSV文件的教程,作者特別講到了防止CSV文件中的中文避免出現(xiàn)亂碼等問(wèn)題,需要的朋友可以參考下2015-05-05
Python實(shí)現(xiàn)OCR識(shí)別之pytesseract案例詳解
這篇文章主要介紹了Python實(shí)現(xiàn)OCR識(shí)別之pytesseract案例詳解,本篇文章通過(guò)簡(jiǎn)要的案例,講解了該項(xiàng)技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-07-07
利用Python實(shí)現(xiàn)QQ實(shí)時(shí)到賬免簽支付功能
這篇文章主要介紹了利用Python實(shí)現(xiàn)QQ實(shí)時(shí)到賬免簽支付功能,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2021-03-03
Python psutil模塊簡(jiǎn)單使用實(shí)例
這篇文章主要介紹了Python psutil模塊簡(jiǎn)單使用實(shí)例,本文直接給出使用腳本,實(shí)現(xiàn)查看cpu的信息、查看內(nèi)存信息、查看系統(tǒng)啟動(dòng)時(shí)間、查看網(wǎng)卡信息等,需要的朋友可以參考下2015-04-04
python scrapy簡(jiǎn)單模擬登錄的代碼分析
在本篇文章里小編給大家整理的是一篇關(guān)于python scrapy簡(jiǎn)單模擬登錄的代碼分析,有興趣的朋友們可以學(xué)習(xí)參考下。2021-07-07

