Датасет · nvidia
Аннотации действий геймпада к игровым видео
nvidia/NitroGen
Покадровая разметка действий геймпада для публичных игровых видео — для обучения робототехнических моделей.
Что внутри
Датасет содержит аннотации действий для общедоступных видео геймплея. Каждый кадр размечен действиями геймпада с помощью внутренней модели NVIDIA. Для воспроизведения результатов статьи нужна дополнительная фильтрация, например удаление IDLE-кадров. Данные лежат в каталоге actions.
Что можно сделать
- Обучить vision-to-action политику для робототехники
- Проверить imitation learning на игровых видео
- Дообучить LoRA для предсказания действий по кадрам
- Задачи
- Робототехника
- Скачиваний за месяц
- 3 тыс.
- Лайков
- 230
Модели, обученные на нём
Как загрузить
from datasets import load_dataset
ds = load_dataset("nvidia/NitroGen")