tensorflow:可变尺寸图像的批量

问题描述:

当传递给tf.train.batch时,它看起来像元素的形状必须严格定义,否则它会抱怨All shapes must be fully defined如果存在形状为Dimension(None)的张量。那么,如何在不同尺寸的图像上进行一次训练呢?tensorflow:可变尺寸图像的批量

您可以在参数tf.train.batch中设置dynamic_pad=True

dynamic_pad:Boolean。在输入形状中允许可变尺寸。给定的尺寸在出列时填充,以便批次内的张量具有相同的形状。

+0

好吧。这工作。雅知道,我一直在寻找3天的答案。 :d – haxtar

通常,图像被调整至一特定数目的像素。

根据你的任务,你可能能够为了处理不同大小的图像,使用其他技术。例如,对于人脸识别和OCR,使用固定大小的窗口,然后将其移到图像上。在其他任务中,具有汇聚层或递归神经网络的卷积神经网络可能会有所帮助。