时间戳

首页 > 解决方案 > 使用 dl4j 图像到图像 DataSetIterator

deeplearning4j - 使用 dl4j 图像到图像 DataSetIterator

问题描述

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

我是 DL4j 的新手，我找到的所有示例都解释了如何为图像分类构建数据集。

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

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

有人可以解释我该怎么做或指向一些相关的资源吗？

谢谢

标签： deeplearning4junity3d-unetsemantic-segmentationdl4jrecordreader

解决方案

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

  • 数据目录/输入/img.png
  • 数据目录/输出/img.png

推荐阅读