Tensorflow加载具有图像标签的图像数据集

Question

我试图找到一种简单的方法将数据放入TensorFlow数据集中，而不必事先加载并将其作为NumPy数组进行处理。

在这种情况下，我正在使用细分模型，并且我的数据经过结构化，因此一个目录包含训练数据图像，而另一个目录则包含“掩码”，本质上也包含图像。

使用tf.keras.preprocessing.image_dataset_from_directory，我可以加载图像数据集，它将设置标签作为目录的名称，或者让我自己通过函数参数设置它们，但不会让我设置参数，因此它从其他目录获取标签。我正在阅读文档，但看不到一种简单的方法来加载这些类型的数据集，而标签本身就是图像。

Answer 1

这是在Keras中实现此目标的一种可能方法：

busy_handler

这里重要的是设置相同的种子，以便图像和标签匹配。 我从我的一个较旧的项目中复制了它，因此它可能不是最新的。