上周看了一个AI换脸的项目,https://github.com/ai-forever/sber-swap

效果图如下:

源图片(脸) + 目标target(脸) = AI算法=》目标身体中的脸识别后替换为源脸

还可以换视频,大家猜一下用的是谁的脸?

我已经下载了代码,并且安装了相关的软件,耗费了好几天的精力,终于在我的破笔记本上把这个AI换脸系统跑通了。迫不及待的试验,效果如下:

有人想体验的可以告诉我哈。

安装方法包括好几个部分

1)项目安装

  1. Clone this repository
git clone https://github.com/sberbank-ai/sber-swap.git
cd sber-swap
git submodule init
git submodule update
  1. Install dependent packages
pip install -r requirements.txt

If it is not possible to install onnxruntime-gpu, try onnxruntime instead

  1. Download weights
sh download_models.sh

2)Cuda和Cudnn版本安装

3)Nvidia驱动安装

上述流程有不少坑,不多说了,一把泪!