轻量级证件照制作 AI 工具 HivisionIDPhotos 介绍

介绍

HivisionIDPhotos 是一款基于 AI 技术的开源证件照生成工具，支持智能抠图、背景替换、多尺寸生成等功能，适合个人和企业快速制作标准证件照。

1. 智能抠图与背景替换

   • 采用深度学习模型（如 MODNet、RetinaFace）实现高精度人像分离，支持纯色背景替换（红、蓝、白等）。
   • 提供多种抠图模型选择，如 rmbg-1.4、birefnet-v1-lite，适应不同场景需求。

2. 多尺寸证件照生成

   • 支持常见证件照尺寸（1寸、2寸、护照、签证等），并可自定义尺寸。
   • 自动排版为六寸规格，方便批量打印。

3. 轻量化与高效处理

   • 纯离线运行，仅需 CPU 即可快速推理，无需高性能硬件。
   • 未来将推出美颜、智能换正装等进阶功能。

4. 目前在 GitHub 上已经收集到 15.7k 星🌟。
   

5. GitHub：https://github.com/Zeyi-Lin/HivisionIDPhotos

优势与适用场景

1. 隐私安全：支持纯本地部署，避免照片数据外传。
2. 灵活扩展：提供 API 接口，可集成至企业系统或第三方应用。
3. 多场景适用：
   • 个人：居家制作签证、简历照片等。
   • 企业：批量处理员工证件照，或集成到摄影服务中

使用

1. 环境安装与依赖

• Python >= 3.7（项目主要测试在 python 3.10）
• OS: Linux, Windows, MacOS

2. 克隆项目

git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd  HivisionIDPhotos

3. 安装依赖环境

# 建议 conda 创建一个 python3.10 虚拟环境后，执行以下命令
pip install -r requirements.txt
pip install -r requirements-app.txt

4. 下载人像抠图模型权重文件

python scripts/download_model.py --models all
# 如需指定下载某个模型
# python scripts/download_model.py --models modnet_photographic_portrait_matting

运行 Gradio Demo

运行程序将生成一个本地 Web 页面，在页面中可完成证件照的操作与交互。

python app.py

Python 推理

1. 证件照制作：输入 1 张照片，获得 1 张标准证件照和 1 张高清证件照的 4 通道透明 png

python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295

2. 人像抠图：输入 1 张照片，获得 1张 4 通道透明 png

python inference.py -t human_matting -i demo/images/test0.jpg -o ./idphoto_matting.png --matting_model hivision_modnet

3. 透明图增加底色：输入 1 张 4 通道透明 png，获得 1 张增加了底色的 3通道图像

python inference.py -t add_background -i ./idphoto.png -o ./idphoto_ab.jpg  -c 4f83ce -k 30 -r 1

4. 得到六寸排版照：输入 1 张 3 通道照片，获得 1 张六寸排版照

python inference.py -t generate_layout_photos -i ./idphoto_ab.jpg -o ./idphoto_layout.jpg  --height 413 --width 295 -k 200

5. 件照裁剪：输入 1 张 4 通道照片（抠图好的图像），获得 1 张标准证件照和 1 张高清证件照的 4 通道透明 png

python inference.py -t idphoto_crop -i ./idphoto_matting.png -o ./idphoto_crop.png --height 413 --width 295

其余方法

• 部署 API 服务和 Docker 部署请参考原项目说明。