Change Keras preprocessing APIs to stable in the Image segmentation tutorial

PiperOrigin-RevId: 405699120

Author: tensorflower-gardener

site/en/tutorials/images/segmentation.ipynb

@@ -109,7 +109,6 @@
       "outputs": [],
       "source": [
         "import tensorflow as tf\n",
-        "from tensorflow.keras.layers.experimental import preprocessing \n",
         "\n",
         "import tensorflow_datasets as tfds"
       ]
@@ -247,8 +246,8 @@
         "  def __init__(self, seed=42):\n",
         "    super().__init__()\n",
         "    # both use the same seed, so they'll make the same random changes.\n",
-        "    self.augment_inputs = preprocessing.RandomFlip(mode=\"horizontal\", seed=seed)\n",
-        "    self.augment_labels = preprocessing.RandomFlip(mode=\"horizontal\", seed=seed)\n",
+        "    self.augment_inputs = tf.keras.layers.RandomFlip(mode=\"horizontal\", seed=seed)\n",
+        "    self.augment_labels = tf.keras.layers.RandomFlip(mode=\"horizontal\", seed=seed)\n",
         "  \n",
         "  def call(self, inputs, labels):\n",
         "    inputs = self.augment_inputs(inputs)\n",
@@ -310,7 +309,7 @@
         "  for i in range(len(display_list)):\n",
         "    plt.subplot(1, len(display_list), i+1)\n",
         "    plt.title(title[i])\n",
-        "    plt.imshow(tf.keras.preprocessing.image.array_to_img(display_list[i]))\n",
+        "    plt.imshow(tf.keras.utils.array_to_img(display_list[i]))\n",
         "    plt.axis('off')\n",
         "  plt.show()"
       ]