人脸马赛克

使用opencv实现对视频中的人脸进行打码

python代码如下

import math
import cv2
from moviepy.editor import VideoFileClip, AudioFileClip, CompositeVideoClip
import os
directory = 'output/'
if not os.path.exists(directory):
    os.makedirs(directory)
def separate():
    # 加载视频
    video = VideoFileClip("input.mp4")
    # 提取视频
    video_clip = video.subclip(0, 10)
    # 提取视频的播放音频
    audio_clip = video_clip.audio
    # 保存视频和音频
    video_clip.write_videofile("./output/video.mp4")
    audio_clip.write_audiofile("./output/audio.mp3")

def merge():
    # 加载视频和音频
    video = VideoFileClip("output/video1.mp4")
    audio = AudioFileClip("output/audio.mp3")
    # 将音频合成到视频中
    video_with_audio = CompositeVideoClip([video.set_audio(audio)])

    # 保存合并后的视频
    video_with_audio.write_videofile("output/output.mp4")

# 加载人脸检测器（使用OpenCV自带的Haar级联分类器）
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
# 马赛克处理函数
def apply_mosaic(image, x, y, w, h, level):
    # 计算人脸区域的尺寸
    face_roi = image[y:y + h, x:x + w]
    # 根据马赛克级别进行降采样
    mosaic_roi = cv2.resize(face_roi, (w // int(math.pow(2,level)), h // int(math.pow(2,level))), interpolation=cv2.INTER_NEAREST)
    mosaic_roi = cv2.resize(mosaic_roi, (w, h), interpolation=cv2.INTER_NEAREST)
    # 将马赛克区域覆盖回原始图像
    image[y:y + h, x:x + w] = mosaic_roi
    return image
separate()
# 读取视频文件
video = cv2.VideoCapture('output/video.mp4')
print("视频处理中......")
# 获取视频的帧数、宽度和高度
frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
frame_width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
# 初始化空白视频
processed_video = cv2.VideoWriter('output/video1.mp4', cv2.VideoWriter_fourcc(*'mp4v'), video.get(cv2.CAP_PROP_FPS),
                                  (frame_width, frame_height))
# 逐帧处理视频
for i in range(frame_count):
    # 读取当前帧的图像
    ret, frame = video.read()
    if ret:
        # 将图像转换为灰度图像
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

        # 进行人脸检测
        faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))

        # 对于每个检测到的人脸
        for (x, y, w, h) in faces:
            # 根据马赛克级别应用马赛克处理
            frame = apply_mosaic(frame, x, y, w, h, level=4)

        # 将处理后的当前帧添加到空白视频中
        processed_video.write(frame)
# 释放资源
video.release()
processed_video.release()
print("处理完成")
merge()
filenames = ["output/audio.mp3","output/video1.mp4","output/video.mp4"]
for filename in filenames:
    # 检查文件是否存在
    if os.path.exists(filename):
        # 删除文件
        os.remove(filename)
    else:
        pass

输入文件的名称要为input.mp4。需要和代码在同一目录。
打码后的视频输入在./output/video.mp4