用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输出指定月份日历的方法
Apr 23 Python
Python实现给文件添加内容及得到文件信息的方法
May 28 Python
解决PySide+Python子线程更新UI线程的问题
Jan 11 Python
python实现坦克大战游戏 附详细注释
Mar 27 Python
python安装scipy的方法步骤
Jun 26 Python
python django下载大的csv文件实现方法分析
Jul 19 Python
Python对接六大主流数据库(只需三步)
Jul 31 Python
关于Python字符串显示u...的解决方式
Mar 06 Python
tensorflow pb to tflite 精度下降详解
May 25 Python
浅析python 字典嵌套
Sep 29 Python
基于python的opencv图像处理实现对斑马线的检测示例
Nov 29 Python
Python pandas求方差和标准差的方法实例
Aug 04 Python
基于Pygame实现简单的贪吃蛇游戏
Dec 06 #Python
Python可变集合和不可变集合的构造方法大全
Dec 06 #Python
Python实现视频中添加音频工具详解
Dec 06 #Python
Python实现GIF动图以及视频卡通化详解
Python实现照片卡通化
用Python爬取英雄联盟的皮肤详细示例
Python+腾讯云服务器实现每日自动健康打卡
Dec 06 #Python
You might like
php截取字符串函数substr,iconv_substr,mb_substr示例以及优劣分析
2014/06/10 PHP
Yii框架连接mongodb数据库的代码
2016/07/27 PHP
详解php中的implements 使用
2017/06/13 PHP
解析jQuery与其它js(Prototype)库兼容共存
2013/07/04 Javascript
jquery 清空file域示例(兼容个浏览器)
2013/10/11 Javascript
JQuery跳出each循环的方法
2015/04/16 Javascript
JavaScript精炼之构造函数 Constructor及Constructor属性详解
2015/11/05 Javascript
js针对ip地址、子网掩码、网关的逻辑性判断
2016/01/06 Javascript
JavaScript兼容性总结之获取非行间样式案例
2016/08/07 Javascript
使用jsonp实现跨域获取数据实例讲解
2016/12/25 Javascript
Vue组件化通讯的实例代码
2017/06/23 Javascript
基于Bootstrap分页的实例讲解(必看篇)
2017/07/04 Javascript
浅析Angular19 自定义表单控件
2018/01/31 Javascript
element-ui多文件上传的实现示例
2019/04/10 Javascript
小程序按钮避免多次调用接口和点击方案实现(不用showLoading)
2020/04/15 Javascript
Javascript文本框脚本实现方法解析
2020/10/30 Javascript
在漏洞利用Python代码真的很爽
2007/08/26 Python
Python字符串格式化
2015/06/15 Python
解决Python3中的中文字符编码的问题
2018/07/18 Python
删除DataFrame中值全为NaN或者包含有NaN的列或行方法
2018/11/06 Python
Python列表对象实现原理详解
2019/07/01 Python
ML神器:sklearn的快速使用及入门
2019/07/11 Python
使用wxpy实现自动发送微信消息功能
2020/02/28 Python
使用CSS3制作饼状旋转载入效果的实例
2015/06/23 HTML / CSS
设计师珠宝:Ylang 23
2018/05/11 全球购物
美国正版电视节目和电影在线观看:Hulu
2018/05/24 全球购物
定义一结构体数组表示分数,并求两个分数相加之和
2013/06/11 面试题
了解AppleTalk协议吗
2014/04/01 面试题
大学同学聚会邀请函
2014/01/19 职场文书
授权委托书范本
2014/04/03 职场文书
大队干部竞选演讲稿
2014/04/28 职场文书
安全演讲稿大全
2014/05/09 职场文书
2014应届本科生自我评价
2014/09/13 职场文书
学籍证明模板
2014/11/21 职场文书
2015年幼儿园班主任工作总结
2015/05/12 职场文书
Django一小时写出账号密码管理系统
2021/04/29 Python