有没有办法将 tf.image.random_flip_left_right 方法用于批量图片?
感谢。
答案 0 :(得分:2)
当这些功能不支持批处理时,它总是很烦人。使用tf.map_fn
将批处理映射到函数上。虽然请注意,这会将您的问题分解为许多小型操作,这些操作在GPU上的效率不如random_flip_left_right
支持批处理操作时那样。
imgs = tf.map_fn(tf.image.random_flip_left_right, imgs_4d_tensor)