원 프롬프트-원 스토리: 텍스트 프롬프트가 캐릭터 아이덴티티 일관된 이미지를 생성합니다.

일반 소개

원 프롬프트 원 스토리(1Prompt1Story)는 하나의 프롬프트에서 일관된 이미지를 생성할 수 있도록 설계된 혁신적인 텍스트 이미지 생성 툴입니다. 1Prompt1Story는 텍스트 임베딩을 기반으로 하는 모든 텍스트-이미지 생성 모델에서 작동하며 다중 문자 생성, 컨트롤넷 가이드 생성, 개인화된 생성을 지원합니다. 이 도구는 단일 프롬프트 입력을 통해 생성된 이미지가 입력 설명과 매우 일치하도록 보장하며, 단일 값 가중치 및 정체성 보존 교차 주의 기법을 결합합니다.

기능 목록

• 일관된 아이덴티티 이미지 생성한 번의 프롬프트 입력을 통해 일관된 캐릭터 아이덴티티를 유지하는 이미지를 생성합니다.
• Gradio 데모사용자 경험 및 테스트를 위한 온라인 데모가 제공됩니다.
• Consistory+ 벤치마킹200개의 큐 세트를 포함하며, 각 큐는 5개에서 10개까지 8개 슈퍼클래스로 나뉩니다.
• 다중 역할 생성여러 문자가 포함된 이미지 생성 지원.
• 컨트롤넷 부트스트랩 생성컨트롤넷 기술을 통해 생성 프로세스를 안내합니다.
• 개인화사용자 요구 사항에 따라 개인화된 실제 이미지 생성을 지원합니다.

도움말 사용

설치 프로세스

1. 이 리포지토리를 복제합니다:

   git clone https://github.com/byliutao/1Prompt1Story

2. 리포지토리 디렉토리로 이동합니다:

   cd 1Prompt1Story

3. 가상 환경을 만들고 활성화합니다:

   conda create --name 1p1s python=3.10
conda activate 1p1s

4. 종속성을 설치합니다:

   conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
conda install conda-forge::transformers
conda install -c conda-forge diffusers
pip install opencv-python scipy gradio==4.44.1 sympy==1.13.1

사용 프로세스

1. 샘플 코드를 실행합니다:

   python main.py

2. Gradio 데모를 실행합니다:

   python app.py

3. Consistory+ 벤치마크를 실행합니다:

   python -m resource.gen_benchmark --save_dir ./result/benchmark --benchmark_path ./resource/consistory+.yaml

주요 기능

• 일관된 아이덴티티 이미지 생성입력 상자에 캐릭터의 정체성과 장면을 설명하는 문구를 입력하고 생성 버튼을 클릭하면 캐릭터의 정체성이 유지되는 이미지를 얻을 수 있습니다.
• 다중 역할 생성프롬프트에 여러 역할 설명을 포함하면 도구에서 모든 역할이 포함된 이미지를 자동으로 생성합니다.
• 컨트롤넷 부트스트랩 생성컨트롤넷 관련 설명이 프롬프트에 추가되며, 이미지가 설명과 높은 일관성을 유지하도록 컨트롤넷 기술에 따라 생성 프로세스가 안내됩니다.
• 개인화사용자가 입력한 개인화된 설명을 기반으로 요구 사항을 충족하는 실제 이미지를 생성합니다.