一款证件照AI自动化处理神器：AI-ID-Photo-LiYing，几步搞定证件照的后期处理

猫头虎

发布于 2025-07-14 08:18:47

51000

代码可运行

文章被收录于专栏：猫头虎博客专区猫头虎博客专区

运行总次数：0

代码可运行

一款证件照AI自动化处理神器：AI-ID-Photo-LiYing，几步搞定证件照的后期处理

它可以自动识别人体/人脸、纠正角度、更换背景颜色，裁剪成各种尺寸、并自动排版。 AI-ID-Photo-LiYing 是一款基于 AI 的智能证件照自动化处理神器，支持智能证件照、商务照、证件照抠图、AI 制作不同尺寸证件照等核心功能：

自动识别人脸与人体，AI 智能抠图与背景一键替换
智能角度校正，自动调整至最佳拍摄姿态
支持 1″、2″、3″、4″、5″、6″ 等多种多尺寸证件照制作
批量裁剪、压缩与智能排版，一键输出符合公安、出入境、简历、签证、资格证等多场景需求的高质量证件照
完全离线本地运行，无需联网，即刻拥有专属的 AI 证件照处理工作流程。

克隆项目地址：https://github.com/MaoTouHU/AI-ID-Photo-LiYing

项目介绍

AI-ID-Photo-LiYing 是一款完全开源、离线运行的证件照批量自动化处理工具，它集成人体/人脸识别、角度纠正、背景替换、尺寸裁剪、自动排版与压缩等功能，让证件照后期处理只需“几步搞定”。

人体/人脸自动识别：精准定位画面中的人物区域。
角度自动纠正：智能检测人脸/肩线倾斜，自动旋转至正位。
背景色一键替换：任意 RGB 颜色随心挑选。
多尺寸裁切：支持 1″ 至 6″ 各类常用规格。
自动排版：可自定义行列数、添加裁剪线、导出证件照排版表格。
本地运行：无需联网，隐私安全。
批量处理：一键处理整个文件夹，告别繁琐操作。

工作流程

整个处理流程一图看懂：

输入符合规范的单人肖像照片
人体/人脸检测与关键点定位
倾斜角度修正
自动扣图 & 背景填充
按指定规格裁剪
排版输出证件照表格

注意：AI-ID-Photo-LiYing 针对常规单人肖像照片效果最佳，复杂背景或多人画面可能出现意外结果。

效果展示

上图展示了从原图到背景替换、尺寸裁剪、自动排版的一体化流程。

开始使用

一、整合包（Windows 用户推荐）

如果你是 Windows 用户且只想快速体验，无需编译源码，可直接下载整合包 Release（已在 Windows 7 SP1 & Windows 10 测试通过）。

提示：整合包不含模型文件，请参考下文“下载对应模型”章节，将模型放入正确路径后再运行。

运行整合包

cd AI-ID-Photo-LiYing
run.bat ./images/test1.jpg

也可启动 WebUI 界面：

# 运行 WebUI
cd AI-ID-Photo-LiYing
run_webui.bat
# 浏览器访问 http://127.0.0.1:7860

二、先决条件

依赖项
- AGPicCompress（内部用于图像压缩）
- mozjpeg、pngquant（AGPicCompress 依赖）
- Windows 用户请安装最新的 Microsoft Visual C++ Redistributable
pngquant 安装与配置
- 推荐：添加到环境变量
- 或放置于 AI-ID-Photo-LiYing/src 目录
- 或放置于 AI-ID-Photo-LiYing/src/ext 子目录
系统要求
- Windows 7 SP1 及以上
- Linux/macOS 用户请自行安装对应依赖

三、从源码构建

克隆项目并进入目录：

git clone https://github.com/MaoTouHU/AI-ID-Photo-LiYing
cd AI-ID-Photo-LiYing
pip install -r requirements.txt

如果在 Windows 7 上运行，请确保 onnxruntime==1.14.0、orjson==3.10.7、gradio==4.44.1 等依赖版本匹配。

四、下载对应模型

将以下模型文件放置于 AI-ID-Photo-LiYing/src/model 目录，或启动时通过参数指定路径。

