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 будет действовать одинаково независимо от размера пакета.