这几年AIGC特别火,我跟风做了一个头像制作软件“托尼头像”,哪知昨天宫崎骏看了我这个工具制作的hayao风格的滤镜,由于效果太震撼,宫崎骏直言:我强烈感觉到对生命的侮辱。

下面我简单介绍一下这个软件的制作过程,这个软件的前端是uniapp,不过因为后端是restful api, 这些图片处理功能可以集成到任意客户端中,包括将api直接开放给开发者使用。重点说一下后端的图片处理功能,托尼头像的功能主要分为三大块:

1. 热门头像的增删改查,这个用的是一个ES的集群,然后使用SCF云函数对头像元数据进行检索,图片文件内容放在COS中,并使用CDN进行缓存。这一块搜索的效果当然比不上语义检索,目前我也在基于LLM做一些语义搜索方面的尝试。

2. CPU推理。这里的模型我是使用的开源的AnimeGAN,这个模型各种推理框架都支持了,比如Pytorch,tensorflow, 以及百度的paddlepaddle等,AnimeGAN是唯一CPU可以跑起来,效果还凑合的人像卡通画模型。

https://github.com/TachibanaYoshino/AnimeGANv2

这个效果在几年前还是可以的,不过现在已经被AI大模型甩得尾灯都看不到了。

3. GPU推理。我在自己的游戏主机上部署了一个大模型,我们先看一下卡通化的效果,原图是这样(图片来源于网络或者AI生成,侵权联系我删除)

卡通化之后是这样。

是不是比AnimeGAN要强很多?可是代价就是必须有显卡才能跑起来,我目前用的游戏主机有一张刚好能跑一张图片的卡,我把它作为后端推理服务器使用了。

Categories:

Tags:

Comments are closed

湘ICP备2024066431号-1