Создание обучения/проверки LMDB для NVIDIA Цифры в Python

Question

Я пытаюсь сделать LMDB для обучения/проверки, установленным для использования с цифрами NVIDIA, но я не могу найти хороших примеров/учебников.

Я понимаю, как создать базу данных LMDB, но я не уверен, как правильно форматировать данные. Я получаю, как создать изображение с помощью caffe_pb2 Datum, установив каналы/ширину/высоту/данные и сохраните их.

Но как создать ярлыки LMDB? Я все еще использую Caffe Datum? Если да, то каким образом я устанавливаю каналы/ширину/высоту? Будет ли это работать, если у меня есть единая метка значения?

Благодаря

Answer 1

ЦИФРЫ только действительно поддерживает данные в LMDBs на данный момент. Каждое значение в LMDB ключ/валь магазин должен быть Caffe Datum, что ограничивает число измерений до 3.

Даже если Caffe Геодезические позволяют одной числовой метки (datum.label), при загрузке скомпилированную LMDB к цифрам, которые нужно для указания отдельной базы данных для этикеток. Это неэффективно, если у вас есть только одна цифровая метка (поскольку вы могли бы сделать все это в одном БД), но она более универсальна и масштабируема для других типов меток.

Извините, вы правы, что это не задокументировано очень хорошо прямо сейчас. Вот некоторые исходные файлы можно просматривать на вдохе, если вы так склонны:

• Data are images, labels are (gradientX, gradientY)
• Data are image pairs, labels are 1 or 0
• Data are text snippets, labels are numeric classes
• Generic script for creating LMDBs from any data blobs coming from extensions/plugins