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 页面,在页面中可完成证件照的操作与交互。
请登录后发表评论
注册
社交账号登录
停留在世界边缘,与之惜别