USO: 통합된 스타일과 주제 중심 이미지 생성 모델

1. 튜토리얼 소개

USO는 Bytedance의 UXO 팀이 2025년 8월에 출시한 콘텐츠와 스타일을 분리하고 재구성하는 통합 프레임워크입니다. 어떤 장면에서든 어떤 주제와 어떤 스타일이든 자유롭게 결합하여 높은 주제 일관성, 강력한 스타일 충실도, 그리고 자연스럽고 가소적이지 않은 느낌을 가진 이미지를 생성할 수 있습니다. USO는 대규모 트리플릿 데이터셋을 구축하고, 스타일 특징을 동시에 정렬하고 콘텐츠와 스타일을 분리하는 분리된 학습 체계를 채택하며, 모델 성능을 더욱 향상시키기 위해 스타일 보상 학습(SRL)을 도입합니다. USO는 스타일 유사성과 주제 충실도를 종합적으로 평가하기 위한 USO-Bench 벤치마크를 발표했습니다. 실험 결과 USO는 주제 일관성과 스타일 유사성 모두에서 오픈 소스 모델 중 최상위 수준에 도달했습니다. 관련 논문 결과는 다음과 같습니다.USO: 분리된 학습과 보상 학습을 통한 통합된 스타일과 주제 중심 세대".

이 튜토리얼에서 사용된 컴퓨팅 리소스는 RTX 4090 카드 1개입니다.

2. 효과 표시

테마/아이덴티티 기반 세대

피사체를 새로운 장면에 배치하려면 "개/남자/여자가..."와 같이 자연스러운 표현을 사용하세요. 레이아웃은 유지하면서 스타일만 전달하려면 "스타일을 ~의 스타일로 전달하세요..."와 같은 안내 신호를 사용하세요. 인물 사진 생성 시 USO는 피부 디테일이 풍부한 이미지를 생성하는 데 탁월합니다. 실용 가이드: 반신 촬영 시에는 반신 클로즈업을 사용하고, 포즈나 프레이밍이 크게 바뀌는 경우에는 전신 이미지를 사용하세요.

스타일 중심 세대

원하는 스타일의 이미지를 한두 장 업로드하고 자연어를 사용하여 원하는 이미지를 만들어 보세요. USO가 안내에 따라 업로드하신 스타일에 맞는 이미지를 자동으로 생성합니다.

스타일 테마 기반 생성

USO는 하나 또는 두 개의 스타일 참조를 사용하여 단일 콘텐츠 참조의 스타일을 지정할 수 있습니다. 레이아웃을 유지하는 빌드의 경우, 힌트를 비워두면 됩니다.

레이아웃 보존 빌드

레이아웃 오프셋 생성

3. 작업 단계

1. 컨테이너를 시작하세요

2. 사용 단계

"잘못된 게이트웨이"가 표시되면 모델이 초기화 중임을 의미합니다. 모델이 크기 때문에 약 2~3분 정도 기다리신 후 페이지를 새로고침해 주시기 바랍니다.

구체적인 매개변수:

• 생성 너비: 이미지 너비를 생성합니다.
• 생성 높이: 생성된 이미지의 높이입니다.
• 입력 크기 유지: 스타일 편집만 필요하거나 레이아웃을 유지하려는 경우 이 값을 True로 설정합니다.
• 고급 옵션:
  • 단계 수: 확산 모델 생성 프로세스의 반복 횟수를 제어합니다. 이론적으로 단계 수가 많을수록 이미지 품질이 향상되지만 생성 시간도 길어집니다.
  • 지침: 생성된 이미지가 프롬프트 단어와 참조 이미지를 얼마나 따르는지 제어합니다.
  • 콘텐츠 참조 크기: 콘텐츠 참조 이미지를 처리할 때, 피처 추출 전에 지정된 가장 긴 측면 길이로 크기가 조정될 수 있습니다(가로 세로 비율은 유지됨).
  • 시드(-1은 난수): 난수 생성기의 초기 상태를 제어합니다.

4. 토론

🖌️ 고품질 프로젝트를 발견하시면, 백그라운드에 메시지를 남겨 추천해주세요! 또한, 튜토리얼 교환 그룹도 만들었습니다. 친구들의 QR코드 스캔과 [SD 튜토리얼] 댓글을 통해 그룹에 가입하여 다양한 기술 이슈에 대해 논의하고 신청 결과를 공유해 주시기 바랍니다.↓

인용 정보

Github 사용자에게 감사드립니다 슈퍼양  이 튜토리얼의 배포. 이 프로젝트에 대한 인용 정보는 다음과 같습니다.

@article{wu2025uso,
    title={USO: Unified Style and Subject-Driven Generation via Disentangled and Reward Learning},
    author={Shaojin Wu and Mengqi Huang and Yufeng Cheng and Wenxu Wu and Jiahe Tian and Yiming Luo and Fei Ding and Qian He},
    year={2025},
    eprint={2508.18966},
    archivePrefix={arXiv},
    primaryClass={cs.CV},
}