Предположим, во время обучения сети мы изменили размеры всех изображений до 512 * 512, поэтому может быть tf.Tensor
с именем input:0
, который имеет форму (batch_size, 512, 512, 3)
.
Однако, делая прогнозы, можно подавать изображения разных размеров в сеть. Таким образом, форма тензора input:0
должна выглядеть примерно так: (batch_size, None, None, 3)
, поскольку размер изображений неизвестен.
Итак, если у меня есть Тензор формы (batch_size, 512, 512, 3)
, как мне "изменить" его значение до (batch_size, None, None, 3)
? Я пытался
inputs=tf.reshape(inputs, (batch_size, tf.shape(inputs)[1], tf.shape(inputs)[2], 3)
но вывод все еще имеет форму (batch_size, 512, 512, 3)
.