用Python生成会跳舞的美女


Posted in Python onJanuary 18, 2022

导语

表妹心疼我,为了逗我开心,教我用Python制作会跳舞的美女。作为新时代的活雷锋,在这里分享给大家。

开发工具

Python版本:3.6.4

相关模块:

torch0.4.1模块;
torchvision0.2.2模块;
opencv-python模块;
argparse模块;
numpy模块;
以及一些Python自带的模块。

环境搭建

安装Python并添加到环境变量,pip安装需要的相关模块即可。

(PS:CUDA环境配置请自行Google教程。)

原理简介

这里我们利用VAE的原理来生成会跳舞的小姐姐呗。先简单介绍一下理论知识,要说VAE,就得先谈谈自动编码器,自动编码器最开始用于数据压缩,其原理图十分简单直观:

用Python生成会跳舞的美女

在模型训练时,输入图片首先通过一个编码器(现在当然是神经网络啦)将输入数据降维到一个code,接着将code输入到解码器(当然还是神经网络啦)生成一个与输入相同大小的图片,网络的优化目标就是使得输入编码器的图片与解码器输出的图片尽可能相似。 当模型训练完之后,我们就可以拿出这个解码器,随机传入一些code作为它的输入,来生成一些差不多的图片。但是这种做法存在一个问题,就是编码器生成的code包含了原图信息,因此,我们就无法自己构造一些随机的code来作为解码器输入从而生成任意图片(必须是现有图片过了编码器之后生成的code作为输入)。为了解决这个问题,VAE就诞生啦~VAE的全称是:

Variational Autoencoder 即变分自编码器
说起来,它的原理图也十分简单:

用Python生成会跳舞的美女

与自编码器唯一不同就是在训练过程中对code也进行了约束,使得code满足某种分布(譬如正态分布)。这样的话,模型训练完之后,我们只需要随机生成一个满足该分布的code,就可以拿来作为解码器的输入,从而生成任意图片啦~ 一般地,我们用均方损失函数保证原图和生成图相似,用KL损失函数来保证code满足我们所要求的分布。均方损失大家肯定知道,KL散度的定义这里我也不想过多地去介绍,贴下维基百科上定义的公式吧:

用Python生成会跳舞的美女

P和Q是在同一个概率空间上定义的离散概率分布。 同时,为了方便计算KL散度,VAE引入了一个trick,叫:

reparameterization

以正态分布为例子,就是现在不直接预测code了,而是预测code的均值和方差,有了均值向量和方差向量,恢复code是很方便的(用标准正态分布乘以方差再加个均值就行了)。

再回过头来,我们现在要用VAE来生成会跳舞的小姐姐,无非就是先找到一个训练数据集,这里我找了个Youtube上的视频作为数据集:

https://www.youtube.com/watch?v=NdSqAAT28v0

然后根据VAE的原理构造一个编码和解码网络,用两个损失函数作为约束训练网络就行啦。代码就不一个个贴了,写起来很简单的,贴上来倒感觉有点麻烦,有需要的直接在公众号后台回复“dancenet”即可获取。简单说下代码使用方式吧,模型训练运行train.py文件即可,命令格式为:

python train.py --videopath xxx.mp4
模型测试运行demo.py文件即可,命令格式为:
python demo.py --mode random/fromtrain --checkpointspath xxx.pth --outputpath xxx.avi
这里模式的含义是code随机生成/从训练集中提取一些图片的code作为输入。

效果展示

随机生成code模式作为输入:

用Python生成会跳舞的美女

数据集中挑选图片过编码器获得code作为输入:

用Python生成会跳舞的美女


好了,我的分享到此结束了。

总结

