HivisionIDPhoto 旨在开发一种实用、系统性的证件照智能制作算法。GitHub地址
它利用一套完善的AI模型工作流程，实现对多种用户拍照场景的识别、抠图与证件照生成。
HivisionIDPhoto 可以做到：
- 轻量级抠图（纯离线，仅需 CPU 即可快速推理）
- 根据不同尺寸规格生成不同的标准证件照、六寸排版照
- 支持 纯离线 或 端云 推理
- 美颜
- 智能换正装（waiting）
Docker 部署
以下方式三选一
方式一：拉取最新镜像：
docker pull linzeyi/hivision_idphotos
方式二：Dockrfile 直接构建镜像：
在确保将至少一个抠图模型权重文件放到hivision/creator/weights下后，在项目根目录执行：
docker build -t linzeyi/hivision_idphotos .
方式三：Docker compose 构建：
在确保将至少一个抠图模型权重文件放到hivision/creator/weights下后，在项目根目录下执行：
docker compose build
2. 运行服务
启动 Gradio Demo 服务
运行下面的命令，在你的本地访问 http://127.0.0.1:7860 即可使用。
docker run -d -p 7860:7860 linzeyi/hivision_idphotos
启动 API 后端服务
docker run -d -p 8080:8080 linzeyi/hivision_idphotos python3 deploy_api.py
两个服务同时启动
docker compose up -d
环境变量
本项目提供了一些额外的配置项，使用环境变量进行设置：
| 环境变量 | 类型 | 描述 | 示例 |
|---|---|---|---|
| FACE_PLUS_API_KEY | 可选 | 这是你在 Face++ 控制台申请的 API 密钥 | 7-fZStDJ···· |
| FACE_PLUS_API_SECRET | 可选 | Face++ API密钥对应的Secret | VTee824E···· |
| RUN_MODE | 可选 | 运行模式，可选值为beast(野兽模式)。野兽模式下人脸检测和抠图模型将不释放内存，从而获得更快的二次推理速度。建议内存16GB以上尝试。 | beast |
| DEFAULT_LANG | 可选 | Gradio Demo启动时的默认语言 | en |
docker使用环境变量示例：
docker run -d -p 7860:7860 \
-e FACE_PLUS_API_KEY=7-fZStDJ···· \
-e FACE_PLUS_API_SECRET=VTee824E···· \
-e RUN_MODE=beast \
-e DEFAULT_LANG=en \
linzeyi/hivision_idphotos
