让美颜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月前
|
人工智能 算法 搜索推荐
电商API的“AI革命”:全球万亿市场如何被算法重新定义?
AI+电商API正引领智能商业变革,通过智能推荐、动态定价与自动化运营三大核心场景,大幅提升转化率、利润率与用户体验。2025年,75%电商API将具备个性化能力,90%业务实现智能决策,AI与API的深度融合将成为未来电商竞争的关键基石。
|
3月前
|
人工智能 小程序 JavaScript
【一步步开发AI运动APP】十、微调优化内置运动分析器,灵活适配不同的应用场景
本文介绍了如何通过【一步步开发AI运动APP】系列博文,利用`ISportOptimizer`对内置运动分析器进行微调优化。相比小程序,APP框架(如uni-app)因技术差异,无法直接修改运动分析器参数,因此提供了统一的优化机制。开发者可通过`ISportOptimizer`获取和更新检测规则、动作样本等,灵活适应不同场景需求,如全民运动赛事的宽松模式或学生体测的严格模式。文中还提供了示例代码,展示如何对具体运动项目(如仰卧起坐)进行优化。需要注意的是,微调优化适用于标准动作的小范围调整,若动作变化过大(如花式跳绳),可期待后续自定义扩展功能。
|
23天前
|
人工智能 自然语言处理 供应链
从体验到系统工程丨上手评测国内首款 AI 电商 App
近期,1688 推出了 1688 AI App,这貌似是国内第一个电商领域的独立 AI App 应用(若不是,欢迎评论指正)。本文试图通过产品界面这一入口,窥探其背后的系统工程。
120 19
|
10天前
|
机器学习/深度学习 人工智能 算法
当AI提示词遇见精密算法:TimeGuessr如何用数学魔法打造文化游戏新体验
TimeGuessr融合AI与历史文化,首创时间与空间双维度评分体系,结合分段惩罚、Haversine距离计算与加权算法,辅以连击、速度与完美奖励机制,实现公平且富挑战性的游戏体验。
|
26天前
|
人工智能 小程序 开发者
【一步步开发AI运动APP】十一、同时检测识别多人运动,分别进行运动计时计数
本文介绍了如何开发支持多人运动检测的AI运动APP,涵盖多人人体检测、定位及运动分析实现方法,助力开发者打造高性能AI运动应用。
|
2月前
|
机器学习/深度学习 人工智能 算法
AI-Compass RLHF人类反馈强化学习技术栈:集成TRL、OpenRLHF、veRL等框架,涵盖PPO、DPO算法实现大模型人类价值对齐
AI-Compass RLHF人类反馈强化学习技术栈:集成TRL、OpenRLHF、veRL等框架,涵盖PPO、DPO算法实现大模型人类价值对齐
 AI-Compass RLHF人类反馈强化学习技术栈:集成TRL、OpenRLHF、veRL等框架,涵盖PPO、DPO算法实现大模型人类价值对齐
|
2月前
|
机器学习/深度学习 人工智能 算法
AI-Compass 强化学习模块:理论到实战完整RL技术生态,涵盖10+主流框架、多智能体算法、游戏AI与金融量化应用
AI-Compass 强化学习模块:理论到实战完整RL技术生态,涵盖10+主流框架、多智能体算法、游戏AI与金融量化应用
|
1月前
|
人工智能 算法 计算机视觉
只需完成手画线稿,让AI算法帮你自动上色
本文介绍了如何利用图像处理技术生成手绘风格图像及自动上色的方法。内容涵盖图像灰度化、梯度调整、虚拟深度实现手绘效果,以及使用 Python 编程实现相关算法。此外,还介绍了 AI 工具 Style2Paints V4.5,其可为线稿自动上色并支持多种线稿类型,如插画和手绘铅笔稿,适用于艺术创作与图像处理领域。
|
3月前
|
人工智能 UED
AI教育热潮,如何衡量教育App渠道投放效果?
AI教育是高潜力场景,但如何衡量AI热潮下教育App的真实拉新与转化效果呢?
99 6