AI 证件照制作工具：HivisionIDPhoto源码

AI 证件照制作工具：HivisionIDPhoto源码

项目介绍

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)'