Docker安装轻量级AI算法证件照工具HivisionIDPhotos
项目优点
HivisionIDPhotos是一款开源免费的图片处理工具,可以利用AI模型对照片进行轻量级智能抠图、通过调整尺寸生成不同的标准证件照、替换背景、美颜、智能换正装(后期支持)等操作。
通过它,自己在家也能轻松搞定证件照,而且在本地部署,无需再去网上付费制作证件照,且不需要担心个人信息泄露。
官方在线体验地址
项目介绍
HivisionIDPhoto 旨在开发一种实用、系统性的证件照智能制作算法。
它利用一套完善的AI模型工作流程,实现对多种用户拍照场景的识别、抠图与证件照生成。
HivisionIDPhoto 可以做到:
轻量级抠图(纯离线,仅需 CPU 即可快速推理)
根据不同尺寸规格生成不同的标准证件照、六寸排版照
支持 纯离线 或 端云 推理
美颜
智能换正装(waiting)
🤩 最近更新(20241010记录)
2024.09.25: 增加五寸相纸和JPEG下载选项|默认照片下载支持300DPI
2024.09.24: API接口增加base64图像传入选项 | Gradio Demo增加排版照裁剪线功能
2024.09.22: Gradio Demo增加野兽模式,可设置内存加载策略 | API接口增加dpi、face_alignment参数
2024.09.18: Gradio Demo增加分享模版照功能、增加美式证件照背景选项
2024.09.17: Gradio Demo增加自定义底色-HEX输入功能 | (社区贡献)C++版本 - HivisionIDPhotos-cpp 贡献 by zjkhahah
2024.09.16: Gradio Demo增加人脸旋转对齐功能,自定义尺寸输入支持毫米单位
2024.09.14: Gradio Demo增加自定义DPI功能,增加日语和韩语支持,增加调整亮度、对比度、锐度功能
2024.09.12: Gradio Demo增加美白功能 | API接口增加加水印、设置照片KB值大小、证件照裁切
安装教程
Docker Compose方式安装
演示以飞牛NAS系统为例(其他系统请自行安装好最新版本Docker、docker-compose)
打开系统的SSH功能,使用终端软件进行连接,并切换到root状态下。
在NAS系统的文件管理中新建一个文件夹,右键选择文件详细信息(属性),点击
复制原始路径
。
(这一步主要是为了获取系统文件夹路径,照顾小白,如果你熟悉的话可以不创建,直接在终端软件内进入这个根目录路径即可。)在终端内执行下面的命令,进入文件夹内。
# 格式:cd 复制的文件夹路径(换成你自己的,别照抄路径。) cd /vol1/1000/Photo
克隆项目到本地
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
查看克隆到本地的文件
ll
进入项目文件夹
cd HivisionIDPhotos
如果你非常熟悉Docker工具,可在查看完项目说明后自行修改docker-compose.yml模板文件。(小白略过这一句)
编辑docker-compose.yml文件(可选)
vi docker-compose.yml
按字母
i
键进入编辑模式,修改下面的配置文件,按Esc
键退出编辑模式,输入:wq
退出并保存。(可选)version: '3.8' services: hivision_idphotos: build: context: . dockerfile: Dockerfile image: linzeyi/hivision_idphotos command: python3 -u app.py --host 0.0.0.0 --port 7860 ports: - '7860:7860' # 第一个7860是容器外部访问端口,可自行修改。 hivision_idphotos_api: build: context: . dockerfile: Dockerfile image: linzeyi/hivision_idphotos command: python3 deploy_api.py ports: - '8080:8080'
执行命令,启动docker-compose文件,拉取镜像并创建容器。
docker compose up -d 或者 docker-compose up -d
查看正在运行的项目容器
docker compose ps 或者 docker-compose ps
访问HivisionIDPhotos
打开浏览器,以NAS的IP+设置的端口进行访问。
以本机为例:http://192.168.2.225:7860使用方式也很简单,直接将你拍好的自拍照放进去。
根据自己的需要选择好对应的参数,然后点击开始制作。
在网页底部还有预设的图片,可以自己先点击使用,看看效果。
(人脸检测模型:face+(联网Online API)这个选项需参考官方文档单独配置才可使用。)然后等待系统自动生成即可,点击图片旁边的图标可全屏查看也可保存到本地。
下面还有社交媒体模板照和抠图照可以用。
整体用下来效果是不错的,不过图片中过于曝光的白色区域会被系统自动抠除,会显示缺一块。
同一个照片文件中只允许出现一个人脸,出现两个就会报错,提示你重新上传图片。
选择不同的模型会消耗不同的性能,实测教程中的图片直接默认配置进行制作,消耗50%CPU3秒(i5-4570),内存消耗约1.5GB。
如果换了其他的模型/配置,CPU约占50%,内存是直接拉满了,不过在制作完成之后就会恢复正常占用。
项目地址
GitHub项目地址:https://github.com/Zeyi-Lin/HivisionIDPhotos
👇👇👇
- 感谢你赐予我前进的力量