def preprocessing_fn(inputs):
"""Preprocess input columns into transformed columns."""
# Since we are modifying some features and leaving others unchanged, we
# start by setting `outputs` to a copy of `inputs.
outputs = inputs.copy()
# Convert the Image from Color to Grey Scale.
# NUMERIC_FEATURE_KEYS is the list of names of Columns of Values of Pixels
for key in NUMERIC_FEATURE_KEYS:
outputs[key] = tf.divide(outputs[key], 255)
outputs[LABEL_KEY] = inputs[LABEL_KEY]
return outputs
1条答案
按热度按时间mzsu5hc01#
是的,TFX也可以用于涉及图像的管道。
特别是,在与数据预处理相关的部分,据我所知,Tensorflow Transform中没有内置函数。
但转换可以使用Tensorflow Ops进行。例如,图像增强可以使用tf来完成。图像等等。
图像变换的示例代码,即。例如,通过将每个像素的值除以255,使用Tensorflow变换将图像从彩色转换为灰度,如下所示: