Датасет · zzliang
GRIT: пары изображение-текст с привязкой к регионам
zzliang/GRIT
Крупный корпус пар изображение-текст, где фразы из подписи привязаны к соответствующим областям картинки.
Что внутри
GRIT — крупномасштабный корпус Grounded Image-Text pairs, построенный на парах изображение-текст из COYO-700M и LAION-2B. В нём извлекают текстовые фрагменты (существительные фразы и референциальные выражения) и связывают их с областями изображения. Размер — от 10 до 100 млн примеров; модальности — изображения, табличные данные и текст, язык английский.
Что можно сделать
- Обучить модель text-to-image с привязкой фраз к регионам
- Дообучить детектор объектов по текстовым описаниям
- Оценить visual grounding и zero-shot-классификацию
- Задачи
- Текст → картинка, Картинка → текст, Детекция объектов, Zero-shot классификация
- Языки
- Английский
- Размер
- 10M–100M записей
- Скачиваний за месяц
- 2 тыс.
- Лайков
- 169
Модели, обученные на нём
Как загрузить
from datasets import load_dataset
ds = load_dataset("zzliang/GRIT")