让美颜app瞬间弱爆的AI算法 附教程

简介: StyleFlow 算法介绍运行算法1、linux环境2、Windows10环境

先来猜猜看。


下面这张照片中的人,是真实存在的,还是算法生成的?


image.png


答案是通过StyleGAN算法生成的。


这个人,在世界上并不存在。


没错——


头发、皱纹、光线、表情等等,都是AI算法自己生成的。


完全看不出跟真人有什么区别。


这就是人脸属性编辑算法的强大之处。


大家可以登录下面这个网站直接体验。


https://thispersondoesnotexist.com/


每刷新一次,都会随机生成这么一张照片。


抖音上很火的性别转换特效,还有变老人、变小孩特效。


都是这种人脸属性编辑算法实现的。


StyleFlow 算法介绍


今天我们来讲讲 StyleFlow 算法,是基于 StyleGAN 算法的。


也是一个非常牛逼的人脸属性编辑算法。


不仅可以瘦脸、美白。


还可以编辑自己的表情、发型、角度、性别等。


先来看看马斯克的各种变脸:


image.png


效果非常强大!


算法原理是在隐空间(latent space)控制隐藏特征(latent code)来控制图片的属性。


用人话说就是,将图片的各种属性进行分解,分别编辑,避免一个属性变化影响另外一个属性。


比如我们编辑一张人脸,表情、头发、性别、肤色、年龄、光线等都是照片的属性。


算法将这些属性进行解耦分类,分别控制编辑,得到我们想要的效果。


具体可以查看论文:


https://arxiv.org/abs/2008.02401


运行算法


我们来分别看看linux环境和windows环境,怎样来运行算法。


1、linux环境


第一步:下载代码


项目地址:


https://github.com/RameenAbdal/StyleFlow


第二步:搭建环境


需要配置PyTorch、TensorFlow等。配置信息都在根目录下面的 environment.yml 文件中,我们根据这个文件安装即可。


conda env create -f environment.yml


安装过程中如果碰到插件错误,可以单独安装插件。比如:


# 单独安装torch插件
pip install torch==1.1.0 -f https://download.pytorch.org/whl/torch_stable.html
# 单独安装tensorflow-gpu插件
conda install tensorflow-gpu==1.15.0


还要搭建StyleGAN2 的环境,跟上面一样的搭建思路。


项目地址:


https://github.com/NVlabs/stylegan2


第三步:下载训练好的模型权重文件。


实现上方动图的效果。


我们使用的权重文件是stylegan2-ffhq-config-f.pkl,可以实现人脸属性编辑。


文件要登陆谷歌下载,知道大家fq不易,贴心的我给大家放到了百度云盘里面,大家可以直接下载。


链接(提取码:6666):


https://pan.baidu.com/s/10bwCYPLzN7fbu3Dd0Eh3MA


如果想要尝试其他效果,比如编辑汽车属性,可以到下面网址下载对应的权重文件


权重文件:


https://drive.google.com/drive/folders/1QHc-yF5C3DChRwSdZKcx1w6K8JvSxQi7


第四步:运行程序。


python main.py


就会打开下方界面。


image.png


2、Windows10环境


如果你是Windows10环境,跟着上面部署,大概率会非常不顺利,别问我是怎么知道的。


那个凌晨4点的晚上——


来看看Windows10环境怎样部署这个算法。


第一步:下载代码


项目地址:


https://github.com/justinjohn0306/StyleFlow-Windows-10


第二步:搭建环境


conda env create -f env_windows.yml
conda activate styleflow
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch


继续安装streamlit插件,用来可视化界面


pip install streamlit


第三步:运行项目


cd webui
streamlit run app.py


现在就可以编辑自己想要编辑的图片效果了。


image.png

目录
相关文章
|
2月前
|
机器学习/深度学习 人工智能 监控
AI算法分析,智慧城管AI智能识别系统源码
AI视频分析技术应用于智慧城管系统,通过监控摄像头实时识别违法行为,如违规摆摊、垃圾、违章停车等,实现非现场执法和预警。算法平台检测街面秩序(出店、游商、机动车、占道)和市容环境(垃圾、晾晒、垃圾桶、路面不洁、漂浮物、乱堆物料),助力及时处理问题,提升城市管理效率。
AI算法分析,智慧城管AI智能识别系统源码
|
2月前
|
监控 安全 数据可视化
【教程】为什么要为 App 应用加固 ?如何为 App 应用加固 ?
【教程】为什么要为 App 应用加固 ?如何为 App 应用加固 ?
|
2月前
|
开发者 iOS开发
UniApp打包教程:使用HBuilder X和AppUploader完成原生App云打包和上架指南“
UniApp打包教程:使用HBuilder X和AppUploader完成原生App云打包和上架指南“
55 3
|
7天前
|
搜索推荐 前端开发 JavaScript
【专栏】从网页到应用:简易教程教你如何在线生成 App
【4月更文挑战第27天】本文介绍了将网页转换为移动应用的在线生成App过程,包括原理(封装网页内容,添加移动特性)、优势(便捷高效,节省成本,快速上线)以及具体步骤(选择工具,准备内容,配置选项,生成测试,发布推广)。同时,注意版权、兼容性和用户体验问题。通过案例分析和实践经验分享,帮助读者深入了解如何通过在线工具实现App梦想,强调了网页优化、用户界面设计和个性化定制的重要性。
|
2月前
|
机器学习/深度学习 人工智能 运维
【人工智能技术专题】「入门到精通系列教程」打好AI基础带你进军人工智能领域的全流程技术体系(机器学习知识导论)(二)
【人工智能技术专题】「入门到精通系列教程」打好AI基础带你进军人工智能领域的全流程技术体系(机器学习知识导论)
60 1
|
2月前
|
机器学习/深度学习 人工智能 自然语言处理
【人工智能技术专题】「入门到精通系列教程」打好AI基础带你进军人工智能领域的全流程技术体系(机器学习知识导论)(一)
【人工智能技术专题】「入门到精通系列教程」打好AI基础带你进军人工智能领域的全流程技术体系(机器学习知识导论)
78 1
|
2天前
|
机器学习/深度学习 人工智能 算法
【AI 初识】描述遗传算法概念
【5月更文挑战第2天】【AI 初识】描述遗传算法概念
|
2天前
|
机器学习/深度学习 存储 人工智能
【AI 初识】人工智能中使用了哪些不同的搜索算法?
【5月更文挑战第2天】【AI 初识】人工智能中使用了哪些不同的搜索算法?
|
7天前
|
机器学习/深度学习 数据采集 人工智能
【热门话题】AI作画算法原理解析
本文解析了AI作画算法的原理,介绍了基于机器学习和深度学习的CNNs及GANs在艺术创作中的应用。从数据预处理到模型训练、优化,再到风格迁移、图像合成等实际应用,阐述了AI如何生成艺术作品。同时,文章指出未来发展中面临的版权、伦理等问题,强调理解这些算法对于探索艺术新境地的重要性。
20 3
|
8天前
|
机器学习/深度学习 人工智能 算法
详解AI作画算法原理
AI作画算法运用深度学习和生成对抗网络(GAN),通过学习大量艺术作品,模拟艺术家风格。卷积神经网络(CNN)提取图像特征,GAN中的生成器和判别器通过对抗训练生成艺术图像。循环神经网络和注意力机制可提升作品质量。这种技术开创了艺术创作新途径。