Skip to content

nICEnnnnnnnLee/FaceSwap

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

VideoFaceSwap 
简单的视频换脸程序
源自 ButterAndButterfly
Butter, 寓意宅男; Butterfly, 寓意美好的事物。
美好的世界由我们创造!

技术原理

参考了以下两个项目:

  • simple_faceswap

    • 实现了摄像头捕捉的视频流的简单换脸,将其实时显示。
    • 换脸思路是直接整个面部到面部的转换。
  • AI-Change-face-in-the-video

    • 实现了视频中的人脸到指定照片中的人脸的更换,并且输出视频。
      但中间有很多冗余步骤,将视频转换为图片保存,再处理图片,再合并视频。
      这其中多出了大量磁盘读写IO,且视频帧频率信息丢失。
    • 换脸思路是五官到五官的转换。

考虑到项目2的缺陷,我们可以参考一个简单的opencv读取复制视频的例子,对其进行改善。

# opencv读取复制视频
import cv2

video = cv2.VideoCapture("faces/trump.avi")
fps = video.get(cv2.CAP_PROP_FPS)
frameCount = video.get(cv2.CAP_PROP_FRAME_COUNT)
size = (int(video.get(cv2.CAP_PROP_FRAME_WIDTH)), int(video.get(cv2.CAP_PROP_FRAME_HEIGHT)))
videoWriter = cv2.VideoWriter('trans.mp4', cv2.VideoWriter_fourcc(*'MP4V'), fps, size)  
success, frame = video.read()  
index = 1
while success :  
	videoWriter.write(frame)
	success, frame = video.read()
	index += 1
video.release()

效果预览

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

可能的坑

  • 安装dlib报错
    根据报错信息提示,使用pip安装Cmake即可。
    但安装Cmake可能也会报错,你需要根据提示再安装Visual Studio,主要是要保证C++的编译环境。

  • 转换后的视频花屏
    原始的image矩阵是0~255的整型的,经过处理后可能变成了浮点型,需要再进行img.astype(np.uint8)操作。

如何使用

  • 根据requirements.txt安装依赖
  • 修改L15~L18行的参数,指定输入的图片、视频即可。