0

當傳遞給tf.train.batch時,它看起來像元素的形狀必須嚴格定義,否則它會抱怨All shapes must be fully defined如果存在形狀爲Dimension(None)的張量。那麼,如何在不同尺寸的圖像上進行一次訓練呢?tensorflow:可變尺寸圖像的批量

回答

3

您可以在參數tf.train.batch中設置dynamic_pad=True

dynamic_pad:Boolean。在輸入形狀中允許可變尺寸。給定的尺寸在出列時填充,以便批次內的張量具有相同的形狀。

+0

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

2

通常,圖像被調整至一特定數目的像素。

根據你的任務,你可能能夠爲了處理不同大小的圖像,使用其他技術。例如,對於人臉識別和OCR,使用固定大小的窗口,然後將其移到圖像上。在其他任務中,具有匯聚層或遞歸神經網絡的卷積神經網絡可能會有所幫助。