>백엔드 개발 >파이썬 튜토리얼 >DeepFakes를 플레이하는 것은 이번이 처음입니다.

DeepFakes를 플레이하는 것은 이번이 처음입니다.

coldplay.xixi
coldplay.xixi앞으로
2020-11-13 16:55:244246검색

python 동영상 튜토리얼 칼럼에서 DeepFakes를 소개합니다. DeepFakes를 플레이하는 것은 이번이 처음입니다.

Goal

저는 DeepFakes를 한번도 만져본 적이 없는데, 갑자기 Bilibili의 영상을 올려서 가지고 놀고 싶습니다. 시도하기가 상당히 번거로워서 제가 겪은 함정에 대한 기록을 여기에 올려드립니다.

이 기사의 목표는 The Singing Trump의 영상을 우리 Chuan Jianguo 동지로 대체하는 것입니다. The Singing Trump的视频换成我们的川建国同志。

最后效果:

视频链接:https://www.bilibili.com/video/BV12p4y1k7E8/

环境说明

本文尝试的环境为linux服务器的环境,因为跑得比较快吧。

Python环境:Anoconda python3.7版本

GPU:K80, 12G显存

DeepFake版本:2.0

其他工具:ffmpeg

素材准备

首先需要准备一个或者多个The Singing Trump的视频,以及川建国同志的视频。用作换脸素材。

视频切分

首先通过ffmpeg将视频素材切分成多个图片。

mkdir output
ffmpeg -i 你的视频.mp4 -r 2 output/video-frame-t-%d.png复制代码

这里视频不一定要mp4,其他格式也行,然后 -r 2表示是2帧,也就是每秒钟采集两张图片,各位可以按照自己的视频尝试。 最后是输出到output文件夹里面,前缀随便定义就好了,名字也不是关键。

这里最好多找几个视频,因为deepfake会提示要保证人脸个数大于200张才会比较好,我这里分别准备了3个视频,一共6个视频。

ffmpeg -i sing_trump1.mp4 -r 2 sing_trump_output/st1-%d.png
ffmpeg -i sing_trump2.flv -r 2 sing_trump_output/st2-%d.png
ffmpeg -i sing_trump3.mp4 -r 2 sing_trump_output/st3-%d.png复制代码
ffmpeg -i trump1.webm -r 2 trump_output/t1-%d.png
ffmpeg -i trump2.mp4 -r 2 trump_output/t2-%d.png
ffmpeg -i trump3.mp4 -r 2 trump_output/t3-%d.png复制代码

弄完了还挺大,乱七八糟加起来3.7个G。

clone代码+装依赖

这里没啥可说的,从github上下代码。

git clone https://github.com/deepfakes/faceswap.git复制代码

然后根据自己的实际情况装环境,我这里是现在PC上装cpu这个,然后在服务器上装nvidia。

抽取脸部

接下来将所有的脸部抽出来。

python3 faceswap.py extract -i trump_output -o trump_output_face
python3 faceswap.py extract -i sing_trump_output -o sing_trump_output_face复制代码

这里抽完了脸就是这样了。

筛选脸部

接下来需要手工把我们不需要的脸都删掉。

修改alignment

在我们调用extract生成脸部时,会自动生成一个校对文件,用于在原图上保存脸部的信息。删除脸部之后,需要将脸部和原图片进行对齐。

这里可以打开gui工具

python3 faceswap.py gui复制代码

然后选择Tools下的Alignments。

接下来选择Remove-Faces,然后输入对齐文件路径,脸的路径,以及原图的路径。

然后点击绿色按钮开始,运行即可。

然后将sing_trump_out也执行同样的操作。

开始训练

接下来就可以开始训练了,-m

🎜🎜최종 효과: 🎜🎜동영상 링크: https://www.bilibili.com/video/BV12p4y1k7E8/🎜

환경 설명🎜🎜본 글에서 시도한 환경은 Linux 서버 환경으로, 실행 속도가 더 빠르기 때문입니다. 🎜🎜Python 환경: Anoconda python3.7 버전🎜🎜GPU: K80, 12G 비디오 메모리🎜🎜DeepFake 버전: 2.0🎜🎜기타 도구: ffmpeg🎜

재료 준비🎜🎜먼저 노래하는 트럼프 동영상과 Chuan Jianguo 동지의 동영상을 하나 이상 준비해야 합니다. 얼굴을 바꾸는 재료로 사용됩니다. 🎜

동영상 분할

🎜먼저 ffmpeg를 통해 동영상 자료를 여러 장의 사진으로 분할합니다. 🎜
python3 ./faceswap.py train -A sing_trump_output_face -ala sing_trump_output/alignments.fsa -B trump_output_face -alb trump_output/alignments.fsa  -m model复制代码
🎜여기 동영상은 mp4일 필요는 없으며 다른 형식도 허용되며 -r 2는 2프레임을 의미합니다. 즉, 초당 2장의 사진이 수집된다는 뜻입니다. 나만의 영상. 마지막으로 출력 폴더에 출력됩니다. 접두사는 임의로 정의할 수 있으며 이름은 중요하지 않습니다. 🎜🎜딥페이크에서는 얼굴 수가 200개 이상인지 확인하라는 메시지가 표시되므로 여기에서 더 많은 동영상을 찾는 것이 가장 좋습니다. 여기에는 3개의 동영상, 총 6개의 동영상을 준비했습니다. 🎜🎜🎜
ffmpeg –i sing_trump2.flv input_frames/video-frame-%d.png 
复制代码
python3 faceswap.py extract -i input_frames -o input_frames_face复制代码
🎜완성되면 꽤 크네요. 엉망진창의 총량은 3.7G입니다. 🎜

