gpt4 book ai didi

python - TensorFlow "tf.image"在 4D 图像批处理上运行

转载 作者:太空狗 更新时间:2023-10-30 01:37:13 24 4
gpt4 key购买 nike

<分区>

我想在 TensorFlow 中使用这个函数,但是它在 3D 张量而不是 4D 张量上运行:我有一个 batch_size 的外部维度。

tf.image.random_flip_left_right(input_image_data)

也就是说,这个函数需要一个形状的张量(图像):

(width, height, channels)

但我有多个图像,例如:

(batch_size, width, height, channels)

我如何将随机翻转函数映射到批量大小的每个图像,并获得具有与我已有的相同 4D 形状的张量作为输出?

我的猜测是它需要在函数入口处进行整形并在函数后进行整形,但我不确定这是否会破坏数据结构并在应用镜像时将批处理中的图像混合在一起.此外,这种方法将对整个批处理进行单一随机化,而不是对每张图像进行随机化。

任何建议表示赞赏!

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com