diff --git a/keras_hub/src/models/preprocessor.py b/keras_hub/src/models/preprocessor.py index a53955640d..f338b45339 100644 --- a/keras_hub/src/models/preprocessor.py +++ b/keras_hub/src/models/preprocessor.py @@ -71,6 +71,22 @@ def image_converter(self): def image_converter(self, value): self._image_converter = value + @property + def image_size(self): + """Shortcut to get/set the image size of the image converter.""" + if self.image_converter is None: + return None + return self.image_converter.image_size + + @image_size.setter + def image_size(self, value): + if self.image_converter is None: + raise ValueError( + "Cannot set `image_size` on preprocessor if `image_converter` " + " is `None`." + ) + self.image_converter.image_size = value + def get_config(self): config = super().get_config() if self.tokenizer: diff --git a/keras_hub/src/models/task.py b/keras_hub/src/models/task.py index 5bb954beab..af12f1cb1c 100644 --- a/keras_hub/src/models/task.py +++ b/keras_hub/src/models/task.py @@ -280,7 +280,7 @@ def summary( def highlight_number(x): if x is None: - f"[color(45)]{x}[/]" + return f"[color(45)]{x}[/]" return f"[color(34)]{x:,}[/]" # Format number with commas. def highlight_symbol(x): @@ -339,7 +339,10 @@ def add_layer(layer, info): add_layer(layer, info) elif isinstance(layer, ImageConverter): info = "Image size: " - info += highlight_shape(layer.image_size) + image_size = layer.image_size + if image_size is None: + image_size = (None, None) + info += highlight_shape(image_size) add_layer(layer, info) elif isinstance(layer, AudioConverter): info = "Audio shape: "