前言
人脸识别是一门比较成熟的技术。
它的身影随处可见,刷脸支付,信息审核,监控搜索等,除了这些常规操作,还可以对视频里的特定人物进行打码。
知识点:
- OpenCV
- face_recognition
- subprocess
- ffmpeg
开发环境:
- Python 3.6
- Pycharm
face_recognition
face_recognition使用世界上最简单的人脸识别工具,在Python或命令行中识别和操作人脸。
使用dlib最先进的人脸识别技术构建而成,并具有深度学习功能。 该模型在Labeled Faces in the Wild基准中的准确率为99.38%。
如何安装
WIN R 输入CMD,打开指命令提示符
代码语言:javascript复制pip install face_recognition
OpenCV
OpenCV是一个用于图像处理、分析、机器视觉方面的开源函数库. 无论你是做科学研究,还是商业应用,opencv都可以作为你理想的工具库,因为,对于这两者,它完全是免费的。
如何安装
WIN R 输入CMD,打开指命令提示符
代码语言:javascript复制pip install opencv-python
部分代码
导入工具
代码语言:javascript复制import cv2
import face_recognitionimport matplotlib.pyplot as plt
# %matplotlib inline # 在 jupyter 中使用的时候,去掉注释
import ffmpy3
import subprocessimport osfrom PIL import Image
图像处理,自动跟踪打码
代码语言:javascript复制def mask_video(input_video, output_video, mask_path='mask.jpg'):
# 打码图片 mask = cv2.imread(mask_path) # 读取视频 cap = cv2.VideoCapture(input_video)
# 读取视频参数,fps、width、heigth CV_CAP_PROP_FPS = 5
CV_CAP_PROP_FRAME_WIDTH = 3
CV_CAP_PROP_FRAME_HEIGHT = 4
v_fps = cap.get(CV_CAP_PROP_FPS)
v_width = cap.get(CV_CAP_PROP_FRAME_WIDTH)
v_height = cap.get(CV_CAP_PROP_FRAME_HEIGHT)
# 设置写视频参数,格式为 mp4 size = (int(v_width), int(v_height))
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
out = cv2.VideoWriter(output_video, fourcc, v_fps, size)
已知人脸
代码语言:javascript复制known_image = face_recognition.load_image_file("tmr.jpg")
biden_encoding = face_recognition.face_encodings(known_image)[0]
读取视频
代码语言:javascript复制 cap = cv2.VideoCapture(input_video)
while (cap.isOpened()):
ret, frame = cap.read()
if ret:
# 检测人脸 face_locations = face_recognition.face_locations(frame) # print(face_locations)
# 检测每一个人脸 for (top_right_y, top_right_x, left_bottom_y, left_bottom_x) in face_locations:
unknown_image = frame[top_right_y - 50:left_bottom_y 50, left_bottom_x - 50:top_right_x 50]
print(face_recognition.face_encodings(unknown_image))
if face_recognition.face_encodings(unknown_image) != []:
unknown_encoding = face_recognition.face_encodings(unknown_image)[0]
# 对比结果 results = face_recognition.compare_faces([biden_encoding], unknown_encoding) # 是仝卓,就将打码贴图。 if results[0] == 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
保存视频
代码语言:javascript复制if __name__ == '__main__':
# 将音频保存为cut.mp3 video2mp3(file_name='cut.mp4')
# 处理视频,自动打码,输出视频为output.mp4
mask_video(input_video='cut.mp4', output_video='output.mp4')
# 为 output.mp4 处理好的视频添加声音
video_add_mp3(file_name='output.mp4', mp3_file='cut.mp3')