LoRA Dataset Auto-Sorter
This skill utilizes Qwen-VL (Vision-Language LLM) to analyze, inspect, and automatically classify training images for LoRA models into structured sub-directories.
Standard Operating Procedure (SOP)
When activated, follow these steps to organize the dataset:
-
Verify Parameters:
source_folder: Directory containing raw image files (.jpg, .png, .webp).categories: Target sub-categories (Default:full_body,half_body,close_up).
-
Classification Logic:
- Inspect image framing and details.
- Categorize into corresponding folders (
full_body,half_body,close_up). - Flag corrupted/distorted images into
bad_quality.
-
Execution:
- Execute script
scripts/sort_images.pyto move files.
- Execute script
Scan to join WeChat group