前言
最近AIGC圈没有什么大新闻,看到一个非常火的项目,AI证件照制作
项目地址: github.com/Zeyi-Lin/Hi…
现在项目功能在不断地积极迭代更新中。几天时间目前就有6.7K Star 了。

看着这个项目这么给力,自己马上把代码拉了下来安装了测试记录一下,认真看看到底是什么魔力这么火。
安装记录
使用docker-compose 安装非常简单,注意的是需要手动下载模型文件,github上只有代码的源码

使用git clone 代码拉取到本地过后进入项目文件夹使用wget 命令下载模型文件

这个项目代码看起来更新非常频繁,写这篇文章的时候发现代码结构改了一点。
wget https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/download/pretrained-model/hivision_modnet.onnx
然后使用docker-compose 打包运行
docker-compose build
docker-compose up -d
此时我这里出现意外了, Dockerfile 里面的 ubuntu:22.04 基础镜像拉取失败,提示没有权限,最终使用其他镜像加速源才正常,在这里分享记录一下我自己的配置:
"registry-mirrors": [
"https://lb5mpmbx.mirror.aliyuncs.com",
"https://mirror.gcr.io"
]
我这里安装了两次,还可以更换一下带ffmpeg的Python3.10的基础镜像也可以解决
FROM insoxin/python3.10-ffmpeg
...
测试
项目正常启动过后,访问127.0.0.1:7860,可以看到使用常见的gradio搭建的交互页面

可以直接上传带人脸的图片即可,尽量背景单一,这样效果非常不错

简单体验了一下,主要是有自动P图美化脸部瑕疵,自动裁剪,自定义背景颜色,自定义尺寸 这些核心功能,效果堪比商业照相馆人工处理了,而且非常快,几秒钟就可以处理完毕了。
最后
自己之前的公司也做过类似的产品,开发的抖音小程序,利用的是腾讯的产品接口,先识别人脸,进行风险检测,然后再进行人脸抠图接口,替换背景图,有一定的成本,所以需要先看几十秒的广告。如今开源项目想过都做得这么好了,非常的少见,而且提供接口,可以快速开发包装其他产品。