Создайте DataSet с несколькими метками и неизвестным количеством классов в deeplearning4j

Question

Какой DataSetIterator следует использовать для создания объекта DataSet, который содержит множество функций и меток? Я видел только примеры, похожие на «Пример Iris», где есть только одна метка, и известно, сколько существует разных меток. В моей задаче есть четыре метки (позиция X, позиция Y, ширина и высота фигуры) и много объектов (значения в пикселях), и невозможно рассчитать, сколько разных меток может быть.

Я хочу что-то как это

RecordReader recordReader = new CSVRecordReader(0, ',');
recordReader.initialize(new FileSplit(new File(fileName)));

DataSetIterator iterator = new CustomDataSetIterator(recordReader, numRows, numFeatures, numLables);
DataSet allData = iterator.next();

Использование данных, которые выглядят следующим образом

feature0;feature1;feature2;feature3;label0;label1;

Я знаю, что этот вопрос кажется очень простым c, но мне действительно было трудно найти какую-либо информацию о это топи c в официальных руководствах или в документации.

Answer 1

похоже, что вы ищете данные типа обнаружения объекта с ограничивающими рамками для нескольких возможных объектов на вашем изображении.

посмотрите на этот пример для этого: https://github.com/eclipse/deeplearning4j-examples/blob/master/dl4j-examples/src/main/java/org/deeplearning4j/examples/convolution/objectdetection/HouseNumberDetection.java

в общем случае существует MultiDataSet, который может принимать несколько входов и иметь несколько выходов.