HivisionIDPhotos-证件照-免费开源的AI证件照制作工具

HivisionIDPhoto 旨在开发一种实用的证件照智能制作算法。
它利用一套完满的模子工作流程，实现对多种用户照相场景的辨认、抠图与证件照天生。
HivisionIDPhoto 可以做到：

• 轻量级抠图（仅需 CPU 即可快速推理）
  
• 根据差异尺寸规格天生差异的标准证件照、六寸排版照
  
• 美颜（waiting）
  
• 智能换正装（waiting）
  
  
  
  

HivisionIDPhotos的项目所在

• GitHub堆栈：https://github.com/Zeyi-Lin/HivisionIDPhotos
  
• 在线Demo体验所在：https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo
  

🔧 环境安装与依赖

• Python >= 3.7（项目重要测试在 python 3.10）
  
• onnxruntime
  
• OpenCV
  
• Option: Linux, Windows, MacOS
  

1. 克隆项目

1. git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
2. cd  HivisionIDPhotos

复制代码

2. （告急）安装依赖环境
   发起 conda 创建一个 python3.10 假造环境后，实行以下下令

1. pip install -r requirements.txt

复制代码

3. 下载权重文件
在Release下载权重文件hivision_modnet.onnx (24.7MB)，存到根目次下。
🚀 运行 Gradio Demo

1. python app.py

复制代码

运行步调将天生一个当地 Web 页面，在页面中可完成证件照的利用与交互。
🚀 Python 推理

1. 证件照制作

输入 1 张照片，得到 1 张标准证件照和 1 张高清证件照的 4 通道透明 png

1. python inference.py -i images/test.jpg -o ./idphoto.png -s '(413,295)'

复制代码

2. 增长底色

输入 1 张 4 通道透明 png，得到 1 张增长了底色的图像）

1. python inference.py -t add_background -i ./idphoto.png -o ./idhoto_ab.jpg  -c '(0,0,0)' -k 30

复制代码

3. 得到六寸排版照

输入 1 张 3 通道照片，得到 1 张六寸排版照

1. python inference.py -t generate_layout_photos -i ./idhoto_ab.jpg -o ./idhoto_layout.jpg  -s '(413,295)' -k 200

复制代码

⚡️ 摆设 API 服务

API 文档：Python | RestAPI（waiting）
启动后端

2. python deploy_api.py

复制代码

哀求 API 服务 - Python

1. 证件照制作

输入 1 张照片，得到 1 张标准证件照和 1 张高清证件照的 4 通道透明 png

1. python requests_api.py -u http://127.0.0.1:8080 -i images/test.jpg -o ./idphoto.png -s '(413,295)'

复制代码

2. 增长底色

输入 1 张 4 通道透明 png，得到 1 张增长了底色的图像

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

复制代码

3. 得到六寸排版照

输入 1 张 3 通道照片，得到 1 张六寸排版照

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

复制代码

🐳 Docker 摆设

1. 拉取或构建镜像

   以下方式三选一
  拉取镜像：
   该镜像构建于 ARM 架构呆板（如 Mac M1），x86 架构呆板利用请用 Dockerfile 构建
用这种方式的，在第 2 步开始把镜像名hivision_idphotos改为linzeyi/hivision_idphotos

1. docker pull linzeyi/hivision_idphotos:v1

复制代码

Dockrfile 构建镜像：
在确保将模子权重文件hivision_modnet.onnx放到根目次下后，在根目次实行：

1. docker build -t hivision_idphotos .

复制代码

Docker compose：
确保将模子权重文件 hivision_modnet.onnx 放在根目次下后，在根目次下实行：

1. docker compose build

复制代码

镜像打包完成后，运行以下下令启动 Gradio 服务：

1. docker compose up -d

复制代码

2. 运行 Gradio Demo

等候镜像封装完毕后，运行以下指令，即可开启 Gradio Demo 服务：

1. docker run -p 7860:7860 hivision_idphotos

复制代码

在你的当地访问http://127.0.0.1:7860即可利用。
3. 运行 API 后端服务

   现在docker镜像大概须要科学上网

1. docker run -p 8080:8080 hivision_idphotos python3 deploy_api.py

复制代码