【问题标题】:How do I turn a numpy array into a tensor in "Tensorflow"?如何在“Tensorflow”中将 numpy 数组转换为张量?
【发布时间】:2018-11-09 09:29:15
【问题描述】:

我在使用 tensorflow 完成训练后尝试测试一些学习网络。

但我的测试图像是通道 1 的 [512 512 1] 数据,水平为 512 个像素,垂直为 512 个像素。

我将图像数据更改为 numpy 数组。

张量网络应该是 [? 512 512 1] 看起来像这样。

如何将 numpy 数组转换为张量? ([512 512 1] -> [?512 512 1])

【问题讨论】:

  • ? 的前导维度是什么?
  • 一条消息说... ValueError:无法为张量“images:0”提供形状(512、512、1)的值,其形状为“(?、512、512、1)”
  • 是的,但是? 维度是什么?批量大小?

标签: numpy tensorflow


【解决方案1】:

你只需要附加一个维度

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

tensor = tf.convert_to_tensor(new_arr)

现在你可以使用 feed dict 或其他东西了。

【讨论】:

  • 非常非常非常感谢你!
猜你喜欢
  • 2020-12-31
  • 2016-03-09
  • 2021-10-13
  • 2021-04-28
  • 1970-01-01
  • 2021-01-08
  • 1970-01-01
  • 2021-10-17
  • 2018-11-28
相关资源
最近更新 更多