Erweitern Sie ein tf.data.Dataset um ein Bild und eine Maske. Beides muss erweitert werden
Posted: 03 Jan 2025, 10:38
Ich habe einen tf.data.Dataset mit Bildpfaden und Masken
Zuordnung einer Funktion, um Pfade in Bilder umzuwandeln
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.
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))
Code: Select all
#mapping the function
val_data = val_data.map(make_image)
Hinweis: Beides Bild und Maske müssen auf die gleiche Weise erweitert werden.