使用 dl4j 图像到图像 DataSetIterator
Posted
技术标签:
【中文标题】使用 dl4j 图像到图像 DataSetIterator【英文标题】：Image to image DataSetIterator using dl4j 【发布时间】：2021-01-31 02:48:28 【问题描述】：我想使用 DeepLearning4j 构建和训练 U-Net 网络。为此，我需要一个数据集迭代器，为网络提供输入图像和输出图像。
我是 DL4j 的新手，我发现的所有示例都说明了如何为图像分类构建数据集。
据我了解，我需要用我的文件实例化一个 ImageRecordReader，然后使用 RecordReaderDataSetIterator 类创建我的数据集迭代器。
问题是，我不知道如何在 ImageRecordReader 中指定我的标签也是图像。
谁能解释一下我该怎么做或指向一些相关的资源？
谢谢
【问题讨论】：
【参考方案1】：您仍然需要一个 ImageRecordReader，但需要一个自定义标签生成器。如果您的数据集按以下方式组织，您可以自定义 PathLabelGenerator 类以通过替换文件路径中的文件夹名称来查找输出图像：
dataDir/输入/img.png dataDir/输出/img.png【讨论】：
以上是关于使用 dl4j 图像到图像 DataSetIterator的主要内容，如果未能解决你的问题，请参考以下文章