今天给大家推荐一个轻量级的 AI 证件照制作工具 —— HivisionIDPhoto。
简介
HivisionIDPhoto 旨在开发一种实用、系统性的证件照智能制作算法。
它利用一套完善的AI模型工作流程，实现对多种用户拍照场景的识别、抠图与证件照生成。
主要功能
-
轻量级抠图（纯离线，仅需 CPU 即可快速推理） -
根据不同尺寸规格生成不同的标准证件照、六寸排版照 -
支持 纯离线 或 端云 推理 -
美颜 -
智能换正装（waiting）
运行
0.准备工作
环境安装与依赖：
-
Python >= 3.7（项目主要测试在 python 3.10） -
OS: Linux, Windows, MacOS
1. 克隆项目
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
2. 安装依赖环境
建议 conda 创建一个 python3.10 虚拟环境后，执行以下命令
pip install -r requirements.txt
pip install -r requirements-app.txt
3. 下载人像抠图模型权重文件
脚本下载
python scripts/download_model.py --models all
# 如需指定下载某个模型
# python scripts/download_model.py --models modnet_photographic_portrait_matting
4. 人脸检测模型配置（可选）
5. 性能参考
测试环境为Mac M1 Max 64GB，非GPU加速，测试图片分辨率为 512×715(1) 与 764×1146(2)。
6. GPU推理加速（可选）
在当前版本，可被英伟达GPU加速的模型为birefnet-v1-lite，并请确保你有16GB左右的显存。
如需使用英伟达GPU加速推理，在确保你已经安装CUDA与cuDNN后，根据onnxruntime-gpu文档找到对应的onnxruntime-gpu版本安装，以及根据pytorch官网找到对应的torch版本安装。
# 假如你的电脑安装的是CUDA 12.x, cuDNN 8
# 安装torch是可选的，如果你始终配置不好cuDNN，那么试试安装torch
pip install onnxruntime-gpu==1.18.0
pip install torch --index-url https://download.pytorch.org/whl/cu121
完成安装后，调用birefnet-v1-lite模型即可利用GPU加速推理。
TIPS: CUDA 支持向下兼容。比如你的 CUDA 版本为 12.6，torch 官方目前支持的最高版本为 12.4（<12.6），torch仍可以正常使用CUDA。
7.运行 Gradio Demo
python app.py
运行程序将生成一个本地 Web 页面，在页面中可完成证件照的操作与交互。
开源地址
https://github.com/Zeyi-Lin/HivisionIDPhotos
构建高质量的技术交流社群，欢迎从事编程开发、技术招聘HR进群，也欢迎大家分享自己公司的内推信息，相互帮助，一起进步！
文明发言，以
交流技术、职位内推、行业探讨为主
广告人士勿入，切勿轻信私聊，防止被骗
加我好友，拉你进群
原文始发于微信公众号（极客之家）：11.8k star，一个轻量级的 AI 证件照制作工具！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/305970.html
