证件照处理的自动化神器!

作者


项目简介

一键搞定证件照的裁切、换底色和排版！亮点：1. 完全离线运行，所有操作本地完成，安全又高效；2. 支持多种尺寸证件照自动裁切，满足不同需求；3. 提供WebUI界面，操作简单易上手

LiYing 可以完成人体、人脸自动识别，角度自动纠正，自动更换任意背景色，任意尺寸证件照自动裁切，并自动排版。

LiYing 可以完全离线运行。所有图像处理操作都在本地运行。

工作流程


效果展示
(1寸-5寸相片纸-3*3)
(2寸-5寸相片纸-2*2)
(1寸-6寸相片纸-4*2)


注：本项目仅针对证件照图像处理，而非要求任意照片图像都可以完美执行，所以该项目的输入图片应该是符合一般要求的单人肖像照片。

如果您使用复杂图片制作证件照出现意外情况属于正常现象。


开始使用

整合包

如果你是 Windows 用户且没有代码阅览需求，可以下载整合包（已在 Windows 7 SP1 & Windows 10 测试）

整合包从未包含模型，您可以参考 下载对应模型 章节说明来下载模型并放入正确的位置。

同时如果运行存在问题，请先尝试按照 先决条件 章节完善环境，如果没问题可以忽略。

运行整合包

运行 BAT 脚本：

cd LiYingrun.bat ./images/test1.jpg

运行 WebUI 界面：

# 运行 WebUIcd LiYingrun_webui.bat# 浏览器访问 127.0.0.1:7860

先决条件

  1. 依赖项

    • LiYing 依赖于 AGPicCompress
    • AGPicCompress 需要依赖于 mozjpeg 和 pngquant
    • 您可能需要手动安装 pngquant，参考 pngquant 官方文档

  2. pngquant 配置位置

    • 环境变量（推荐）
    • LiYing/src 目录下
    • LiYing/src 目录下的 ext 目录

  3. 系统要求

    • Windows 用户需要安装最新 Microsoft Visual C++ Redistributable 依赖
    • Windows 系统最低要求 Windows 7 SP1 及以上

从源码构建

  1. 克隆项目：
git clone https://github.com/aoguai/LiYingcd LiYing ## 进入 LiYing 目录pip install -r requirements.txt # install Python helpers' dependencies

注： 如果您使用的是 Windows 7 系统请您至少需要是 Windows 7 SP1 以上版本，且要求 onnxruntime==1.14.0, orjson==3.10.7, gradio==4.44.1

下载对应模型

您需要下载该项目使用到的模型并将其放置在 LiYing/src/model 中。或者您可以在 CIL 中指定模型路径。

用途
模型名称
下载链接
来源
人脸识别
Yunnet
下载链接
Yunnet
主体识别替换背景
RMBG-1.4/2.0
1.4 下载链接/2.0 下载链接
BRIA AI
人体识别
yolov8n-pose
下载链接
ultralytics

注： 对于 yolov8n-pose 模型，您需要将其导出为 ONNX 模型，您可以参考官方文档实现

同时，我们提供了转换好的 ONNX 模型，您可以直接下载使用：

下载方式
链接
Google Drive
下载链接
百度网盘
下载链接(提取码：ahr9)
Github releases
下载链接

运行

查看 CIL 帮助：

cd LiYing/srcpython main.py --help

对于 Window 用户，项目提供了 bat 运行脚本：

# 运行 BAT 脚本cd LiYingrun.bat ./images/test1.jpg

运行 WebUI：

cd LiYing/src/webuipython app.py

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.




扫码加入技术交流群，备注「开发语言-城市-昵称」

（文：GitHubStore）

发表评论