roop视频换脸技术如何实现效果?
摘要:roop: one click face swap. 只用一张人脸图片,就能完成视频换脸。 - 项目地址: https:github.coms0md3vroop - Colab 部署: https:github.comdrea
roop: one click face swap. 只用一张人脸图片,就能完成视频换脸。
项目地址: https://github.com/s0md3v/roop
Colab 部署: https://github.com/dream80/roop_colab
本文是本地部署的实践记录。
环境基础
OS: Ubuntu 22.04.2 LTS
Kernel: 5.19.0
CPU: Intel(R) Core(TM) i7-10750H
GPU: NVIDIA GeForce RTX 2060
RAM: 16.0GiB
环境安装
Nvidia 驱动
# 查看可用驱动信息
ubuntu-drivers devices
# 安装 Nvidia 驱动(可选提示 recommended 的)
sudo apt install nvidia-driver-530
# 查看 Nvidia 驱动(查看不了,可重登录或重启)
nvidia-smi
Anaconda
Anaconda: https://www.anaconda.com/download
# 安装 Anaconda, 都 yes
bash Anaconda3-2023.07-1-Linux-x86_64.sh
创建虚拟环境,
conda create -n roop python=3.10 -y
conda activate roop
roop 安装
获取代码,
sudo apt install git -y
git clone --depth 1 https://github.com/s0md3v/roop.git
安装依赖,
cd roop
pip install -r requirements.txt
# 如遇错误,可依提示。例如:
# pip install -r requirements.txt --use-pep517
# 如遇 dependency conflict,可修改依赖版本。例如:
# numpy>=1.23.5
# 若配置镜像
pip config set global.index-url http://mirrors.aliyun.com/pypi/simple
pip config set install.trusted-host mirrors.aliyun.com
cat ~/.config/pip/pip.conf
roop 体验
运行,
cd roop
python run.py --execution-provider cuda
# 如果报错类似 ONNXRuntimeError CUBLAS_STATUS_ALLOC_FAILED,需减少执行线程
# https://github.com/s0md3v/roop/issues/413
python run.py --execution-provider cuda --execution-threads 1
可见窗口,
选择一张人脸图片、一个视频,就能完成转换了。
