unravel 让图片唱歌详细教程 Real Time Image Animation 项目

01 下载资源

项目资源

原始项目

Real Time Image Animation

项目地址：https://github.com/anandpawara/

打包修改好的代码、权重文件、视频图片素材, 下载下面这个即可

下载链接（密码:amz5）：https://pan.baidu.com/s/1TEd7SOaO5mzPaxpOh2pALQ

资源来源于Jack-Cui 知乎文章https://zhuanlan.zhihu.com/p/193119216

02 环境安装

1. 安装Python

为避免报错, 选择和项目一样的Python版本Python3.7

https://www.python.org/ftp/python/3.7.9/python-3.7.9-amd64.exe

安装过程: 略

2.安装依赖包

安装项目依赖包, 进入到进入到项目目录Real_Time_Image_Animation-master, 按住shift右键点击目录空白处在此处打开powershell窗口, 输入下面命令安装

python -m pip install -r requirements.txt 

3.安装pytorch

进入到pytorch官网https://pytorch.org/get-started/locally/ 安装合适的版本

我选择了最新的版本

在命令行输入

pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html 

这个包比较大, 大约有1个G, 如果网络不好可以下载包再安装

4.安装CUDA

安装CUDA必须是nvidia显卡

建议先升级显卡驱动, 驱动版本越新能安装的cuda版本越高, 也能安装旧版本的cuda

我安装的是CUDA Toolkit 11.1 实测也能兼容pytorch 1.6.0

https://developer.nvidia.com/cuda-toolkit-archive

安装时选择自定义安装, 为避免不必要的错误, 只勾选如图所CUDA下那四项就可以了

5.安装ffmpeg

下载https://www.gyan.dev/ffmpeg/builds/packages/ffmpeg-4.3.1-2020-10-01-full_build.7z

或蓝奏云盘https://wws.lanzous.com/iRzjbhg3xod

下载好后解压把bin目录添加到环境变量

cmd输入ffmpeg -version 显示版本号等信息即添加成功

使用

在项目Real_Time_Image_Animation-master文件夹下打开命令行运行

python `项目主程序` -i `需要制作的源图路径` -c  `权重文件路径` -v `输入的视频文件(unravel源视频, 也可以自己制作)` 

例如

python image_animation.py -i Inputs/trump2.png -c checkpoints/vox-cpk.pth.tar -v 1.mp4 

注意输入的源图最好是长宽比1:1的正方形分辨率256x256像素以上, 并且五官清晰

自己做的输入源视频分辨率是256x256像素, 时长2分钟内, 不然会报内存不足