批量跨帧提取人脸视频中的脸部区域帧图像python

qq_37256906

已于 2022-09-16 11:58:54 修改

阅读量1.5k

点赞数 2

CC 4.0 BY-SA版权

文章标签： python 音视频 opencv

于 2022-09-16 11:55:22 首次发布

本文链接：https://blog.csdn.net/qq_37256906/article/details/126887854

代码实现功能

1、所有人脸视频在一个文件夹中。
2、跨帧提取帧图像。
3、截取帧图像的脸部区域。
3、将截取到的所有脸部图像保存到新文件夹中。

代码语言：python3.6

#这个代码实现了批量跨帧提取视频帧图像，并使用MTCNN截取帧图像中的脸部区域，然后保存
#（1）视频来自一个文件夹中
#（2）截取的人脸图像会全部存在新文件夹中

import os
import cv2
from mtcnn.mtcnn import MTCNN
from PIL import Image
import numpy as np
import os

videos_src_path = 'E:/wangqin/forensics++ datasets/manipulated_sequences/NeuralTextures/c23/videos/'
videos_save_path = 'E:/wangqin/forensics++ datasets/Image/fake/NeuralTextures/'

videos = os.listdir(videos_src_path)
detector = MTCNN()

i = 1

for each_video in videos:
    # if not os.path.exists(videos_save_path + '/' + str(i)):
    #     os.mkdir(videos_save_path + '/' + str(i))
    # each_video_save_full_path = os.path.join(videos_save_path, str(i)) + '/'
    each_video_full_path = os.path.join(videos_src_path, each_video)
    cap = cv2.VideoCapture(each_video_full_path)

    print('正在处理第{}视频，一共{}个视频'.format(i,