Python tf.keras.applications.imagenet_utils.preprocess_input用法及代码示例

预处理编码一批图像的张量或 Numpy 数组。

用法

tf.keras.applications.imagenet_utils.preprocess_input(
    x, data_format=None, mode='caffe'
)

参数

x 具有 3 个颜色通道的浮点 numpy.array 或 tf.Tensor 、3D 或 4D，其值在 [0, 255] 范围内。如果数据类型兼容，则预处理数据将覆盖输入数据。为了避免这种行为，可以使用numpy.copy(x)。
data_format 图像张量/数组的可选数据格式。默认为无，在这种情况下使用全局设置tf.keras.backend.image_data_format()(除非您更改它，否则默认为"channels_last")。
mode
"caffe"、"tf" 或 "torch" 之一。默认为"caffe"。
- caffe：将图像从 RGB 转换为 BGR，然后将 zero-center 每个颜色通道相对于 ImageNet 数据集，不进行缩放。
- tf：将在 -1 和 1 之间缩放像素，sample-wise。
- torch：将在 0 和 1 之间缩放像素，然后将相对于 ImageNet 数据集的每个通道进行归一化。

预处理 numpy.array 或类型为 float32 的 tf.Tensor 。

抛出

ValueError 如果未知 mode 或 data_format 参数。

applications.MobileNet 的用法示例：

i = tf.keras.layers.Input([None, None, 3], dtype = tf.uint8)
x = tf.cast(i, tf.float32)
x = tf.keras.applications.mobilenet.preprocess_input(x)
core = tf.keras.applications.MobileNet()
x = core(x)
model = tf.keras.Model(inputs=[i], outputs=[x])

image = tf.image.decode_png(tf.io.read_file('file.png'))
result = model(image)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.applications.imagenet_utils.preprocess_input。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法

参数

返回

抛出