在 deeplearning4j 中创建具有多个标签和未知数量类的数据集

Question

我应该使用什么 DataSetIterator 来创建包含多个特征和标签的 DataSet 对象？我只见过类似于 'Iris example' 的例子，其中只有一个标签，并且知道有多少个不同的标签。在我的问题中，有四个标签（位置 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;

我知道这个问题看起来很基础，但我真的很难在官方教程或文档中找到关于这个主题的任何信息。

Answer 1

您似乎正在寻找一种对象检测类型的数据，其中包含图片中的多个可能对象的边界框。

看看这个例子：https://github.com/eclipse/deeplearning4j-examples/blob/master/dl4j-examples/src/main/java/org/deeplearning4j/examples/convolution/objectdetection/HouseNumberDetection.java

一般来说，有一个 MultiDataSet 可以接受多个输入，也可以有多个输出。

在 deeplearning4j 中创建具有多个标签和未知数量类的数据集

Create a DataSet with multiple labels and unknown number of classes in deeplearning4j

neural-network

deep-learning

deeplearning4j

在 deeplearning4j 中创建具有多个标签和未知数量 类 的数据集

Create a DataSet with multiple labels and unknown number of classes in deeplearning4j

neural-network

deep-learning

deeplearning4j

在 deeplearning4j 中创建具有多个标签和未知数量类的数据集