Дифференцируемая библиотека компьютерного зрения на базе PyTorch с поддержкой GPU для обработки изображений, геометрии, аугментаций и сопоставления признаков.
NVIDIA/DALI
Библиотека с ускорением на GPU для загрузки и предобработки изображений, видео и аудио, ускоряющая обучение и инференс моделей глубокого обучения.
- Звёзды
- 5,792
- Форки
- 691
- Открытые issues
- 222
- Последний коммит
- 20 часов назад
Что это?
NVIDIA DALI (Data Loading Library) переносит загрузку и предобработку данных для глубокого обучения с CPU на GPU. Пайплайны охватывают декодирование, кадрирование, изменение размера и другие аугментации изображений, видео и аудио, а встроенный движок выполнения отвечает за предварительную загрузку данных, параллельное выполнение и формирование батчей.
Пайплайны создаются с помощью функционального Python API и могут заменить загрузчики данных и итераторы в TensorFlow, PyTorch, PaddlePaddle и JAX без изменений в остальном коде. Библиотека читает данные в форматах LMDB, RecordIO, TFRecord, COCO, JPEG, JPEG 2000, WAV, FLAC, OGG, H.264, VP9 и HEVC, а GPUDirect Storage позволяет загружать их напрямую из хранилища в память GPU.
DALI также интегрируется с NVIDIA Triton Inference Server через специальный backend и поддерживает пользовательские операторы для специфичных для проекта этапов обработки.
Ключевые возможности
- 01Функциональный Python API для создания пайплайнов обработки данных.
- 02Выполняет декодирование и аугментацию на CPU или GPU с масштабированием на несколько GPU.
- 03Читает данные во множестве форматов, включая JPEG, TFRecord, COCO, WAV и H.264.
- 04Может заменить загрузчики данных в TensorFlow, PyTorch, PaddlePaddle и JAX.
- 05Поддерживает пользовательские операторы и GPUDirect Storage.
Сценарии использования
- Устранение узкого места на CPU при обучении моделей классификации изображений или обнаружения объектов.
- Использование одного и того же пайплайна предобработки в проектах на TensorFlow и PyTorch.
- Запуск предобработки вместе с моделями на Triton Inference Server.
- Загрузка и аугментация аудио- или видеодатасетов для моделей речи и компьютерного зрения.
Быстрый старт
- 1
Установите последнюю версию DALI для CUDA 12 с помощью pip
pip install nvidia-dali-cuda120 - 2
Убедитесь, что установлены драйвер NVIDIA и CUDA Toolkit с поддержкой CUDA 12
- 3
Следуйте руководству Getting Started в документации
Что важно знать
- Требуется графический процессор NVIDIA с совместимыми драйвером и CUDA Toolkit.
- Для других способов установки и более старых версий CUDA обратитесь к отдельному руководству Installation Guide.
В подборках
Темы
Частые вопросы
- Что такое DALI?
- Библиотека с ускорением на GPU для загрузки и предобработки изображений, видео и аудио, ускоряющая обучение и инференс моделей глубокого обучения. NVIDIA DALI (Data Loading Library) переносит загрузку и предобработку данных для глубокого обучения с CPU на GPU. Пайплайны охватывают декодирование, кадрирование, изменение размера и другие аугментации изображений, видео и аудио, а встроенный движок выполнения отвечает за предварительную загрузку данных, параллельное выполнение и формирование батчей.
- DALI можно использовать бесплатно? Какая у него лицензия?
- DALI распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- DALI всё ещё поддерживается?
- Активно поддерживается. Последний коммит в DALI — 6 нояб. 2026 г..
- Для чего можно использовать DALI?
- Устранение узкого места на CPU при обучении моделей классификации изображений или обнаружения объектов. Использование одного и того же пайплайна предобработки в проектах на TensorFlow и PyTorch. Запуск предобработки вместе с моделями на Triton Inference Server. Загрузка и аугментация аудио- или видеодатасетов для моделей речи и компьютерного зрения.
- Как установить DALI?
- Установите последнюю версию DALI для CUDA 12 с помощью pip: pip install nvidia-dali-cuda120 Убедитесь, что установлены драйвер NVIDIA и CUDA Toolkit с поддержкой CUDA 12 Следуйте руководству Getting Started в документации
- Где скачать DALI?
- Исходный код и инструкции по установке — на GitHub: https://github.com/NVIDIA/DALI
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Библиотека Python для аугментации изображений в компьютерном зрении: пиксельные, пространственные, медицинские и 3D-преобразования с согласованной обработкой масок, рамок и ключевых точек.
Библиотека Python и CLI от Deezer, которая разделяет песни на вокал, ударные, бас и другие дорожки с помощью предварительно обученных моделей.
Библиотека Python для обработки аудио: чтение и запись файлов, встроенные эффекты и поддержка плагинов VST3 и Audio Unit.
Библиотека Python, которая преобразует датасеты в чанковый формат для быстрой потоковой загрузки из облачного или локального хранилища во время обучения моделей.
Высокопроизводительная система для работы с ML-моделями в production: версионируемые gRPC- и HTTP-endpoint’ы для инференса.