Vectorised crop and resize

keras_cv/layers/preprocessing/random_crop_and_resize.py

@@ -1,4 +1,4 @@
-# Copyright 2022 The KerasCV Authors
+# Copyright 2023 The KerasCV Authors
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -101,6 +101,7 @@ def get_random_transformation_batch(
         new_heights = tf.clip_by_value(
             tf.sqrt(crop_area_factor / aspect_ratio), 0.0, 1.0
         )  # to avoid unwanted/unintuitive effects
+
         new_widths = tf.clip_by_value(
             tf.sqrt(crop_area_factor * aspect_ratio), 0.0, 1.0
         )
@@ -292,4 +293,12 @@ def get_config(self):
 
     @classmethod
     def from_config(cls, config):
+        if isinstance(config["crop_area_factor"], dict):
+            config["crop_area_factor"] = tf.keras.utils.deserialize_keras_object(
+                config["crop_area_factor"]
+            )
+        if isinstance(config["aspect_ratio_factor"], dict):
+            config["aspect_ratio_factor"] = tf.keras.utils.deserialize_keras_object(
+                config["aspect_ratio_factor"]
+            )
         return cls(**config)