使用tf.image.random_flip_left_right方法批处理图像

时间:2018-03-26 17:35:09

标签: tensorflow

有没有办法将 tf.image.random_flip_left_right 方法用于批量图片?

感谢。

1 个答案:

答案 0 :(得分:2)

当这些功能不支持批处理时,它总是很烦人。使用tf.map_fn将批处理映射到函数上。虽然请注意,这会将您的问题分解为许多小型操作,这些操作在GPU上的效率不如random_flip_left_right支持批处理操作时那样。

imgs = tf.map_fn(tf.image.random_flip_left_right, imgs_4d_tensor)