将不同形状的Tensor对象传递给tf.train.batch

Question

我正在执行此操作：

images, labels, labels2 = tf.train.batch([image, lbl1, lbl2], batch_size=32);

其中image，lbl1和lbl2的类型为Tensor和Dimension(None)。

image是一个三维矩阵。 lbl1和lb2将代表浮点数组。

问题是，这些都是不同的大小，tf.train.batch要求Tensor对象具有已定义的形状。所以，当然，我收到如下错误：

ValueError: All shapes must be fully defined: [TensorShape([Dimension(None), Dimension(None), Dimension(3)]), TensorShape([Dimension(None)]), TensorShape([Dimension(None)])]

如果有某些不同尺寸的图像和标签，我该怎么做？

Answer 1

将dynamic_pad设置为True可以使用形状None的尺寸。

images, labels, labels2 = tf.train.batch([image, lbl1, lbl2], batch_size=32, dynamic_pad=True);