首页 文章

如何在tensorflow中水平翻转标记的方向

提问于
浏览
0

我正在使用数据集API扩充数据以在tensorflow中学习,其中扩充被写为图的一部分 . 对于我的应用程序,我需要修改标签和图像 . 标签将方向编码为:

0: up
1: right
2: down
3: left

对于旋转增强,我可以做:

rotated_image = tf.image.rot90(image, 1)
rotated_label = (label + 1) % 4

我可以使用以下方法水平翻转特征图像:

hflipped_image = tf.image.flip_left_right(image)

但我无法弄清楚如何左右翻转标签< - > .

如果标签为1则应为3,反之亦然 . 0和2应保持不变 . 我怎样才能做到这一点?

1 回答

  • 1
    flipped_label = tf.gather([0, 3, 2, 1], label)
    

相关问题