Numpy 4d 数组到 tf.data.dataset答案

【问题标题】：Numpy 4d array to tf.data.datasetNumpy 4d 数组到 tf.data.dataset
【发布时间】：2020-12-02 19:53:47
【问题描述】：

我正在关注本教程https://www.tensorflow.org/tutorials/generative/pix2pix，但我正在尝试制作自己的输入管道。我有一个 4d numpy 数组（Num samples、Height、Width、Channels），我使用ds = tf.data.Dataset.from_tensor_slices() 创建我的数据集。但是，当我调用 ds.take(1) 时，它没有批量大小的 a 维度。我可以通过在必要时插入 tf.expand_dims() 来解决此问题，但我觉得应该有一种方法可以在数据集中执行此操作。

【问题讨论】：

Tensorflow 提供了一个.batch() 方法，它接受一个数字作为批量大小。
欢迎来到 StackOverflow。另外，@RichardX，老实说，我在注意到您的评论之前就写了答案！ :)

标签： tensorflow tensorflow-datasets

【解决方案1】：

你可以试试：

for image in ds.batch(1).take(1):
    assert image.shape[0] == 1
    # do something with the image

【讨论】：