七夕啦,小情侣们又又又又又要正大光明“撒狗粮”了!在这个特别的日子里,还是“单身喵”的你我他,是不是更对未来的 TA 充满了期待呢?来,AI 来帮你找到心仪的另一半!
不信?你看!只需要文字描绘出,你心目中未来的 TA 的样子,就能立马生成一张你心仪对象的照片,是不是很神奇?
一个黑色头发微笑着的美丽女孩
一个紫色头发大笑着的漂亮女孩
一个有着粗眉毛的亚洲少年
一个大鼻子的阳光男孩
是不是很神奇?不仅如此,你还可以为心仪的 TA,一键生成卡爱的卡通头像!
爱了爱了,如此好玩又有趣,那背后的 AI 技术又是怎样的呢?
以上两种应用都是由 MMGeneration 算法库支持实现的,目前支持图像生成、图像转换、图像编辑等任务,支持了 16 种算法,提供了 80+ 预训练模型。
其中,通过文字生成对象是通过一种叫 StyleClip 的算法实现的,通过计算生成图像和描述文本的 clip loss 来优化生成结果,经过迭代优化,最终得到符合描述的图片。
而给心仪对象生成可爱卡通头像则是通过 MMGEN-FaceStylor 实现的,MMGEN-FaceStylor 是基于 MMGeneration 和 agilegan 算法开发的人脸风格化应用。
看到这里,是不是已经迫不及待啦,想要立刻动手试试?so easy,我们只需按以下步骤操作,就能快速实现效果~
第一步,利用 MMGeneration 下 apps/styleclip.py 脚本通过文本生成图像。
参考 https://github.com/open-mmlab/mmgeneration#installation 安装 MMGeneration。
在 `--description` 后,你可以输入自己的描述,step 设置迭代次数,lr 设置学习率。
- cd mmgeneration
- python apps/styleclip.py\
- configs/styleganv2/stylegan2_c2_ffhq_256_b4x8_800k.py \
- https://download.openmmlab.com/mmgen/stylegan2/stylegan2_c2_ffhq_256_b4x8_20210407_160709-7890ae1f.pth\
- --description "a beautiful girl with purple hair" \
- --results-dir ~/outputs/girl_friends/ \
- --step 120 \
- --lr 0.1\
运行完这一步后,可以在 ~/outputs/girl_friends/ 下找到生成的心仪的 TA 的图片。
第二步,利用 MMGen-FaceStylor 生成头像。
参考 https://github.com/open-mmlab/MMGEN-FaceStylor 安装 FaceStylor,将 style 设成你想要的风格,然后运行如下命令,在 work_dirs/demos/ 下找到生成的头像。
是不是简单又好玩?
MMGen-FaceStylor 的玩法可以参考
OpenMMLab:MMGEN-FaceStylor | 因为是你,所以每一种样子我都喜欢13 赞同 · 13 评论文章正在上传…重新上传取消
更多进阶玩法,欢迎访问算法库。
https://github.com/open-mmlab/mmgenerationgithub.com/open-mmlab/mmgeneration
https://github.com/open-mmlab/MMGEN-FaceStylorgithub.com/open-mmlab/MMGEN-FaceStylor
欢迎大家体验,觉得好用欢迎点亮小星星,也欢迎大家成为我们的贡献者,一起建设更好的算法库!
值此七夕,但愿朝夕。祝愿大家都能遇到一生所爱~