让美颜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月前
|
存储 人工智能 前端开发
Qoder + ADB Supabase :5分钟GET超火AI手办生图APP
本文介绍如何利用Qoder、阿里云ADB Supabase和通义千问图像编辑模型,快速搭建AI手办生图Flutter应用。无需传统后端,实现从前端生成到数据存储、AI服务集成的全链路敏捷开发,展现Vibe Coding的高效实践。
Qoder + ADB Supabase :5分钟GET超火AI手办生图APP
|
5月前
|
人工智能 算法 搜索推荐
电商API的“AI革命”:全球万亿市场如何被算法重新定义?
AI+电商API正引领智能商业变革,通过智能推荐、动态定价与自动化运营三大核心场景,大幅提升转化率、利润率与用户体验。2025年,75%电商API将具备个性化能力,90%业务实现智能决策,AI与API的深度融合将成为未来电商竞争的关键基石。
|
2月前
|
人工智能 小程序 开发者
【一步步开发AI运动APP】十二、自定义扩展新运动项目03
继【一步步开发AI运动小程序】后,我们推出新系列【一步步开发AI运动APP】,助开发者打造高性能、优体验的AI运动应用。本文详解自定义扩展运动分析器的统一管理实现,提升代码复用性与可维护性,涵盖APP与小程序插件差异及完整代码示例,助力AI运动场景深度拓展。
|
3月前
|
人工智能 自然语言处理 供应链
从体验到系统工程丨上手评测国内首款 AI 电商 App
近期,1688 推出了 1688 AI App,这貌似是国内第一个电商领域的独立 AI App 应用(若不是,欢迎评论指正)。本文试图通过产品界面这一入口,窥探其背后的系统工程。
463 32
|
3月前
|
机器学习/深度学习 人工智能 算法
当AI提示词遇见精密算法:TimeGuessr如何用数学魔法打造文化游戏新体验
TimeGuessr融合AI与历史文化,首创时间与空间双维度评分体系,结合分段惩罚、Haversine距离计算与加权算法,辅以连击、速度与完美奖励机制,实现公平且富挑战性的游戏体验。
|
2月前
|
人工智能 小程序 搜索推荐
【一步步开发AI运动APP】十二、自定义扩展新运动项目2
本文介绍如何基于uni-app运动识别插件实现“双手并举”自定义扩展运动,涵盖动作拆解、姿态检测规则构建及运动分析器代码实现,助力开发者打造个性化AI运动APP。
|
3月前
|
人工智能 小程序 开发者
【一步步开发AI运动APP】十一、同时检测识别多人运动,分别进行运动计时计数
本文介绍了如何开发支持多人运动检测的AI运动APP,涵盖多人人体检测、定位及运动分析实现方法,助力开发者打造高性能AI运动应用。
|
5月前
|
机器学习/深度学习 人工智能 算法
AI-Compass RLHF人类反馈强化学习技术栈:集成TRL、OpenRLHF、veRL等框架,涵盖PPO、DPO算法实现大模型人类价值对齐
AI-Compass RLHF人类反馈强化学习技术栈:集成TRL、OpenRLHF、veRL等框架,涵盖PPO、DPO算法实现大模型人类价值对齐
 AI-Compass RLHF人类反馈强化学习技术栈:集成TRL、OpenRLHF、veRL等框架,涵盖PPO、DPO算法实现大模型人类价值对齐

热门文章

最新文章