免费开源 AI 证件照制作工具
今天给大家推荐 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
- 安装依赖环境
pip install -r requirements.txt - 下载体重文件
下载权限重文件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)’
———————————————
原文链接:https://blog.csdn.net/xs1997/article/details/141833034