perfectgf/lora-dataset-studio

해결하는 문제

LoRA Dataset Studio는 LoRA 학습용 이미지 데이터셋 생성의 전체 수명 주기를 위한 통합 브라우저 기반 인터페이스를 제공합니다. 이미지 수집, 정리, 캡셔닝, 학습을 위해 여러 개별 도구를 오가야 하는 필요성을 없애고, AI 모델에 특정 캐릭터, 개념, 스타일을 가르치는 과정을 간소화합니다.

작동 방식

이 소프트웨어는 여러 전문 AI 도구를 통합하는 중앙 허브 역할을 합니다. 사용자 머신에서 로컬로 실행되며, 실제 학습 프로세스에는 ai-toolkit을, 로컬 이미지 생성 및 체크포인트 테스트에는 ComfyUI를 활용합니다. 워크플로우를 여러 단계로 구성합니다.

• 데이터셋 관리: 이미지 가져오기, 뱅크로 구성, 크롭, 미러링, 품질 점수를 통한 큐레이션.
• 캡셔닝: 로컬 비전 모델 또는 JoyCaption을 사용하여 이미지에 대한 설명이나 태그 생성.
• 학습: 다양한 모델 패밀리(SDXL, FLUX.1, Qwen-Image 2.1 등)에 대한 안내 레시피 제공 및 학습 큐 관리.
• 검토: 고정 시드와 프롬프트 배치를 사용하여 다른 체크포인트를 비교하고 학습된 LoRA의 품질을 평가.

대상 사용자

복잡한 명령줄 도구나 수동 파일 구성을 관리하지 않고도 이미지 생성 모델용 고품질 LoRA를 만들고자 하는 AI 아티스트와 개발자를 위해 설계되었습니다.

주요 기능

• 엔드투엔드 워크플로우: 이미지 뱅크 큐레이션과 텍스트 감지부터 학습 및 최종 테스트까지 모든 것을 다룹니다.
• 확장 가능한 플러그인 시스템: 클라우드 학습(vast.ai 경유), 웹 스크래핑, Civitai 또는 Hugging Face에 게시하기 위한 선택적 플러그인을 제공합니다.
• 다중 모델 지원: FLUX.1, SDXL, Krea 2를 포함한 광범위한 학습 패밀리를 지원합니다.
• 통합 테스트: 계보 및 순위 도구를 사용하여 체크포인트와 강도를 비교하는 "스튜디오"를 포함합니다.