Как очистить информацию о форме tf.Tensor в TensorFlow? - PullRequest
0 голосов
/ 02 мая 2019

Предположим, во время обучения сети мы изменили размеры всех изображений до 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).

1 Ответ

0 голосов
/ 03 мая 2019

Я не верю, что вы можете изменить размер / масштабировать термины веса / смещения в нейронной сети. Но было бы довольно легко изменить размер входного изображения до 512 * 512. Вы рассматривали это?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...