🚀 Playground v2 – 1024px美学模型
本项目包含一个能够生成1024x1024高分辨率美学图像的模型。你可以使用Hugging Face 🧨 Diffusers库来调用该模型。

Playground v2 是一个基于扩散模型的文生图生成模型。该模型由 Playground 的研究团队从头开始训练。
根据Playground的 用户研究,Playground v2生成的图像比Stable Diffusion XL生成的图像受欢迎程度高出 2.5 倍。
我们很高兴能向社区发布不同训练阶段的 中间检查点,包括评估指标,希望这能促进对图像生成基础模型的进一步研究。
最后,我们引入了一个新的基准测试 MJHQ - 30K,用于自动评估模型的美学质量。
更多详情请查看我们的 博客。
🚀 快速开始
本项目的模型可以借助Hugging Face 🧨 Diffusers库来使用,下面为你介绍具体的使用步骤。
✨ 主要特性
- 高质量图像生成:能够生成1024x1024分辨率的高美学图像。
- 用户偏好高:根据用户研究,生成的图像比Stable Diffusion XL生成的图像受欢迎程度高出2.5倍。
- 多阶段模型发布:发布不同训练阶段的中间检查点,促进基础模型研究。
- 新基准测试:引入MJHQ - 30K基准测试,用于自动评估模型的美学质量。
📦 安装指南
安装 diffusers >= 0.24.0
及一些依赖项:
pip install transformers accelerate safetensors
💻 使用示例
基础用法
使用 🧨 Diffusers 调用模型,运行以下代码片段。
注意:建议使用 guidance_scale = 3.0
。
from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained(
"playgroundai/playground-v2-1024px-aesthetic",
torch_dtype=torch.float16,
use_safetensors=True,
add_watermarker=False,
variant="fp16"
)
pipe.to("cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt=prompt, guidance_scale=3.0).images[0]
其他使用方式
若要在诸如Automatic1111或ComfyUI等软件中使用该模型,可使用 playground-v2.fp16.safetensors
文件。
📚 详细文档
模型描述
用户研究

根据Playground进行的用户研究,涉及2600多个提示和数千名用户,Playground v2生成的图像比 Stable Diffusion XL 生成的图像受欢迎程度高出 2.5 倍。
我们按照标准做法,在 PartiPrompts 和Playground团队策划的内部提示数据集上报告用户偏好指标。“Internal 1K” 提示数据集具有多样性,涵盖各种类别和任务。
在用户研究期间,我们指导用户从(1)美学偏好和(2)图像 - 文本对齐两个方面评估图像对。
MJHQ - 30K基准测试

我们引入了一个新的基准测试 MJHQ - 30K,用于自动评估模型的美学质量。该基准测试在高质量数据集上计算FID以衡量美学质量。
我们从Midjourney精心策划了一个高质量数据集,包含10个常见类别,每个类别有3000个样本。按照惯例,我们使用美学分数和CLIP分数来确保高图像质量和高图像 - 文本对齐。此外,我们还特别注意使每个类别内的数据具有多样性。
对于Playground v2,我们报告了整体FID和每个类别的FID。所有FID指标均在1024x1024分辨率下计算。我们的基准测试结果表明,我们的模型在整体FID和所有类别FID上均优于SDXL - 1 - 0 - refiner,特别是在人物和时尚类别上。这与用户研究结果一致,表明人类偏好与MJHQ - 30K基准测试的FID分数之间存在相关性。
我们将此基准测试向公众发布,并鼓励社区采用它来评估其模型的美学质量。
中间基础模型
除了 playground - v2 - 1024px - aesthetic,我们还向社区发布了不同训练阶段的中间检查点,以促进像素级基础模型研究。在这里,我们报告了在MSCOCO14评估集上的FID分数和CLIP分数以供参考。(请注意,我们报告的数字可能与SDXL发布结果中的数字不同,因为我们的提示列表可能不同。)
引用方式
@misc{playground-v2,
url={[https://huggingface.co/playgroundai/playground-v2-1024px-aesthetic](https://huggingface.co/playgroundai/playground-v2-1024px-aesthetic)},
title={Playground v2},
author={Li, Daiqing and Kamko, Aleks and Sabet, Ali and Akhgari, Ehsan and Xu, Linmiao and Doshi, Suhail}
}
📄 许可证
本项目采用 Playground v2社区许可证。