KNX Tagger

세부 정보

파일 다운로드

모델 설명

KNX-Tagger

Discord Support us on Ko-fi


Stable Diffusion/Flux LoRA 학습을 위해 Google의 Gemini API를 사용하는 이미지 태깅 도구입니다.

모든 이슈/기능 요청은 Github에서 처리해야 합니다:

https://github.com/Ktiseos-Nyx/Knx-Tagger

요구 사항

  • Python 3.10+

  • PyQt6

  • Google Gemini API 키(무료 계층 제공 가능)

  • 지원되는 모델: gemini-2.5-flash, gemini-2.5-pro, gemini-flash-latest

기능

  • 다중 태깅 모드:

    • Danbooru 스타일 태그(쉼표로 구분됨)

    • 자연어 캡션

    • 사용자 지정 프롬프트 및 사용자 지정 지시문

  • 초점 태깅을 위한 편향 모드:

    • 캐릭터 중심: 캐릭터 세부 사항, 표정, 자세 강조

    • 스타일 중심: 예술적 기법, 구성, 조명 강조

    • 사실성 중심: 환경 및 사진적 요소 상세화

    • 환경 중심: 장면 맥락, 설정, 분위기

  • 유연한 워크플로우:

    • 테스트 모드: 먼저 10장의 이미지에 태그 미리보기

    • 일괄 모드: 전체 폴더 처리

    • 이미 태그된 이미지 건너뛰기

    • 태그 필터링 및 제외

    • 활성화 태그 앞추가(LoRA 트리거용)

  • Kohya 호환 출력:

    • 이미지와 함께 .txt 또는 .caption 파일 생성

    • SD/Flux 학습 워크플로우에 완벽하게 적합

  • 테마 시스템:

    • KNX 컬렉션에서 제공하는 213개 이상의 사전 제작된 테마

    • qt-material, unreal-stylesheet 및 사용자 정의 QSS 지원

    • 검색 가능한 테마 선택 대화상자

    • 사용자 정의 테마 생성을 위한 완전한 스킨 문서

    • Krita, GIMP, Photoshop, Affinity, Illustrator용 가이드

  • 비용 추적:

    • 처리 전 API 비용 예측

    • 효율적인 gemini-2.5-flash 모델 사용

    • 속도 향상을 위한 병렬 처리(10개 워커)

설치

권장: 패키지로 설치(knx-tagger 명령어 사용)

# 리포지토리 복제
git clone https://github.com/Ktiseos-Nyx/Knx-Tagger.git
cd Knx-Tagger

# 편집 가능 모드로 설치
pip install -e .

# 이제 어디서든지 실행 가능:
knx-tagger

대안 1: venv 사용(표준)

# 가상 환경 생성
python -m venv venv

# 활성화
# macOS/Linux(bash/zsh):
source venv/bin/activate

# macOS/Linux(fish 쉘):
source venv/bin/activate.fish

# Windows:
venv\Scripts\activate

# 패키지로 설치
pip install -e .

대안 2: uv 사용(더 빠름, 선택 사항)

uv는 빠른 Python 패키지 설치 프로그램입니다. 속도를 원한다면 꼭 추천합니다!

# uv 설치(없는 경우)
curl -LsSf https://astral.sh/uv/install.sh | sh

# venv 생성 및 설치를 한 번에
uv venv
source .venv/bin/activate  # 또는 fish 쉘의 경우 .venv/bin/activate.fish
uv pip install -e .

API 키 발급

  1. Gemini API 키 발급:

  2. API 키 구성(선택 사항):

    cp .env.example .env
    # .env 파일을 편집하고 키를 추가
    

    또는 첫 실행 시 프롬프트에 직접 입력하세요. 앱이 자동으로 저장해 줍니다!

사용법

패키지로 설치한 경우:

knx-tagger

또는 직접 실행:

python main.py
# 또는
python -m knx_tagger

참고: 콘솔에 ALTS creds ignored. Not running on GCP...와 같은 무해한 경고가 표시될 수 있습니다. 이는 Google 라이브러리가 Google Cloud Platform이 아닌 환경임을 알리는 것이며, 안전히 무시하셔도 됩니다. 앱은 올바르게 작동하는 API 키 인증을 사용합니다.

워크플로우:

  1. 이미지가 있는 폴더 선택(JPEG, PNG, WebP)

  2. 태깅 모드 선택:

    • Danbooru: 창의적이고 설명적인 태그

    • 자연어: 전체 문장 캡션

    • 사용자 지정: 직접 프롬프트 작성

  3. 먼저 테스트: 10장의 이미지에서 테스트하고 품질 확인

  4. 일괄 처리: 만족스러우면 전체 폴더 태깅

  5. 출력: 각 이미지와 함께 .txt 또는 .caption 파일 생성

테마

KNX-Tagger에는 213개 이상의 사전 제작된 테마가 포함되어 있습니다! 상단 메뉴의 🎨 테마를 통해 접근하세요.

테마 둘러보기

  • 🎨 테마를 클릭하여 테마 선택기를 열기

  • 카테고리별로 탐색하거나 이름으로 검색

  • 더블 클릭하여 즉시 테마 적용

  • 현재 테마는 강조 표시됨

사용자 정의 테마 제작

Winamp 스타일 스킨을 만들고 싶으신가요? 다음 문서를 참고하세요:

  • knx_tagger/themes/SKINNING_GUIDE.md - 완전한 QSS 스타일링 참조

  • knx_tagger/themes/WIDGET_LAYOUT_REFERENCE.md - 템플릿 생성을 위한 위젯 위치

  • knx_tagger/themes/SOFTWARE_SPECIFIC_GUIDES.md - Krita, GIMP, Photoshop 등에 대한 튜토리얼

테마 카테고리

  • AI 테마(ChatGPT, Claude, Gemini, Llama)

  • 레트로 UI(MS-DOS, Windows XP, Mac OS 9, Linux 배포판)

  • 게임(Genshin Impact, Pokemon, Final Fantasy, Playstation)

  • 감성(Vaporwave, Kawaii, Neon, 일몰 그라데이션)

  • 음식 및 브랜드(Taco Bell, Domino's, Coca-Cola)

  • 대중 문화(MLP, Deadpool, X-Men, VTubers)

  • 그리고 더 많은 신기한 옵션!

  • Danbooru 모드는 기존 booru 집합을 넘어서는 새로운 태그를 창조하는 데 탁월합니다

  • 일괄 처리 전에 테스트하여 프롬프트 품질을 확인하세요

  • 기존 이미지 건너뛰기를 통해 이미 처리된 이미지를 재태깅하지 마세요

  • 비용 예측은 근사치이며 실제 비용은 다를 수 있습니다

  • 편향 모드는 특정 요소(캐릭터, 스타일, 환경, 사실성)에 집중하는 데 도움이 됩니다

  • 태그 제외 기능을 사용하여 AI가 생성한 태그에서 원치 않는 항목을 필터링하세요

  • 일부 테마는 약간 어색해 보일 수 있습니다. 이들은 원래 Dataset-Tools용으로 설계되었습니다!

확장

KNX-Tagger는 모듈식으로 설계되었습니다:

  • knx_tagger/core/gemini_tagger.py - API 통합

  • knx_tagger/utils/file_handler.py - 파일 작업

  • knx_tagger/ui/main_window.py - PyQt6 인터페이스

대규모 애플리케이션에 통합하거나 라이브러리로 사용하기 쉽습니다.

라이선스

MIT

이 모델로 만든 이미지

이미지를 찾을 수 없습니다.