我有从tf记录中解码的imgs和标签。我想把它们想象出来。
为此,我必须将其从uint8转换为numpy。
<tf.Tensor: shape=(224, 224, 3), dtype=uint8, numpy=
array([[[0, 0, 0],
[0, 0, 0],
[0, 0, 0],
...,
[0, 0, 0],
[0, 0, 0],
[0, 0, 0]],
[0, 0, 0],
[0, 0, 0],
[0, 0, 0]]], dtype=uint8)>
这就是我打印imgs 时显示的内容
您只需运行以下命令即可。
imgs,labels = imgs.numpy(), labels.numpy()
它将imgs和标签转换为numpy。