← 返回 Skill 列表
extension
分类: 内容与媒体API Key 暂未确认

lora-dataset-sorter

基于 Qwen2.5-VL 视觉大模型的 LoRA 训练集 AI 智能分类工具。专门解决 AIGC 炼丹师手动整理数据集耗时费力的痛点,可自动识别图片构图与视角(全身、半身、面部特写等),并智能过滤崩坏废图,一键将图片自动分类移至指定子文件夹,大幅提升数据集准备与打标效率。

person作者: atent2Lhubgithub

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:

  1. Verify Parameters:

    • source_folder: Directory containing raw image files (.jpg, .png, .webp).
    • categories: Target sub-categories (Default: full_body, half_body, close_up).
  2. Classification Logic:

    • Inspect image framing and details.
    • Categorize into corresponding folders (full_body, half_body, close_up).
    • Flag corrupted/distorted images into bad_quality.
  3. Execution:

    • Execute script scripts/sort_images.py to move files.