轻量级证件照制作 AI 工具 HivisionIDPhotos 介绍
介绍
HivisionIDPhotos 是一款基于 AI 技术的开源证件照生成工具，支持智能抠图、背景替换、多尺寸生成等功能，适合个人和企业快速制作标准证件照。
-
智能抠图与背景替换
- 采用深度学习模型（如 MODNet、RetinaFace）实现高精度人像分离，支持纯色背景替换（红、蓝、白等）。
- 提供多种抠图模型选择，如 rmbg-1.4、birefnet-v1-lite，适应不同场景需求。
-
多尺寸证件照生成
- 支持常见证件照尺寸（1寸、2寸、护照、签证等），并可自定义尺寸。
- 自动排版为六寸规格，方便批量打印。
-
轻量化与高效处理
- 纯离线运行，仅需 CPU 即可快速推理，无需高性能硬件。
- 未来将推出美颜、智能换正装等进阶功能。
-
目前在 GitHub 上已经收集到 15.7k 星🌟。
-
GitHub：https://github.com/Zeyi-Lin/HivisionIDPhotos
优势与适用场景
- 隐私安全：支持纯本地部署，避免照片数据外传。
- 灵活扩展：提供 API 接口，可集成至企业系统或第三方应用。
- 多场景适用：
- 个人：居家制作签证、简历照片等。
- 企业：批量处理员工证件照，或集成到摄影服务中
使用
- 环境安装与依赖
- Python >= 3.7（项目主要测试在 python 3.10）
- OS: Linux, Windows, MacOS
- 克隆项目
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
- 安装依赖环境
# 建议 conda 创建一个 python3.10 虚拟环境后，执行以下命令
pip install -r requirements.txt
pip install -r requirements-app.txt
- 下载人像抠图模型权重文件
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 张高清证件照的 4 通道透明 png
python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295
- 人像抠图：输入 1 张照片，获得 1张 4 通道透明 png
python inference.py -t human_matting -i demo/images/test0.jpg -o ./idphoto_matting.png --matting_model hivision_modnet
- 透明图增加底色：输入 1 张 4 通道透明 png，获得 1 张增加了底色的 3通道图像
python inference.py -t add_background -i ./idphoto.png -o ./idphoto_ab.jpg -c 4f83ce -k 30 -r 1
- 得到六寸排版照：输入 1 张 3 通道照片，获得 1 张六寸排版照
python inference.py -t generate_layout_photos -i ./idphoto_ab.jpg -o ./idphoto_layout.jpg --height 413 --width 295 -k 200
- 件照裁剪：输入 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 部署请参考原项目说明。