到此这篇关于用Python生成会跳舞的美女的文章就介绍到这了,更多相关Python美女内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python编写检测数据库SA用户的方法
Jul 11 Python
Django中的Signal代码详解
Feb 05 Python
python将文本中的空格替换为换行的方法
Mar 19 Python
python实现决策树分类
Aug 30 Python
Python实现统计英文文章词频的方法分析
Jan 28 Python
python使用配置文件过程详解
Dec 28 Python
Python使用进程Process模块管理资源
Mar 05 Python
parser.add_argument中的action使用
Apr 20 Python
Django-migrate报错问题解决方案
Apr 21 Python
使用pyecharts1.7进行简单的可视化大全
May 17 Python
Django多个app urls配置代码实例
Nov 26 Python
Python实现视频自动打码的示例代码
Apr 08 Python
基于Pygame实现简单的贪吃蛇游戏
Dec 06 #Python
Python可变集合和不可变集合的构造方法大全
Dec 06 #Python
Python实现视频中添加音频工具详解
Dec 06 #Python
Python实现GIF动图以及视频卡通化详解
Python实现照片卡通化
用Python爬取英雄联盟的皮肤详细示例
Python+腾讯云服务器实现每日自动健康打卡
Dec 06 #Python
You might like
输出控制类
2006/10/09 PHP
让你同时上传 1000 个文件 (二)
2006/10/09 PHP
两个强悍的php 图像处理类1
2009/06/15 PHP
php ob_flush,flush在ie中缓冲无效的解决方法
2010/05/09 PHP
详解PHP防止盗链防止迅雷下载的方法
2017/04/26 PHP
jscript之Read an Excel Spreadsheet
2007/06/13 Javascript
一段效率很高的for循环语句使用方法
2007/08/13 Javascript
jQuery中的val()示例应用
2014/02/26 Javascript
学习JavaScript事件流和事件处理程序
2016/01/25 Javascript
node.js插件nodeclipse安装图文教程
2020/10/19 Javascript
JavaScript操作 url 中 search 部分方法函数
2016/06/15 Javascript
url中的特殊符号有什么含义(推荐)
2016/06/17 Javascript
Node.js连接postgreSQL并进行数据操作
2016/12/18 Javascript
Bootstrap风格的zTree右键菜单
2017/02/17 Javascript
JS实现定时任务每隔N秒请求后台setInterval定时和ajax请求问题
2017/10/15 Javascript
vue + element-ui实现简洁的导入导出功能
2017/12/22 Javascript
基于vue-resource jsonp跨域问题的解决方法
2018/02/03 Javascript
JS简单实现动态添加HTML标记的方法示例
2018/04/08 Javascript
基于nodejs的微信JS-SDK简单应用实现
2019/05/21 NodeJs
element-ui table行点击获取行索引(index)并利用索引更换行顺序
2020/02/27 Javascript
Vue数据双向绑定原理实例解析
2020/05/15 Javascript
[04:11]DOTA2亚洲邀请赛小组赛第一日 TOP10精彩集锦
2015/01/30 DOTA
[01:36]DOTA2完美大师赛趣味视频之与队友相处的十万个技巧
2017/11/19 DOTA
TensorFlow实现Batch Normalization
2018/03/08 Python
python实现csv格式文件转为asc格式文件的方法
2018/03/23 Python
python bmp转换为jpg 并删除原图的方法
2018/10/25 Python
python中的subprocess.Popen()使用详解
2019/12/25 Python
15行Python代码实现免费发送手机短信推送消息功能
2020/02/27 Python
浅析Python面向对象编程
2020/07/10 Python
video结合canvas实现视频在线截图功能
2018/06/25 HTML / CSS
html5 css3实例教程 一款html5和css3实现的小机器人走路动画
2014/10/20 HTML / CSS
北美主要的汽车零部件零售商:AutoShack.com
2019/02/23 全球购物
将时尚融入珠宝:Adornmonde
2019/10/17 全球购物
娇韵诗俄罗斯官方网站:Clarins俄罗斯
2020/10/03 全球购物
初中生物教学随笔
2015/08/15 职场文书
python文件名批量重命名脚本实例代码
2021/04/22 Python