再不去照相馆,最火Ai证件照片!HivisionIDPhotos本地部署使用教程

要求

项目依靠CPU,对显卡无要求:

  1. 安装 & 配置 Git (已安装,可跳过) Git 下载链接
  2. 安装 & 配置 Conda (已安装,可跳过) Conda 安装指南

输入以下 CMD 指令查看是否安装成功:

conda --version

安装步骤

  1. 克隆项目代码:
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd  HivisionIDPhotos

2. 创建 & 激活虚拟环境:

#创建虚拟环境
conda create -n HivisionIDPhotos python==3.10.6 -y 
#激活虚拟环境
conda activate HivisionIDPhotos

3.安装Python三方库:强烈推荐安装python 3.10版

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

4. 下载模型权重文件 (最新更新,以指南为准):

方式一:脚本下载

python scripts/download_model.py --models all

方式二:直接下载

存到项目的hivision/creator/weights目录下:

  • modnet_photographic_portrait_matting.onnx (24.7MB): MODNet官方权重,下载
  • hivision_modnet.onnx (24.7MB): 对纯色换底适配性更好的抠图模型,下载
  • mnn_hivision_modnet.mnn (24.7MB): mnn转换后的抠图模型 by zjkhahah下载
  • rmbg-1.4.onnx (176.2MB): BRIA AI 开源的抠图模型,下载后重命名为rmbg-1.4.onnx

5. 运行项目,多种方式:

5.1 手动运行代码

# 激活虚拟环境,每次运行均要选激活
conda activate HivisionIDPhotos
# 运行程序
python app.py


5.2 制作 .bat 启动文件并点击运行,内容如下:

@echo off

call conda activate HivisionIDPhotos

start http://127.0.0.1:7860

python app.py

pause


评论