简体中文 | English
LiYing is an automated photo processing program designed for automating general post-processing workflows in photo studios.
LiYing can automatically identify human bodies and faces, correct angles, change background colors, crop passport photos to any size, and automatically arrange them.
LiYing can run completely offline. All image processing operations are performed locally.
 |
 |
 |
|---|---|---|
 (1-inch on 5-inch photo paper - 3x3) |
 (2-inch on 5-inch photo paper - 2x2) |
 (1-inch on 6-inch photo paper - 4x2) |
Note: This project is specifically for processing passport photos and may not work perfectly on any arbitrary image. The input images should be standard single-person portrait photos.
It is normal for unexpected results to occur if you use complex images to create passport photos.
If you are a Windows user and do not need to review the code, you can download the bundled package (tested on Windows 7 SP1 & Windows 10). Extract it and drag your images or directory into run_en.bat to start LiYing.
You can install and configure LiYing locally by following the instructions below.
LiYing depends on AGPicCompress, which in turn requires mozjpeg and pngquant.
You may need to manually install pngquant. Refer to the official pngquant documentation and add it to the appropriate location.
LiYing checks for pngquant in the following locations, which you can configure:
- Environment variables (recommended)
- LiYing/src directory
extdirectory under LiYing/src
This allows AGPicCompress to locate and use pngquant for PNG image compression.
You need to install the latest Microsoft Visual C++ Redistributable.
If you are using Windows, your minimum version should be Windows 7 SP1 or higher.
You can obtain the LiYing project code by running:
git clone https://github.com/aoguai/LiYing
cd LiYing ## Enter the LiYing directory
pip install -r requirements.txt # Install Python helpers' dependenciesNote: If you are using Windows 7, ensure you have at least Windows 7 SP1 and onnxruntime==1.14.0.
Download the models used by the project and place them in LiYing/src/model, or specify the model paths in the command line.
| Purpose | Model Name | Download Link | Source |
|---|---|---|---|
| Face Recognition | Yunnet | Download Link | Yunnet |
| Subject Recognition and Background Replacement | RMBG-1.4 | Download Link | RMBG-1.4 |
| Body Recognition | yolov8n-pose | Download Link | ultralytics |
Note: For the yolov8n-pose model, you need to export it to an ONNX model. Refer to the official documentation for instructions.
cd LiYing/src
python main.py --helpFor Windows users, the project provides a batch script for convenience:
cd LiYing
run_en.bat ./images/test1.jpgpython main.py --help
Usage: main.py [OPTIONS] IMG_PATH
Options:
-y, --yolov8-model-path PATH Path to YOLOv8 model
-u, --yunet-model-path PATH Path to YuNet model
-r, --rmbg-model-path PATH Path to RMBG model
-b, --rgb-list RGB_LIST RGB channel values list (comma-separated)
for image composition
-s, --save-path PATH Path to save the output image
-p, --photo-type TEXT Photo types(supporting formats of XXpx x
XXpx or those specified in the data.ini)
--photo-sheet-size TEXT Size of the photo sheet(supporting formats
of XXpx x XXpx or those specified in the
data.ini)
-c, --compress / --no-compress Whether to compress the image
-sc, --save-corrected / --no-save-corrected
Whether to save the corrected image
-bg, --change-background / --no-change-background
Whether to change the background
-sb, --save-background / --no-save-background
Whether to save the image with changed
background
-sr, --sheet-rows INTEGER Number of rows in the photo sheet
-sc, --sheet-cols INTEGER Number of columns in the photo sheet
--rotate / --no-rotate Whether to rotate the photo by 90 degrees
-rs, --resize / --no-resize Whether to resize the image
-srz, --save-resized / --no-save-resized
Whether to save the resized image
--help Show this message and exit.
In this version, the standard ID photo settings are configured in data/data_en.ini. You can use the photo-type and photo-sheet-size parameters.
Additionally, you can modify this configuration file to customize the ID photo types. For the Chinese environment, the format is as follows:
[XXX]
print_size = XXXcm x XXXcm
electronic_size = XXXpx x XXXpx
resolution = XXXdpi
The section name [XXX] and the line electronic_size = XXXpx x XXXpx are mandatory.
The section name represents the values for the photo-type and photo-sheet-size parameters.
Furthermore, photo-type and photo-sheet-size also support direct input of strings in the format XXXpx x XXXpx, representing width and height.
- 2024/08/06 Update
- Added support for entering width and height in pixels directly for
photo-typeandphoto-sheet-size, and support for configuration viadata.ini. - Fixed issues related to some i18n configurations; now compatible with both English and Chinese settings.
- Fixed other known bugs.
- Added support for entering width and height in pixels directly for
The project was created to help my parents complete their work more easily. I would like to thank my parents for their support.
Special thanks to the following projects and contributors for providing models and theories:
You might also be interested in the image compression part, which is another open-source project of mine:
It depends on:
LiYing is an open-source project, and community participation is highly welcomed. To contribute to this project, please follow the Contribution Guide.
LiYing is open-sourced under the AGPL-3.0 license. For details, please refer to the LICENSE file.