zoukankan html css js c++ java

【python】30行代码实现视频中的动漫人脸检测（opencv）

视频展示

https://www.bilibili.com/video/BV1hA411p7R9

动图展示

在这里插入图片描述

单图展示

在这里插入图片描述

依赖模块

pip install opencv-python

pip install tqdm

完整代码

import cv2
from tqdm import tqdm


def face_detect(img, cascade_name):
    img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    img_gray = cv2.equalizeHist(img_gray)
    face_cascade = cv2.CascadeClassifier(cascade_name)
    faces = face_cascade.detectMultiScale(img)
    for (x, y, w, h) in faces:
        img = cv2.rectangle(img, (x, y), (x+w, y+h), (255, 0, 255), 5)
    return img


def video_detect(file_name, cascade_name):
    video = cv2.VideoCapture(file_name)  # 加载视频
    fps = video.get(cv2.CAP_PROP_FPS)  # 帧率
    w = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))  # 宽
    h = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))  # 高
    fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')  # 指定视频编码方式
    videoWriter = cv2.VideoWriter('result.mp4', 0x7634706d, fps, (w, h))  # 创建视频写对象
    frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))  # 视频总帧数
    for i in tqdm(range(frame_count)):  # 帧数遍历
        success, img = video.read()  # 读取视频帧
        img = face_detect(img, cascade_name)  # 帧检测
        videoWriter.write(img)  # 视频对象写入


if __name__ == "__main__":
    video_detect('test.mp4', 'lbpcascade_animeface.xml')

必要组件

lbpcascade_animeface.xml：点击下载

温馨提示

lbpcascade_animeface.xml是用于检测动漫人脸的，如果你想检测三次元人脸或猫脸，可以分别使用以下文件：

haarcascade_frontalface_alt.xml：点击下载
haarcascade_frontalcatface.xml：点击下载

注意事项

fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')指定视频的编码方式为mp4v。cv2.VideoWriter的第2个参数需要填视频的编码方式。但为什么我这里填的是0x7634706d而不是fourcc？

因为直接填fourcc会报错：

OpenCV: FFMPEG: tag 0x5634504d/'MP4V' is not supported with codec id 12 and format 'mp4 / MP4 (MPEG-4 Part 14)'
OpenCV: FFMPEG: fallback to use tag 0x7634706d/'mp4v'

而填0x7634706d可以避免报错。