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)>