Написать пост

NVIDIA представила библиотеки DALI и nvJPEG для работы с изображениями

Аватар Сергей Штукатуров

DALI предназначена для обработки изображений и оптимизации входящих данных для фреймворков глубинного обучения, а nvJPEG позволяет ускорить загрузку файлов.

NVIDIA предоставила независимым разработчикам доступ к новым программным библиотекам. Одна из них — DALI — имеет открытый исходный код, предназначена для работы с такими платформами глубокого обучения, как MXNet, TensorFlow, и PyTorch, и служит для оптимизации потоков данных и загрузки изображений. Библиотека nvJPEG позволяет извлекать информацию из JPEG-файлов.

Основные особенности DALI

Компания предлагает использовать DALI для обработки обучающих данных перед их передачей непосредственно во фреймворк. Вычислительная нагрузка перенаправляется на графический процессор. Загрузить библиотеку можно с GitHub.

Среди характеристик DALI числятся:

  • простая интеграция с MXNet, TensorFlow и PyTorch за счёт использования соответствующих плагинов;
  • быстрое извлечение данных формата JPEG с использованием библиотеки nvJPEG;
  • настраиваемые потоки данных с регулируемыми графами и пользовательскими операторами;
  • обучающие процессы с поддержкой различных форматов данных: JPEG, RAW, LMDB, RecordIO и TFRecord.

Особенности nvJPEG

Основные преимущества nvJPEG:

  • распределение вычислительной нагрузки между центральным и графическим процессором;
  • возможность обработки как одиночных файлов, так и целых пакетов;
  • конверсия цветового пространства с поддержкой форматов RGB, BGR, RGBI, BGRI и YUV;
  • одно- и многофазное декодирование.

Исходный код nvJPEG компания оставила закрытым, однако сама библиотека доступна для бесплатного скачивания.

NVIDIA не в первый раз оказывает помощь независимым разработчикам, занимающимся вопросами машинного обучения. В 2017 году компания запустила облачную платформу NVIDIA GPU Cloud, оптимизированную для работы с подобными технологиями.

Следите за новыми постами
Следите за новыми постами по любимым темам
669 открытий671 показов