我正在使用数据集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 回答