用途	模型名称	下载链接	来源
人脸识别	Yunnet	face_detection_yunet_2023mar.onnx	OpenCV Zoo
主体识别 & 换背景	RMBG-1.4/2.0	RMBG-1.4.onnx / RMBG-2.0 ONNX 目录	BRIA AI
人体关键点识别	yolov8n-pose	yolov8n-pose.pt	ultralytics

注意：yolov8n-pose 模型需根据官方指南导出为 ONNX，或直接下载本项目 Release 中提供的 ONNX 版本。

五、运行

查看 CLI 帮助：

cd AI-ID-Photo-LiYing/src
python main.py --help

通过参数灵活控制处理流程：

# 样例：更换背景为纯白，裁剪 1 寸照片，输出排版表
python main.py \
  --rmbg-model-path src/model/model.onnx \
  --yunet-model-path src/model/face_detection_yunet_2023mar.onnx \
  --yolov8-model-path src/model/yolov8n-pose.onnx \
  --photo-type 1in \
  --change-background \
  --rgb-list 255,255,255 \
  --save-path output/ \
  --layout-only

Windows BAT 脚本同理：

cd AI-ID-Photo-LiYing
run.bat ./images/test1.jpg

启动 WebUI（Gradio）：

cd AI-ID-Photo-LiYing/src/webui
python app.py
# 浏览器访问 http://127.0.0.1:7860

六、CIL 参数信息与帮助

python main.py --help 
Usage: main.py [OPTIONS] IMG_PATH

Options:
  -y, --yolov8-model-path PATH    YOLOv8 模型路径
  -u, --yunet-model-path PATH     YuNet 模型路径
  -r, --rmbg-model-path PATH      RMBG 模型路径
  -sz, --size-config PATH         尺寸配置文件路径
  -cl, --color-config PATH        颜色配置文件路径
  -b, --rgb-list RGB_LIST         RGB 通道值列表（英文逗号分隔），用于图像合成
  -s, --save-path PATH            保存路径
  -p, --photo-type TEXT           照片类型
  -ps, --photo-sheet-size TEXT    选择照片表格的尺寸
  -c, --compress / --no-compress  是否压缩图像（使用 AGPicCompress 压缩）
  -sv, --save-corrected / --no-save-corrected
                                  是否保存修正图像后的图片
  -bg, --change-background / --no-change-background
                                  是否替换背景
  -sb, --save-background / --no-save-background
                                  是否保存替换背景后的图像
  -lo, --layout-only              仅排版照片，不更换背景
  -sr, --sheet-rows INTEGER       照片表格的行数
  -sc, --sheet-cols INTEGER       照片表格的列数
  -rt, --rotate / --no-rotate     是否旋转照片90度
  -rs, --resize / --no-resize     是否调整图像尺寸
  -svr, --save-resized / --no-save-resized
                                  是否保存调整尺寸后的图像
  -al, --add-crop-lines / --no-add-crop-lines
                                  在照片表格上添加裁剪线
  -ts, --target-size INTEGER      目标文件大小（KB）。指定后将忽略质量和大小范围参数。
  -szr, --size-range SIZE_RANGE   文件大小范围（KB），格式为最小值,最大值（例如：10,20）
  -uc, --use-csv-size / --no-use-csv-size
                                  是否使用CSV中的文件大小限制
  --help                          Show this message and exit.

七、配置文件

在 data 目录中预置了常用证件照尺寸配置（size_XX.csv）与背景颜色配置（color_XX.csv），可根据示例模板自定义或扩展。

更新日志

2025/06/30 更新

新增 size_range 选项，允许用户指定文件大小范围（KB）
新增 target_size 选项，精确控制输出文件大小
支持 RMBG-2.0 与更高迭代版本的 yolov8
实现 CLI/BAT/WebUI 自动构建
新增模型路径可配置项
修复已知 Bug

一款证件照AI自动化处理神器：AI-ID-Photo-LiYing，几步搞定证件照的后期处理

一款证件照AI自动化处理神器：AI-ID-Photo-LiYing，几步搞定证件照的后期处理

一款证件照AI自动化处理神器：AI-ID-Photo-LiYing，几步搞定证件照的后期处理

项目介绍

工作流程

效果展示

开始使用

一、整合包（Windows 用户推荐）

运行整合包

二、先决条件

三、从源码构建

四、下载对应模型

五、运行

六、CIL 参数信息与帮助

七、配置文件

更新日志

相关项目

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