Как перевернуть помеченные направления по горизонтали в тензорном потоке - PullRequest
0 голосов
/ 21 мая 2018

Я дополняю данные для изучения в тензорном потоке, используя API набора данных, в котором дополнения записываются как часть графика.Для моего приложения мне нужно изменить ярлыки, а также изображения.Метки обозначают направление следующим образом:

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 Ответ

0 голосов
/ 21 мая 2018
flipped_label = tf.gather([0, 3, 2, 1], label)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...