deep-learning - 在 deeplearning4j 中创建具有多个标签和未知数量类的 DataSet

Question

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

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

标签： deep-learningneural-networkdeeplearning4j

Answer 1

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

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

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