什么是"HivisionIDPhotos"?
HivisionIDPhotos 是一款轻量级且高效的 AI 证件照制作工具,旨在为用户提供便捷的证件照生成服务。通过一系列完善的 AI 模型工作流程,HivisionIDPhotos 能够识别多种用户拍照场景,进行精准抠图,并生成符合标准的证件照。
"HivisionIDPhotos"有哪些功能?
- 轻量级抠图:纯离线操作,仅需 CPU 即可快速推理。
- 证件照生成:根据不同尺寸规格生成标准证件照和六寸排版照。
- 美颜功能:提供美白、智能换正装等功能(即将上线)。
- 多语言支持:支持中文、英文、日语和韩语。
- API服务:提供多种 API 接口,支持证件照裁切、加水印、设置照片 KB 值大小等功能。
产品特点:
- 高效性:利用高效的 AI 模型,快速完成抠图和证件照生成。
- 多样性:支持多种人脸检测模型,如 MTCNN、RetinaFace 和 Face++,满足不同精度需求。
- 易用性:提供 Gradio Demo,本地 Web 页面操作简单直观。
- 扩展性:支持 GPU 加速推理,提升处理速度。
- 社区支持:拥有活跃的社区,提供多种应用和扩展,如 Windows 客户端应用、ComfyUI 工作流等。
应用场景:
- 个人用户:快速制作符合标准的证件照,满足签证、护照、身份证等需求。
- 企业用户:批量处理员工证件照,提升工作效率。
- 教育机构:为学生制作标准化证件照,便于管理。
- 摄影工作室:提供高效的证件照制作服务,提升客户满意度。
- 在线服务平台:集成 HivisionIDPhotos API,为用户提供在线证件照制作服务。
"HivisionIDPhotos"如何使用?
环境准备:
安装 Python >= 3.7(推荐使用 Python 3.10)。
克隆项目并安装依赖:
bash
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
pip install -r requirements.txt
pip install -r requirements-app.txt下载权重文件并放置到指定目录:
bash
python scripts/download_model.py --models all
运行 Gradio Demo:
bash
python app.py运行程序后,将生成一个本地 Web 页面,用户可以在页面中完成证件照的操作与交互。
Python 推理:
证件照制作:
bash
python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295人像抠图:
bash
python inference.py -t human_matting -i demo/images/test0.jpg -o ./idphoto_matting.png --matting_model hivision_modnet透明图增加底色:
bash
python inference.py -t add_background -i ./idphoto.png -o ./idphoto_with_bg.png
常见问题:
如何选择合适的人脸检测模型?
- MTCNN:适合高性能 CPU 推理,检测精度较低。
- RetinaFace:适合中等速度的 CPU 推理,检测精度较高。
- Face++:在线人脸检测 API,检测精度最高。
如何使用 GPU 加速推理?
确保已安装 CUDA 和 cuDNN。
安装对应的 onnxruntime-gpu 版本:
bash
pip install onnxruntime-gpu==1.18.0调用如 birefnet-v1-lite 模型时将自动利用 GPU 加速推理。
如何在 Windows 上运行 HivisionIDPhotos?
- 参考社区构建的 Windows 客户端应用 HivisionIDPhotos-windows-GUI。
通过以上步骤,用户可以轻松使用 HivisionIDPhotos 进行证件照制作,满足多种应用场景的需求。