HivisionIDPhotos 最强证件照一键制作工具

官方文档

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

功能介绍

上传一张照片，制作证件照，同时可开启美颜功能（美白强度/对比度/饱和度/锐化程度），添加水印以及选择不同的相纸排版。

安装

前提：2G 显存

为了方便使用，制作了一键整合包，关注本公众号，回复 photo 获取下载地址。

git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd  HivisionIDPhotos
conda create -n python310 python=3.10 -y
conda activate python310

pip install -r requirements.txt
pip install -r requirements-app.txt

# onnxruntime-gpu 版本要与 cuda 和 cudnn相匹配，匹配矩阵见 https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements
pip install onnxruntime-gpu==1.20.0
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

模型下载

需要两类模型：抠图模型和人脸检测模型。

• 抠图模型：通过 python scripts/download_model.py --models all 下载四种抠图模型到 hivision/creator/retinaface/weights；
• 人脸检测模型：三种检测模型如下
  • MTCNN：工程自带，无需下载模型，检测精度最低，实测下来还可以；
  • RetinaFace：从 此处 下载文件，之后放入 hivision/creator/retinaface/weights
  • Face++：旷视推出的在线人脸检测API，检测精度较高，但是需要在线使用，使用文档

启动应用

python app.py