chore: review comments

keras-team · Jun 28, 2023 · 13514d2 · 13514d2
1 parent 1059373
commit 13514d2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/vision/oxford_pets_image_segmentation.py b/examples/vision/oxford_pets_image_segmentation.py
@@ -90,13 +90,13 @@
 format.
 """
 
-key_rename_fn = lambda inputs: {
+rescale_images_and_correct_masks = lambda inputs: {
     "images": tf.cast(inputs["image"], dtype=tf.float32) / 255.0,
     "segmentation_masks": inputs["segmentation_mask"] - 1,
 }
 
-train_ds = orig_train_ds.map(key_rename_fn, num_parallel_calls=AUTOTUNE)
-val_ds = orig_val_ds.map(key_rename_fn, num_parallel_calls=AUTOTUNE)
+train_ds = orig_train_ds.map(rescale_images_and_correct_masks, num_parallel_calls=AUTOTUNE)
+val_ds = orig_val_ds.map(rescale_images_and_correct_masks, num_parallel_calls=AUTOTUNE)
 
 """
 ## Utility Function