Конвертировать все пиксели изображения с 0 значениями в 255 в тензорном потоке

Question

Я новичок в tenorflow,

Мне нужно преобразовать все пиксели с 0 значениями в 255 для тензора трехмерного изображения

Я могу сделать это в cv, как это

img[np.where((img == [0, 0, 0]).all(axis=2))] = [255, 255, 255]

Но мне нужна помощь, чтобы узнать, как это сделать в tenorflow. Как мне нужно сделать это в графе тензорного потока

Answer 1

Вы можете сделать:

img = tf.cast(tf.equal(img, tf.zeros_like(img)), tf.int32)*255 + img