问题描述
我想使用DeepLearning4j来构建和训练U-Net网络。为此，我需要一个数据集迭代器，该迭代器向网络提供输入图像和输出图像。
我是DL4j的新手，我发现的所有示例都说明了如何构建图像分类数据集。
据我了解，我需要实例化一个ImageRecordReader文件，然后使用RecordReaderDataSetIterator类创建数据集迭代器。
问题是，我不知道如何在ImageRecordReader中指定我的标签也是图像。
有人可以解释我该怎么做，还是可以向我介绍一些新的资源？
谢谢
解决方法
您仍将需要一个ImageRecordReader，但需要一个自定义标签生成器。如果您的数据集是按以下方式组织的，则可以通过替换文件路径中的文件夹名称来自定义 PathLabelGenerator 类以查找输出图像：
- dataDir / 输入 /img.png
- dataDir / 输出 /img.png