Tensorflow / Keras 在通道上堆疊圖像

Question

想象一下以下設置：

• 圖像數據形狀：(302, 482, 3)
• 每個樣本的圖像：2
• 批量大小：100

我的數據加載器生成以下形狀：(100, 2, 302, 482, 3) - 目標是將通道軸上的兩個輸入圖像堆疊為 (100, 302, 482, 6)。

沒有批量維度（所以 x 有形狀 (2, 302, 482, 3) ）它很容易：

# x.shape = (2, 302, 482, 3)
stacked = tf.concat(x, axis=-1)
# stacked.shape = (302, 482, 6)

但是當添加批量維度時，我無法弄清楚執行相同的操作。

Answer 1

在我看來，最好的方法是在輸入網絡之前連接 2 個圖像（使用 numpy），為網絡提供維度 (302, 482, 6, )，除非您想在網絡中對其進行更高的處理。 這取決於你的目標。 編寫圖層時，批次大小無關緊要。 無論批量大小， tf.concat都將繼續進行。