原标题:PyTorch实现EverybodyDanceNow
今天介绍一个比较有趣的开源项目,这个项目来自于Berkeley AI实验室,
将一个源视频的舞姿转移给目标视频人物对象,实现人物的动作迁移,非常流畅,
如下图所示(有WIFI的童鞋可以感受一下文章末尾的视频):
想阅读原理的童鞋,可以下载该项目论文地址
https://arxiv.org/pdf/1808.07371.pdf
实现开源的github
https://github.com/nyoki-mtl/pytorch-EverybodyDanceNow
Lotayou大神对其重新整理及开源,比较适合小白复现
https://github.com/Lotayou/everybody_dance_now_pytorch
一、需要的环境:
Ubuntu 18.04(但16.04也应该没问题)
Python 3.6
CUDA 9.0.176
PyTorch 0.4.1post2
对于其他必要的包,请使用pip install -r requirements快速安装。由于姿势估计器在Keras中实现,因此该项目需要tensorflow> 1.9.0。如果您使用的是独立的Keras软件包,请更改相应的导入命令./pose_estimator/compute_coordinates_for_video.py,但是将无法以这种方式使用tensorboard。
该项目使用imageio进行视频处理,这需要在pip命令之后下载并安装ffmpeg核心。只需按照错误消息,