HivisionIDPhoto：一键生成高清证件照

发表于 更新于 分类于 Waline： 阅读次数： 本文字数： 2.3k 阅读时长 ≈ 2 分钟
HivisionIDPhoto 是一个轻量级的AI证件照制作算法。

什么是 HivisionIDPhoto ？

HivisionIDPhoto 是一个轻量级的 AI 证件照制作算法。利用一套完善的 AI 模型工作流程，实现对多种用户拍照场景的识别、抠图与证件照生成。可解决证件照应急制作问题！

软件功能：

  1. 轻量级抠图（纯离线，仅需 CPU 即可快速推理）
  2. 根据不同尺寸规格生成不同的标准证件照、六寸排版照
  3. 支持 纯离线 或 端云 推理
  4. 美颜
  5. 智能换正装（waiting）

如果不想自己安装，可以试试官方提供的在线 demo：

  • 地址一：https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo
  • 地址二：https://huggingface.co/spaces/TheEeeeLin/HivisionIDPhotos
  • 地址三：https://modelscope.cn/studios/SwanLab/HivisionIDPhotos

安装

在群晖上以 Docker 方式安装。

镜像有两种启动方式，一种是 API 后端服务方式，另一种是 Gradio Demo 服务方式。可以启动其中一种，也可以两种都启动，本文只介绍了 Gradio Demo 模式，会生成一个 Web 页面，可在页面中完成证件照的操作与交互；

在注册表中搜索 hivision_idphotos ，选择第一个 linzeyi/hivision_idphotos，版本选择 latest。

本文写作时， latest 版本对应为 v1.2.9；

端口

本地端口不冲突就行，不确定的话可以用命令查一下

1
2
# 查看端口占用
netstat -tunlp | grep 端口号
本地端口 容器端口 说明
7861 7860 Gradio Demo 服务端口
7861 7860 API 后端服务端口

默认暴露了 2 个端口

不用的端口建议点 - 号删了

环境

可变 值
DEFAULT_LANG Gradio Demo启动时的默认语言，默认为 en

更多环境变量请参考官方说明：https://github.com/Zeyi-Lin/HivisionIDPhotos#环境变量

命令行安装

如果你熟悉命令行，可能用 docker cli 更快捷

1
2
3
4
5
6
7
# 运行容器
docker run -d \
   --restart unless-stopped \
   --name idphotos \
   -p 7861:7860 \
   -e DEFAULT_LANG=zh \
   linzeyi/hivision_idphotos

也可以用 docker-compose 安装，将下面的内容保存为 docker-compose.yml 文件

下面的示例中，直接通过命令行，修改了容器端口；

1
2
3
4
5
6
7
8
9
10
11
12
version: '3'

services:
  hivision_idphotos:
    image: linzeyi/hivision_idphotos
    container_name: idphotos
    restart: unless-stopped
    ports:
      - 7861:7861
    environment:
      - DEFAULT_LANG=zh
    command: python3 -u app.py --host 0.0.0.0 --port 7861

然后执行下面的命令

1
2
3
4
5
6
7
8
9
10
# 新建文件夹 hivision_idphotos
mkdir -p /volume1/docker/hivision_idphotos

# 进入 hivision_idphotos 目录
cd /volume1/docker/hivision_idphotos

# 将 docker-compose.yml 放入当前目录

# 一键启动
docker-compose up -d

运行

在浏览器中输入 http://群晖IP:7861 就能看到主界面

在底部的 Examples 中随便选一张照片

支持不同的尺寸

不同的背景演示

1
2
3
4
5
国内证件照常用底色为白底、红底和蓝底。 

- 红底证件照多用于保险、医保、IC卡、结婚证等；
- 白底证件照一般用于身份证、驾驶证、护照、社保、签证、公务员以及事业单位考试等。 
- 蓝底证件照应用场景相对严谨正式一般用于毕业证、工作证、简历及相关考试证书上面。

虽然老苏的机器比较慢，但就是这种复杂背景的照片，速度也是很快的

如果临时需要制作证件照还是很方便的，不需要再找人用 PS 抠图了

如果你是开发人员，可以基于 HivisionIDPhoto 开发自己的应用。官方提供了 API 接口文档，可以启动 API 后端服务进行调试

1
2
3
4
5
6
7
# 运行容器( API 服务)
docker run -d \
   --restart unless-stopped \
   --name idphotos-api \
   -p 7862:8080 \
   linzeyi/hivision_idphotos \
   python3 deploy_api.py

参考文档

Zeyi-Lin/HivisionIDPhotos: ⚡️HivisionIDPhotos: a lightweight and efficient AI ID photos tools. 一个轻量级的AI证件照制作算法。
地址：https://github.com/Zeyi-Lin/HivisionIDPhotos

HivisionIDPhotos/docs/api_CN.md at master · Zeyi-Lin/HivisionIDPhotos
地址：https://github.com/Zeyi-Lin/HivisionIDPhotos/blob/master/docs/api_CN.md

ONG-Leo/HivisionIDPhotos-NAS: AI抠图项目HivisionIDPhotos，使用群晖NAS设置教程
地址：https://github.com/ONG-Leo/HivisionIDPhotos-NAS