比较与tf.image.convert_image_dtype的功能差异

查看源文件

tf.image.convert_image_dtype

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

更多内容详见tf.image.convert_image_dtype

mindspore.dataset.transforms.TypeCast

class mindspore.dataset.transforms.TypeCast(
    output_type
)

更多内容详见mindspore.dataset.transforms.TypeCast

使用方式

TensorFlow:转换Tensor格式图像的数据类型,支持设置是否在转换前进行数值裁切避免溢出。

MindSpore:转换numpy.ndarray格式图像的数据类型。

代码示例

# The following implements TypeCast with MindSpore.
import numpy as np
import mindspore.dataset as ds

image = np.random.random((28, 28, 3))
result = ds.transforms.TypeCast(np.uint8)(image)
print(result.dtype)
# uint8

# The following implements convert_image_dtype with TensorFlow.
import tensorflow as tf

image = tf.random.normal((28, 28, 3), dtype=tf.float32)
result = tf.image.convert_image_dtype(image, tf.uint8)
print(result.dtype)
# uint8