Erweitern Sie ein tf.data.Dataset um ein Bild und eine Maske. Beides muss erweitert werden ⇐ Python

[Guest]

Ich habe einen tf.data.Dataset mit Bildpfaden und Masken

Code: Select all

# Creating list of image and mask path
all_val_img = np.array(sorted([os.path.join(VAL_DIR,i) for i in os.listdir(VAL_DIR)]))
all_val_mask = np.array(sorted([os.path.join(VAL_MASK_DIR,i) for i in os.listdir(VAL_MASK_DIR)]))

# doing tf.data.Dataset
val_data = tf.data.Dataset.from_tensor_slices((all_val_img, all_val_mask))

Zuordnung einer Funktion, um Pfade in Bilder umzuwandeln

Code: Select all

#mapping the function
val_data = val_data.map(make_image)

Wie kann ich es nun erweitern? Ich möchte ImageDataGenerator nicht verwenden, wie tf sagt. Veraltet: tf.keras.preprocessing.image.ImageDataGenerator wird für neuen Code nicht empfohlen.
Hinweis: Beides Bild und Maske müssen auf die gleiche Weise erweitert werden.