Создать LMDB для набора данных изображений с помощью k-hot меток

Question

Я хочу создать классификатор для набора данных изображения, в котором каждое изображение входит в несколько классов из всех классов, поэтому целевые значения представляют собой k-горячие векторы.Теперь я создаю текстовый файл, который содержит адрес, если файл изображения и пространство, и вектор k-hot в каждой строке, но когда я пытаюсь запустить сценарии для создания файлов lmdb, это вызывает ошибки, которые не могут открыть или найти файлы.Я пробую тот же процесс с теми же данными и просто числом, что и метка класса, и все идет хорошо.Поэтому я думаю, что он не может правильно проанализировать файл .txt, когда метки являются векторами.

Любое предложение ... Спасибо

Answer 1

Caffe "Data" слоев и сценарий convert_imageset были написаны с очень конкретным примером использования: классификация изображений.Следовательно, базовый элемент, сохраняемый (и извлекаемый из) LMDB с помощью caffe, равен Datum, в котором есть место для одного целого числа метки .
_{Более длительное обсуждение этой темы можно посмотреть здесь}

Это не означает, что Caffe не может облегчить различные типы ввода / задач.
Вместо этого вы можете использовать слой "HDF5Data".Когда дело доходит до ввода hdf5, caffe практически не имеет ограничений по форме и размеру ввода.
См., Например, этот ответ и этот для получения более подробной информации о том, как на самом деле сделатьэто работает.