HivisionIDPhoto 旨在开发一种实用的证件照智能制作算法。

它利用一套完善的模型工作流程,实现对多种用户拍照场景的识别、抠图与证件照生成。

HivisionIDPhoto 可以做到:

轻量级抠图(仅需 CPU 即可快速推理)
根据不同尺寸规格生成不同的标准证件照、六寸排版照
美颜(waiting)
智能换正装(waiting)
环境安装与依赖

Python >= 3.7(项目主要测试在 python 3.10)
onnxruntime
OpenCV
Option: Linux, Windows, MacOS
克隆项目
(重要)安装依赖环境

建议 conda 创建一个 python3.10 虚拟环境后,执行以下命令

pip install -r requirements.txt

下载权重文件

在我们的Release下载权重文件hivision_modnet.onnx (24.7MB),存到项目根目录下。

运行 Gradio Demo
python app.py

运行程序将生成一个本地 Web 页面,在页面中可完成证件照的操作与交互。

此处内容已隐藏,请登录后刷新页面查看