Python+AI让静态图片动起来

￡神魔★判官ぃ 2023-09-28 13:44 47阅读 0赞

哈喽，大家好。

大家在刷短视频的时候有没有刷到这样的视频，视频里一张静态的人物图片能动起来，如：歪歪头、眨眨眼。

类似于下面这种效果

![在这里插入图片描述][5556bcbecdef41a2b4dfcec45c318c9c.jpeg]  
最左侧的人物是原始的动作，上面是静态图片。通过AI技术便可以让最左侧人物的动作运用到上面的静态图片上，从而让张图都可以做出相同的动作。

这种技术一般是基于的GAN（生成对抗网络）来实现，今天我给大家分享一个开源项目，能够复现上面的效果，可以做些有趣的项目，也可以怀念故人。

### 1. 运行项目 ###

项目地址：https://github.com/AliaksandrSiarohin/first-order-model

首先，git clone将项目下载到本地，进入项目安装依赖

git clone https://github.com/AliaksandrSiarohin/first-order-model.git
    cd first-order-model
    pip install -r requirements.txt

然后，在项目首页Pre-trained checkpoint标题下，找到模型下载链接，下载模型文件，模型有很多，我用的是vox-adv-cpk.pth.tar。

准备好模型文件后，在项目根目录下执行下面命令即可

python demo.py  \
      --config config/vox-adv-256.yaml \
      --driving_video src_video.mp4 \
      --source_image src_img.jpg \
      --checkpoint weights/vox-adv-cpk.pth.tar

解释下参数：

–config：模型配置文件，在源文件中有  
–driving\_video：提供动作的视频  
–source\_image：需要加动效的静态图片  
–checkpoint：刚刚下载的模型文件  
运行完成后，会看到如下输入

![在这里插入图片描述][3cd5744c24a94eb0b96ff498048b8e64.png]

该项目使用PyTorch搭建神经网络，支持GPU和CPU运行，所以如果你的电脑只有CPU，运行会比较慢。

我是在CPU下运行的，从上图可以看到，driving\_video只有 31 帧。如果你也是CPU运行，最好控制driving\_video视频的时长，不然运行时长会比较长。

有了该项目，可以自己做些比较有意思的尝试

![在这里插入图片描述][458703eb447d406c8113a49532b08f0b.jpeg]

### 2. Python API ###

上面教大家按照官网在命令行运行该项目。

有些朋友可能想在Python项目中调用，所以我在demo.py中抽取了核心代码，封装了一个Pyhton API

![在这里插入图片描述][1a59fb82970348e0af1c425f1217fa95.png]

有需要的朋友可以下载这个文件，放到与first-order-model同一级的目录下，按照下面代码调用即可

fom = FOM()
    # 查看驱动视频，驱动视频最好裁剪为480 x 640 大小的视频
    driving_video = ''
    # 被驱动的画面
    source_image = ''
    # 输出视频
    result_video = ''
    # 驱动画面
    fom.img_to_video(driving_video, source_image, result_video)

源代码和模型文件已经整理好了，在公众号回复 图片动起来 即可获取。

最后，感谢你的关注，我将持续分享优秀的 AI 项目。如果本文对你有用就点个 在看 鼓励一下吧。

最后： 可以在公众号：伤心的辣条 ！ 自行领取一份216页软件测试工程师面试宝典文档资料【免费的】。以及相对应的视频学习教程免费分享！，其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。

我推荐一个【Python自动化测试交流群：**746506216**】，大家可以一起探讨交流软件测试，共同学习软件测试技术、面试等软件测试方方面面，助你快速进阶Python自动化测试/测试开发，走向高薪之路。

**喜欢软件测试的小伙伴们，如果我的博客对你有帮助、如果你喜欢我的博客内容，请 “点赞” “评论” “收藏” 一 键三连哦！**

[5556bcbecdef41a2b4dfcec45c318c9c.jpeg]: https://img-blog.csdnimg.cn/5556bcbecdef41a2b4dfcec45c318c9c.jpeg
[3cd5744c24a94eb0b96ff498048b8e64.png]: https://img-blog.csdnimg.cn/3cd5744c24a94eb0b96ff498048b8e64.png
[458703eb447d406c8113a49532b08f0b.jpeg]: https://img-blog.csdnimg.cn/458703eb447d406c8113a49532b08f0b.jpeg
[1a59fb82970348e0af1c425f1217fa95.png]: https://img-blog.csdnimg.cn/1a59fb82970348e0af1c425f1217fa95.png