tf.image.adjust_contrast

Регулировка контраста RGB или градации серого изображений.

tf.image.adjust_contrast(
    images, contrast_factor
)

Это удобный метод, который преобразует RGB-изображения в представление с плавающей запятой, регулирует их контраст и затем преобразует их обратно в исходный тип данных. Если несколько корректировок выполняются последовательно, рекомендуется минимизировать количество избыточных преобразований.

images — это тензор с не менее чем 3 измерениями. Последние 3 измерения интерпретируются как [height, width, channels]. Другие измерения представляют собой лишь набор изображений, например, [batch, height, width, channels].

Контраст регулируется независимо для каждого канала каждого изображения.

Для каждого канала этот оператор вычисляет среднее значение пикселей изображения в канале, а затем корректирует каждый компонент x каждого пикселя до (x - mean) * contrast_factor + mean.

Пример использования:

x = [[[1.0, 2.0, 3.0],
      [4.0, 5.0, 6.0]],
    [[7.0, 8.0, 9.0],
      [10.0, 11.0, 12.0]]]
tf.image.adjust_contrast(x, 2)
<tf.Tensor: shape=(2, 2, 3), dtype=float32, numpy=
array([[[-3.5, -2.5, -1.5],
        [ 2.5,  3.5,  4.5]],
       [[ 8.5,  9.5, 10.5],
        [14.5, 15.5, 16.5]]], dtype=float32)>