Как превратить массив Numpy в тензор в «Tensorflow»?

Question

Я пытался протестировать некоторые обучающие сети после завершения тренировки с тензорным потоком.

Но мое тестовое изображение представляет собой [512 512 1] данные канала 1 в 512 горизонтальных и 512 вертикальных пикселях.

Я изменил данные изображения на пустой массив.

Тензорная сеть должна быть [?512 512 1] Это выглядит следующим образом.

Как преобразовать массив с нулевыми значениями в тензор?([512 512 1] -> [? 512 512 1])

Answer 1

Вы просто должны добавить одно измерение

arr = your_image # [512, 512, 1]
new_arr = np.expand_dims(arr, 0)

tensor = tf.convert_to_tensor(new_arr)

Теперь вы можете использовать команду dict или что-то еще.