LivePortrait 本地部署教程,强大且开源的可控人像AI视频生成

 
 

 

 

1,准备工作,本地下载代码并准备环境,运行命令前需安装git

没有外网环境的朋友可以直接点击【网盘打包下载

git clone https://github.com/KwaiVGI/LivePortrait
cd LivePortrait

# create env using conda
conda create -n LivePortrait python=3.9
conda activate LivePortrait

# install dependencies with pip
# for Linux and Windows users
pip install -r requirements.txt
# for macOS with Apple Silicon users
pip install -r requirements_macOS.txt

注意:确保您的系统已安装FFmpeg,包括ffmpegffprobe!不会安装?看这个FFmpeg 【安装教程

2. 下载预训练权重

下载预训练权重的最简单方法是从 HuggingFace 下载:

# first, ensure git-lfs is installed, see: https://docs.github.com/en/repositories/working-with-files/managing-large-files/installing-git-large-file-storage
git lfs install
# clone and move the weights
git clone https://huggingface.co/KwaiVGI/LivePortrait temp_pretrained_weights
mv temp_pretrained_weights/* pretrained_weights/
rm -rf temp_pretrained_weights

 

非海外用户,没有外网环境的朋友,你可以从Google Drive百度云网盘下载所有预训练权重。解压并将它们放在 中./pretrained_weights

确保目录结构如下,或包含:

pretrained_weights
├── insightface
│   └── models
│       └── buffalo_l
│           ├── 2d106det.onnx
│           └── det_10g.onnx
└── liveportrait
    ├── base_models
    │   ├── appearance_feature_extractor.pth
    │   ├── motion_extractor.pth
    │   ├── spade_generator.pth
    │   └── warping_module.pth
    ├── landmark.onnx
    └── retargeting_models
        └── stitching_retargeting_module.pth

3.推理使用

# For Linux and Windows
python inference.py

# For macOS with Apple Silicon, Intel not supported, this maybe 20x slower than RTX 4090
PYTORCH_ENABLE_MPS_FALLBACK=1 python inference.py

如果脚本成功运行,你会得到一个名为 的输出mp4文件animations/s6--d0_concat.mp4。此文件包含以下结果:驾驶视频,输入图像或视频,以及生成的结果。

图像

或者您可以通过指定-s和参数-d来更改输入

# source input is an image
python inference.py -s assets/examples/source/s9.jpg -d assets/examples/driving/d0.mp4

# source input is a video ✨
python inference.py -s assets/examples/source/s13.mp4 -d assets/examples/driving/d0.mp4

# more options to see
python inference.py -h

参照视频自动裁剪 📢📢📢

 

要使用您自己的参照视频,我们建议:⬇️

  • 将其裁剪为1:1 的宽高比(例如 512×512 或 256×256 像素),或通过 启用自动裁剪--flag_crop_driving_video
  • 重点关注头部区域,与示例视频类似。
  • 尽量减少肩部运动。
  • 确保参照视频的第一帧是正面且表情中性

以下是自动裁剪的案例--flag_crop_driving_video

python inference.py -s assets/examples/source/s9.jpg -d assets/examples/driving/d13.mp4 --flag_crop_driving_video

如果觉得自动裁剪的效果不好,您可以修改--scale_crop_driving_video--vy_ratio_crop_driving_video选项来调整比例和偏移量,或者手动进行调整。

动作模板制作

 

您还可以使用自动生成的以 结尾的运动模板文件来.pkl加速推理,并保护隐私,例如:

python inference.py -s assets/examples/source/s9.jpg -d assets/examples/driving/d5.pkl # portrait animation
python inference.py -s assets/examples/source/s13.mp4 -d assets/examples/driving/d5.pkl # portrait video editing

4. Gradio 可视化界面操作

在Gradio的可视化界面下可以获得更好的体验,适合新手使用,只需运行下面安装代码即可:

# For Linux and Windows users (and macOS with Intel??)
python app.py

# For macOS with Apple Silicon users, Intel not supported, this maybe 20x slower than RTX 4090
PYTORCH_ENABLE_MPS_FALLBACK=1 python app.py

您可以指定--server_port、、--share参数--server_name来满足您的需求!

🚀 它们还提供了加速选项--flag_do_torch_compile。首次推理会触发优化过程(约一分钟),使后续推理速度提高 20-30%。性能提升可能因 CUDA 版本的不同而有所差异。

# enable torch.compile for faster inference
python app.py --flag_do_torch_compile

注意:Windows 和 macOS 不支持此方法。或者,在HuggingFace上轻松尝试一下🤗

5. 推理速度评估

 

下方提供了一个脚本来评估每个模块的推理速度:

# For NVIDIA GPU
python speed.py

以下是使用原生 PyTorch 框架在 RTX 4090 GPU 上推断一帧的结果torch.compile

模型 参数(米) 模型大小(MB) 推理(毫秒)
外观特征提取器 0.84 3.3 0.82
运动提取器 28.12 108 0.84
铲形发电机 55.37 212 7.59
变形模块 45.53 174 5.21
拼接和重定向模块 0.23 2.3 0.31

注意:拼接和重定向模块的值代表三个连续 MLP 网络的组合参数数量和总推理时间。

当然如果你没有一张好的显卡,无法本地运行,那么可以在huggingface上免费体验:【点击前往】在线使用

详细的LivePortrait安装教程如下:

THE END
喜欢就支持一下吧
点赞2178 分享
Claude 3 最新注册使用教程!性能完爆GPT4!-零度博客

Claude 3 最新注册使用教程!性能完爆GPT4!

【重点】 准备好美国 或 英国 IP 节点  (Claude 3 目前仅对部分国家地区开放),如果你没有代理节点,可以使用Warp【点击下载】 访问「Claude 3 官网」注册自己的账号 通过邮箱注册账号,推荐使...
admin的头像-零度博客admin
1.7W+1527
Phi-3 免费开源的最强大模型!本地一键部署!!-零度博客

Phi-3 免费开源的最强大模型!本地一键部署!!

Phi-3  Mini Phi-3 Mini 是一个拥有 38 亿参数的轻量级、最先进的开放模型,使用 Phi-3 数据集进行训练,该数据集包含合成数据和经过过滤的公开网站数据,重点关注高质量和推理密集属性。 该模...
admin的头像-零度博客admin
2.1W+2176
Geek Uninstaller 、CCleaner 电脑优化的必备软件-零度博客

Geek Uninstaller 、CCleaner 电脑优化的必备软件

      1.Geek Uninstaller 卸载神器: 【下载】 2.CCleaner 电脑清理工具:【下载】 选择免费即可! 3.ChromeOS 最新安装教程:   https://youtu.be/j_CDBBRIjk0  
admin的头像-零度博客admin
1.7W+2251
切勿错过!7款“限时免费”软件,下载即赚到!!白嫖好几百刀... | 零度解说-零度博客
macOS 漏洞“Powerdir” 细节 被微软公开!请尽快修复-零度博客

macOS 漏洞“Powerdir” 细节 被微软公开!请尽快修复

最近微软公开了 macOS漏洞“Powerdir”细节 Microsoft's 365 Defender Research 团队今天上午公开了称之为“Powerdir”的 macOS 漏洞细节,该漏洞让攻击者绕过 Transparency、Consent 和 Contro...
是什么导致加密货币价格下跌以及如何在崩溃中幸存?-零度博客

是什么导致加密货币价格下跌以及如何在崩溃中幸存?

加密市场波动很大,经常崩溃。在这里,我们将讨论价格下跌的原因以及如何度过加密货币崩盘。 如果你可以用一个词来描述加密市场,那很可能是“不可预测的”。加密货币的价值不断波动,通常不知...
admin的头像-零度博客admin
1.5W+2251
最火爆的 AI绘图模型: ChilloutMix 等其它5个热门模型下载!含提示词-零度博客

最火爆的 AI绘图模型: ChilloutMix 等其它5个热门模型下载!含提示词

不得不承认的是,最近AI绘图火的一塌糊涂!各种AI美图层出不穷,大家看到的很多网红脸AI小姐姐基本上都来自ChilloutMix 然而在ChilloutMix爆火的时候,作者TASUKU2023突然删掉了自己在网上分享...
admin的头像-零度博客admin
1.6W+6539
一键将Telegram 改成中文版(iOS/Android/电脑版适用)-零度博客

一键将Telegram 改成中文版(iOS/Android/电脑版适用)

Telegram现在已经越来越受大家青睐,许多群组都移到Telegram上,身为使用者的你应该有发现Telegram App内的语言设定并没有中文选项, 对于看英文介面不习惯的人来说还真的相当困扰,今天要来教...
免费开源好用的 Office 软件推荐!-零度博客

免费开源好用的 Office 软件推荐!

  1.Libreoffice  【官网】 LibreOffice 是一款功能强大的办公软件,默认使用开放文档格式 (OpenDocument Format , ODF), 并支持 *.docx, *.xlsx, *.pptx 等其他格式。 它包含了 Writer, ...
admin的头像-零度博客admin
1.8W+2250