tf.image.convert_image_dtype

Преобразует image в dtype, масштабируя значения при необходимости.

tf.image.convert_image_dtype(
    image, dtype, saturate=False, name=None
)

Ожидается, что изображения, представленные с помощью чисел с плавающей точкой, будут иметь значения в диапазоне [0,1). Ожидается, что данные изображений, хранящиеся в целочисленных типах данных, будут иметь значения в диапазоне [0,MAX], где MAX — наибольшее положительное представимое число для типа данных.

Этот оператор выполняет преобразование между типами данных, масштабируя значения соответствующим образом перед приведением типа.

Обратите внимание, что преобразование от входных данных с плавающей точкой к целочисленным типам может привести к проблемам переполнения/подпотока. Установите saturate в True чтобы избежать таких проблем при проблемных преобразованиях. Если включено, насыщение ограничит выходные данные в разрешенном диапазоне перед выполнением потенциально опасного приведения типа (и только перед выполнением такого приведения, то есть при приведении от чисел с плавающей точкой к целочисленному типу, а также при приведении от знакового к беззнаковому типу; saturate не оказывает никакого влияния на преобразования между числами с плавающей точкой или на преобразования, увеличивающие диапазон типа).

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

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.convert_image_dtype(x, dtype=tf.float16, saturate=False)
<tf.Tensor: shape=(2, 2, 3), dtype=float16, numpy=
array([[[ 1.,  2.,  3.],
        [ 4.,  5.,  6.]],
       [[ 7.,  8.,  9.],
        [10., 11., 12.]]], dtype=float16)>