Дополните tf.data.Dataset изображением и маской. Оба должны быть дополнены ⇐ Python

[Anonymous]

У меня есть tf.data.Dataset путей изображения и масок

Код: Выделить всё

# 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))

сопоставление функции для преобразования путей в изображения

Код: Выделить всё

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

Как мне его расширить? Я не хочу использовать ImageDataGenerator, как говорит tf. Устарело: tf.keras.preprocessing.image.ImageDataGenerator не рекомендуется для нового кода.
Примечание: и то, и другое. Изображение и маску необходимо дополнить таким же образом.

Подробнее здесь: https://stackoverflow.com/questions/739 ... -augmented