免费开源 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

  1. 安装依赖环境
    pip install -r requirements.txt
  2. 下载体重文件
    下载权限重文件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

阅读剩余
THE END