上周看了一个AI换脸的项目,https://github.com/ai-forever/sber-swap
效果图如下:
源图片(脸) + 目标target(脸) = AI算法=》目标身体中的脸识别后替换为源脸
还可以换视频,大家猜一下用的是谁的脸?
我已经下载了代码,并且安装了相关的软件,耗费了好几天的精力,终于在我的破笔记本上把这个AI换脸系统跑通了。迫不及待的试验,效果如下:
有人想体验的可以告诉我哈。
安装方法包括好几个部分
1)项目安装
- Clone this repository
git clone https://github.com/sberbank-ai/sber-swap.gitncd sber-swapngit submodule initngit submodule update
- Install dependent packages
pip install -r requirements.txt
If it is not possible to install onnxruntime-gpu, try onnxruntime instead
- Download weights
sh download_models.sh
2)Cuda和Cudnn版本安装
3)Nvidia驱动安装
上述流程有不少坑,不多说了,一把泪!