복제 코드 + 설치 종속성

🎜여기서는 말할 것도 없습니다. github에서 코드를 다운로드하세요. 🎜
python3 faceswap.py convert -i input_frames/ -o output_frames -m model/复制代码
🎜그런 다음 실제 상황에 맞게 환경을 설치합니다. 여기서는 PC에 CPU를 설치한 다음 서버에 nvidia를 설치합니다. 🎜🎜🎜

얼굴 추출

🎜다음으로 모든 얼굴을 추출합니다. 🎜
ffmpeg -i output_frames/video-frame-%d.png -vcodec libx264 -r 30  out.mp4复制代码
🎜뺨을 맞고 나면 그게 다야. 🎜

얼굴 필터링

🎜다음으로 필요하지 않은 모든 얼굴을 수동으로 삭제해야 합니다. 🎜🎜🎜🎜🎜

정렬 수정

🎜추출을 호출하여 얼굴을 생성하면 교정이 자동으로 수행됩니다. 생성된 파일은 원본 이미지에 얼굴 정보를 저장하는 데 사용됩니다. 얼굴을 삭제한 후에는 원본 사진에 얼굴을 맞춰야 합니다. 🎜🎜여기에서 GUI 도구🎜rrreee🎜를 열고 도구 아래에서 정렬을 선택할 수 있습니다. 🎜🎜다음으로 Remove-Faces를 선택하고 정렬 파일 경로, 얼굴 경로, 원본 이미지 경로를 입력하세요. 🎜🎜그런 다음 녹색 버튼을 클릭하여 시작하고 실행하세요. 🎜🎜그런 다음 sing_trump_out으로 동일한 작업을 수행하세요. 🎜

훈련 시작🎜🎜다음에 훈련을 시작할 수 있습니다. -m 매개변수는 모델이 저장되는 위치입니다. 🎜
python3 ./faceswap.py train -A sing_trump_output_face -ala sing_trump_output/alignments.fsa -B trump_output_face -alb trump_output/alignments.fsa  -m model复制代码

小问题

这里如果用gpu的话,我发现tensorflow2.2开始要用cuda10.1以上,但我这边儿没法装,所以需要用tensorflow1.14或者tensorflow1.15,这就需要deepfake的1.0版本才能用。

github.com/deepfakes/f…

训练截图

我发现faceswap1.0和master分支的操作是一样的,没太大变化。

我这里的速度大概是2分钟100个step。

转换视频

准备视频帧

首先要准备我们要转换的视频,然后把视频切分,这里就不是按照之前的帧数了。

ffmpeg –i sing_trump2.flv input_frames/video-frame-%d.png 
复制代码

这里我的视频是1分41秒。

转换完了大概有3050张图片,也就是差不多30帧的,然后一共7.1G(mac就256G真的有点儿遭不住)

再次对齐一遍

接下来,需要对我们要转换的视频图片再来一遍人脸对齐,首先抽脸。

python3 faceswap.py extract -i input_frames -o input_frames_face复制代码

然后再把多余的脸删掉,像前面的步骤一样的操作用gui工具选择Remove-Faces,然后进行对齐。

对每一帧进行AI换脸

通过convert命令进行转换

python3 faceswap.py convert -i input_frames/ -o output_frames -m model/复制代码

我这里的速度大概是每秒1张图片,不过真正的脸只有600多张,如果脸比较密集的话我估计可能没有那么快,所有的图片转换完大概是5分多钟(这个gpu当时有别的程序在跑真实可能会更快一点儿)。

效果

训练20分钟后

在训练了1200step之后,大概是这个样子,效果看着还不是很好哈,不过已经有点儿意思了。

训练一个小时后

训练一天以后

把图片合成视频

最后通过ffmpeg把图片合成一个视频。

ffmpeg -i output_frames/video-frame-%d.png -vcodec libx264 -r 30  out.mp4复制代码

这里合并完了我发现是2分钟,不过影响也不大,毕竟后面还要进行剪辑,用PR等软件再编辑一下就好了。

总结

看视频可以发现当脸比较小的时候,faceswap并没有识别出来脸,所以也就没有做替换,还是有点儿遗憾。

个人感觉整个deepfake的最费时间的流程其实就是在删掉多余的脸上面。

相关免费学习推荐:python视频教程

위 내용은 DeepFakes를 플레이하는 것은 이번이 처음입니다.의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
이 기사는 juejin.im에서 복제됩니다. 침해가 있는 경우 admin@php.cn으로 문의하시기 바랍니다. 삭제