Датасет · poloclub
2 млн изображений Stable Diffusion с промптами
poloclub/diffusiondb
2 млн изображений Stable Diffusion с промптами и гиперпараметрами реальных пользователей.
Что внутри
DiffusionDB — первый крупномасштабный датасет промптов text-to-image. Внутри 2 млн изображений, сгенерированных Stable Diffusion по промптам и гиперпараметрам, которые задавали реальные пользователи. Промпты на английском; метаданные включают параметры генерации.
Что можно сделать
- Обучение и оценка text-to-image моделей
- Детекция сгенерированных изображений и дипфейков
- Изучение связи промптов и генеративных моделей
- Задачи
- Текст → картинка, Картинка → текст
- Языки
- Английский
- Размер
- больше 1T записей
- Скачиваний за месяц
- 41,1 тыс.
- Лайков
- 679
Модели, обученные на нём
- FredZhang7/distilgpt2-stable-diffusion-v2Генерация текста 9 тыс.
- amd/Nitro-1-SDТекст → картинка 1 тыс.
- mradermacher/distilgpt2-stable-diffusion-v2-i1-GGUF 427
- amd/Nitro-1-PixArtТекст → картинка 274
- saggamer/horizon-1Текст → картинка 193
- mradermacher/distilgpt2-stable-diffusion-v2-GGUF 162
- vllm-sr/Decision-1.0-Route-0.6BZero-shot классификация 132
- amd/Nitro-T-0.6BТекст → картинка 121
Как загрузить
from datasets import load_dataset
ds = load_dataset("poloclub/diffusiondb")