Pipeline
類別keras.layers.Pipeline(layers, name=None)
將一系列層應用於輸入。
此類別適用於建立預處理 Pipeline,特別是圖像資料增強 Pipeline。與 Sequential
模型相比,Pipeline
具有一些重要的差異
Model
,只是一個普通的層。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,
)