使用 dl4j 图像到图像 DataSetIterator答案

【问题标题】：Image to image DataSetIterator using dl4j使用 dl4j 图像到图像 DataSetIterator
【发布时间】：2021-01-31 02:48:28
【问题描述】：

我想使用 DeepLearning4j 构建和训练 U-Net 网络。为此，我需要一个数据集迭代器，为网络提供输入图像和输出图像。

我是 DL4j 的新手，我发现的所有示例都说明了如何为图像分类构建数据集。

据我了解，我需要用我的文件实例化一个 ImageRecordReader，然后使用 RecordReaderDataSetIterator 类创建我的数据集迭代器。

问题是，我不知道如何在 ImageRecordReader 中指定我的标签也是图像。

谁能解释一下我该怎么做或指向一些相关的资源？

谢谢

【问题讨论】：

【解决方案1】：

您仍然需要一个 ImageRecordReader，但需要一个自定义标签生成器。如果您的数据集按以下方式组织，您可以自定义 PathLabelGenerator 类以通过替换文件路径中的文件夹名称来查找输出图像：

【讨论】：