今天给大家推荐 GitHub 上一个轻量级的 AI 证件照制作工具：HivisionIDPhoto。
实现了对多种用户拍照场景的识别、抠图与证件照生成，并提供不同尺寸规格的标准证件照。感兴趣的小伙伴可先收藏以备应急使用。
相关链接
GitHub：github.com/Zeyi-Lin/HivisionIDPhotos
在线使用：http://t.cn/A6RJ6ujG
引用项目
MTCNN：https://github.com/ipazc/mtcnn ModNet：https://github.com/ZHKKKe/MODNet
项目介绍
HivisionIDPhoto旨在开发一种实用的证件照智能制作算法。
它利用一套完善的模型工作流程，实现对多个用户拍摄场景的识别、抠图与服饰拍照生成。
HivisionIDPhoto可以实现：
轻量级抠图 根据不同尺寸规格生成不同的标准服装照、六寸排版照 美颜（等待） 智能换正装（等待）
环境安装与依赖
Python >= 3.7 onnxruntime OpenCV 选项：Linux、Windows、MacOS
1.克隆项目
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
2. 安装依赖环境
pip install -r requirements.txt
3. 下载体重文件
下载权限重文件hivision_modnet.onnx，存到根目录下。https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/tag/pretrained-model
4.运行 Gradio Demo
python app.py
运行程序将生成一个本地网页，在页面中可完成文档说明的操作与交互。
5.部署API服务
python deploy_api.py
6.请求API服务（Python）
7.使用Python给服务发送请求
a. 证件照制作（输入1张照片，获得1张标准证件照和1张高清证件照的4通道透明png）
python requests_api.py -u http://127.0.0.1:8080 -i test.jpg -o ./idphoto.png -s '(413,295)'
b. 增加底色（输入1张4通道透明png，获得1张增加了底色的图像）：
python requests_api.py -u http://127.0.0.1:8080 -t add_background -i ./idphoto.png -o ./idhoto_ab.jpg -c '(0,0,0)'
c. 获得六寸排版照（输入1张3通道照片，获得1张六寸排版照）：
python requests_api.py -u http://127.0.0.1:8080 -t generate_layout_photos -i ./idhoto_ab.jpg -o ./idhoto_layout.jpg -s '(413,295)'
项目更新
在线体验:SwanHub 演示 2023.12.1: 更新API部署（基于fastapi） 2023.6.20: 更新预设尺寸菜单 2023.6.19：更新排版照片 2023.6.13：更新中心渐变颜色 2023.6.11：更新上下突变颜色 2023.6.8: 更新自定义尺寸 2023.6.4：更新自定义底色、人脸检测Bug通知 2023.5.10：更新不改尺寸只换底
感谢你看到这里，也欢迎点击关注下方公众号并添加公众号小助手加入官方读者交流群，一个有趣有AI的AIGC公众号:关注AI、深度学习、计算机视觉、AIGC、Stable Diffusion、Sora等相关技术，欢迎一起交流学习💗～
