tensorflow 用TFX设计图像流水线

3htmauhk  于 2023-05-01  发布在  其他
关注(0)|答案(1)|浏览(133)

当阅读TFX文档时,尤其是与数据预处理相关的部分,我认为管道设计更适合分类特性。
我想知道TFX是否也可以用于涉及图像的管道。

mzsu5hc0

mzsu5hc01#

是的,TFX也可以用于涉及图像的管道。
特别是,在与数据预处理相关的部分,据我所知,Tensorflow Transform中没有内置函数。
但转换可以使用Tensorflow Ops进行。例如,图像增强可以使用tf来完成。图像等等。
图像变换的示例代码,即。例如,通过将每个像素的值除以255,使用Tensorflow变换将图像从彩色转换为灰度,如下所示:

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

相关问题