使用张量流完成训练后,我尝试测试一些学习网络。
但是我的测试图像是通道1的[512 512 1]数据(水平512和垂直512)。
我将图像数据更改为一个numpy数组。
张量网络应该是[? 512 512 1]看起来像这样。
如何将numpy数组转换为张量? ([512 512 1]-> [?512 512 1])
答案 0 :(得分:2)
您只需附加一个维度
arr = your_image # [512, 512, 1]
new_arr = np.expand_dims(arr, 0)
tensor = tf.convert_to_tensor(new_arr)
现在您可以使用feed dict或其他东西。