Pipeline 層

[來源]

Pipeline 類別

keras.layers.Pipeline(layers, name=None)

將一系列層應用於輸入。

此類別適用於建立預處理 Pipeline,特別是圖像資料增強 Pipeline。與 Sequential 模型相比,Pipeline 具有一些重要的差異

  • 它不是 Model,只是一個普通的層。
  • 當 Pipeline 中的層與 tf.data 相容時,Pipeline 也將保持 tf.data 相容性。也就是說,在 tf.data 環境中,Pipeline 不會嘗試將其輸入轉換為後端原生張量(與 Sequential 模型不同)。

範例

from keras import layers
preprocessing_pipeline = layers.Pipeline([
    layers.AutoContrast(),
    layers.RandomZoom(0.2),
    layers.RandomRotation(0.2),
])

# `ds` is a tf.data.Dataset
preprocessed_ds = ds.map(
    preprocessing_pipeline,
    num_parallel_calls=4,
)