介绍
HivisionIDPhotos 是一款基于 AI 技术的开源证件照生成工具，支持智能抠图、背景替换、多尺寸生成等功能，适合个人和企业快速制作标准证件照。
- 智能抠图与背景替换
- 采用深度学习模型（如 MODNet、RetinaFace）实现高精度人像分离，支持纯色背景替换（红、蓝、白等）。提供多种抠图模型选择，如 rmbg-1.4、birefnet-v1-lite，适应不同场景需求。
- 支持常见证件照尺寸（1寸、2寸、护照、签证等），并可自定义尺寸。自动排版为六寸规格，方便批量打印。
- 纯离线运行，仅需 CPU 即可快速推理，无需高性能硬件。未来将推出美颜、智能换正装等进阶功能。
使用
- 环境安装与依赖
Python >= 3.7（项目主要测试在 python 3.10）OS: Linux, Windows, MacOS
- 克隆项目
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.gitcd HivisionIDPhotos- 安装依赖环境
# 建议 conda 创建一个 python3.10 虚拟环境后，执行以下命令pip install -r requirements.txtpip 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.pyPython 推理
- 证件照制作：输入 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 部署请参考原项目说明。
