tensorflow：批量可变大小的图像

Question

当传递给tf.train.batch时，似乎必须严格定义元素的形状，否则它将抱怨All shapes must be fully defined是否存在带有shape的张量Dimension(None)。那么，如何训练不同尺寸的图像呢？

Answer 1

您可以设置dynamic_pad=True的参数tf.train.batch。

dynamic_pad：布尔值。允许输入形状中的尺寸可变。给定的尺寸在出队时填充，以使一批中的张量具有相同的形状。