在当今数字化时代，证件照的需求无处不在。而 HivisionIDPhoto 项目的出现，为我们带来了一种全新的、便捷的证件照制作方式。

项目介绍

HivisionIDPhoto 是一个轻量级的 AI 证件照制作算法，它具有以下强大功能：

1.智能识别与抠图：能够识别多种用户拍照场景，并进行精准的抠图操作。

2.多种证件照生成：可以根据不同尺寸规格生成标准证件照和六寸排版照。

3.美颜功能（待开发）：未来将提供美颜功能，让您的证件照更加出色。

4.智能换正装（待开发）：期待后续能实现智能换正装的功能，满足更多需求。

该项目的优势在于其轻量级的特点，以及完善的模型工作流程，能够高效地完成证件照的制作。

安装使用

使用 HivisionIDPhoto 非常简单，只需按照以下步骤进行操作：

1.环境安装与依赖：

  • Python >= 3.7（项目主要测试在 python 3.10）。

  • 安装 onnxruntime、OpenCV 等依赖。

  • 克隆项目并进入目录。

  • 安装依赖环境：pip install -r requirements.txt。

  • 通过Github发布页下载权重文件 hivision_modnet.onnx 并存到根目录下。

2.运行 Gradio Demo：

运行 python app.py，将生成一个本地 Web 页面，在页面中可完成证件照的操作与交互。

3.部署 API 服务：

运行 python deploy_api.py。

4.请求 API 服务（Python）：

证件照制作：

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

增加底色：

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

得到六寸排版照：

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

此外，HivisionIDPhoto 还支持 Docker 部署，确保将模型权重文件放到根目录下后，执行以下命令即可开启 API 服务：

#在确保将模型权重文件hivision_modnet.onnx放到根目录下后，在根目录执行：
docker build -t hivision_idphotos.
docker run -p 8080:8080 hivision_idphotos

HivisionIDPhoto 项目引用了 MTCNN 和 ModNet 等相关技术，不断优化和提升自身性能。

如果您对这个项目感兴趣，不妨亲自尝试一下，让 HivisionIDPhoto 为您解决证件照应急制作的问题！

使用

可通过在线demo体验该项目。

项目地址

https://github.com/Zeyi-Lin/HivisionIDPhotos