keras-hub-nightly 0.21.0.dev202504040358__tar.gz → 0.21.0.dev202504052235__tar.gz

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-hub-nightly
-Version: 0.21.0.dev202504040358
+Version: 0.21.0.dev202504052235
 Summary: Industry-strength Natural Language Processing extensions for Keras.
 Home-page: https://github.com/keras-team/keras-hub
 Author: Keras team

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/keras_hub/api/layers/__init__.py

@@ -84,3 +84,6 @@ from keras_hub.src.models.vit.vit_image_converter import ViTImageConverter
 from keras_hub.src.models.whisper.whisper_audio_converter import (
     WhisperAudioConverter,
 )
+from keras_hub.src.models.xception.xception_image_converter import (
+    XceptionImageConverter,
+)

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/keras_hub/api/models/__init__.py

@@ -388,6 +388,13 @@ from keras_hub.src.models.vit.vit_image_classifier_preprocessor import (
 from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone
 from keras_hub.src.models.whisper.whisper_backbone import WhisperBackbone
 from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
+from keras_hub.src.models.xception.xception_backbone import XceptionBackbone
+from keras_hub.src.models.xception.xception_image_classifier import (
+    XceptionImageClassifier,
+)
+from keras_hub.src.models.xception.xception_image_classifier_preprocessor import (
+    XceptionImageClassifierPreprocessor,
+)
 from keras_hub.src.models.xlm_roberta.xlm_roberta_backbone import (
     XLMRobertaBackbone,
 )

keras_hub_nightly-0.21.0.dev202504052235/keras_hub/src/models/xception/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.xception.xception_backbone import XceptionBackbone
+from keras_hub.src.models.xception.xception_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, XceptionBackbone)

keras_hub_nightly-0.21.0.dev202504052235/keras_hub/src/models/xception/xception_backbone.py

@@ -0,0 +1,188 @@
+import functools
+
+from keras import layers
+
+from keras_hub.src.api_export import keras_hub_export
+from keras_hub.src.models.backbone import Backbone
+from keras_hub.src.utils.keras_utils import standardize_data_format
+
+
+@keras_hub_export("keras_hub.models.XceptionBackbone")
+class XceptionBackbone(Backbone):
+    """Xception core network with hyperparameters.
+
+    This class implements a Xception backbone as described in
+    [Xception: Deep Learning with Depthwise Separable Convolutions](https://arxiv.org/abs/1610.02357).
+
+    Most users will want the pretrained presets available with this model. If
+    you are creating a custom backbone, this model provides customizability
+    through the `stackwise_conv_filters` and `stackwise_pooling` arguments. This
+    backbone assumes the same basic structure as the original Xception mode:
+    * Residuals and pre-activation everywhere but the first and last block.
+    * Conv layers for the first block only, separable conv layers elsewhere.
+
+    Args:
+        stackwise_conv_filters: list of list of ints. Each outermost list
+            entry represents a block, and each innermost list entry a conv
+            layer. The integer value specifies the number of filters for the
+            conv layer.
+        stackwise_pooling: list of bools. A list of booleans per block, where
+            each entry is true if the block should includes a max pooling layer
+            and false if it should not.
+        image_shape: tuple. The input shape without the batch size.
+            Defaults to `(None, None, 3)`.
+        data_format: `None` or str. If specified, either `"channels_last"` or
+            `"channels_first"`. If unspecified, the Keras default will be used.
+        dtype: `None` or str or `keras.mixed_precision.DTypePolicy`. The dtype
+            to use for the model's computations and weights.
+
+    Examples:
+    ```python
+    input_data = np.random.uniform(0, 1, size=(2, 224, 224, 3))
+
+    # Pretrained Xception backbone.
+    model = keras_hub.models.Backbone.from_preset("xception_41_imagenet")
+    model(input_data)
+
+    # Randomly initialized Xception backbone with a custom config.
+    model = keras_hub.models.XceptionBackbone(
+        stackwise_conv_filters=[[32, 64], [64, 128], [256, 256]],
+        stackwise_pooling=[True, True, False],
+    )
+    model(input_data)
+    ```
+    """
+
+    def __init__(
+        self,
+        stackwise_conv_filters,
+        stackwise_pooling,
+        image_shape=(None, None, 3),
+        data_format=None,
+        dtype=None,
+        **kwargs,
+    ):
+        if len(stackwise_conv_filters) != len(stackwise_pooling):
+            raise ValueError("All stackwise args should have the same length.")
+
+        data_format = standardize_data_format(data_format)
+        channel_axis = -1 if data_format == "channels_last" else 1
+        num_blocks = len(stackwise_conv_filters)
+
+        # Layer shorcuts with common args.
+        norm = functools.partial(
+            layers.BatchNormalization,
+            axis=channel_axis,
+            dtype=dtype,
+        )
+        act = functools.partial(
+            layers.Activation,
+            activation="relu",
+            dtype=dtype,
+        )
+        conv = functools.partial(
+            layers.Conv2D,
+            kernel_size=(3, 3),
+            use_bias=False,
+            data_format=data_format,
+            dtype=dtype,
+        )
+        sep_conv = functools.partial(
+            layers.SeparableConv2D,
+            kernel_size=(3, 3),
+            padding="same",
+            use_bias=False,
+            data_format=data_format,
+            dtype=dtype,
+        )
+        point_conv = functools.partial(
+            layers.Conv2D,
+            kernel_size=(1, 1),
+            strides=(2, 2),
+            padding="same",
+            use_bias=False,
+            data_format=data_format,
+            dtype=dtype,
+        )
+        pool = functools.partial(
+            layers.MaxPool2D,
+            pool_size=(3, 3),
+            strides=(2, 2),
+            padding="same",
+            data_format=data_format,
+            dtype=dtype,
+        )
+
+        # === Functional Model ===
+        image_input = layers.Input(shape=image_shape)
+        x = image_input  # Intermediate result.
+
+        # Iterate through the blocks.
+        for block_i in range(num_blocks):
+            first_block, last_block = block_i == 0, block_i == num_blocks - 1
+            block_filters = stackwise_conv_filters[block_i]
+            use_pooling = stackwise_pooling[block_i]
+
+            # Save the block input as a residual.
+            residual = x
+            for conv_i, filters in enumerate(block_filters):
+                # First block has post activation and strides on first conv.
+                if first_block:
+                    prefix = f"block{block_i + 1}_conv{conv_i + 1}"
+                    strides = (2, 2) if conv_i == 0 else (1, 1)
+                    x = conv(filters, strides=strides, name=prefix)(x)
+                    x = norm(name=f"{prefix}_bn")(x)
+                    x = act(name=f"{prefix}_act")(x)
+                # Last block has post activation.
+                elif last_block:
+                    prefix = f"block{block_i + 1}_sepconv{conv_i + 1}"
+                    x = sep_conv(filters, name=prefix)(x)
+                    x = norm(name=f"{prefix}_bn")(x)
+                    x = act(name=f"{prefix}_act")(x)
+                else:
+                    prefix = f"block{block_i + 1}_sepconv{conv_i + 1}"
+                    # The first conv in second block has no activation.
+                    if block_i != 1 or conv_i != 0:
+                        x = act(name=f"{prefix}_act")(x)
+                    x = sep_conv(filters, name=prefix)(x)
+                    x = norm(name=f"{prefix}_bn")(x)
+
+            # Optional block pooling.
+            if use_pooling:
+                x = pool(name=f"block{block_i + 1}_pool")(x)
+
+            # Sum residual, first and last block do not have a residual.
+            if not first_block and not last_block:
+                prefix = f"block{block_i + 1}_residual"
+                filters = x.shape[channel_axis]
+                # Match filters with a pointwise conv if needed.
+                if filters != residual.shape[channel_axis]:
+                    residual = point_conv(filters, name=f"{prefix}_conv")(
+                        residual
+                    )
+                    residual = norm(name=f"{prefix}_bn")(residual)
+                x = layers.Add(name=f"{prefix}_add", dtype=dtype)([x, residual])
+
+        super().__init__(
+            inputs=image_input,
+            outputs=x,
+            dtype=dtype,
+            **kwargs,
+        )
+
+        # === Config ===
+        self.stackwise_conv_filters = stackwise_conv_filters
+        self.stackwise_pooling = stackwise_pooling
+        self.image_shape = image_shape
+        self.data_format = data_format
+
+    def get_config(self):
+        config = super().get_config()
+        config.update(
+            {
+                "stackwise_conv_filters": self.stackwise_conv_filters,
+                "stackwise_pooling": self.stackwise_pooling,
+                "image_shape": self.image_shape,
+            }
+        )
+        return config

keras_hub_nightly-0.21.0.dev202504052235/keras_hub/src/models/xception/xception_image_classifier.py

@@ -0,0 +1,12 @@
+from keras_hub.src.api_export import keras_hub_export
+from keras_hub.src.models.image_classifier import ImageClassifier
+from keras_hub.src.models.xception.xception_backbone import XceptionBackbone
+from keras_hub.src.models.xception.xception_image_classifier_preprocessor import (  # noqa: E501
+    XceptionImageClassifierPreprocessor,
+)
+
+
+@keras_hub_export("keras_hub.models.XceptionImageClassifier")
+class XceptionImageClassifier(ImageClassifier):
+    backbone_cls = XceptionBackbone
+    preprocessor_cls = XceptionImageClassifierPreprocessor

keras_hub_nightly-0.21.0.dev202504052235/keras_hub/src/models/xception/xception_image_classifier_preprocessor.py

@@ -0,0 +1,14 @@
+from keras_hub.src.api_export import keras_hub_export
+from keras_hub.src.models.image_classifier_preprocessor import (
+    ImageClassifierPreprocessor,
+)
+from keras_hub.src.models.xception.xception_backbone import XceptionBackbone
+from keras_hub.src.models.xception.xception_image_converter import (
+    XceptionImageConverter,
+)
+
+
+@keras_hub_export("keras_hub.models.XceptionImageClassifierPreprocessor")
+class XceptionImageClassifierPreprocessor(ImageClassifierPreprocessor):
+    backbone_cls = XceptionBackbone
+    image_converter_cls = XceptionImageConverter

keras_hub_nightly-0.21.0.dev202504052235/keras_hub/src/models/xception/xception_image_converter.py

@@ -0,0 +1,8 @@
+from keras_hub.src.api_export import keras_hub_export
+from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
+from keras_hub.src.models.xception.xception_backbone import XceptionBackbone
+
+
+@keras_hub_export("keras_hub.layers.XceptionImageConverter")
+class XceptionImageConverter(ImageConverter):
+    backbone_cls = XceptionBackbone

keras_hub_nightly-0.21.0.dev202504052235/keras_hub/src/models/xception/xception_presets.py

@@ -0,0 +1,14 @@
+"""Xception preset configurations."""
+
+backbone_presets = {
+    "xception_41_imagenet": {
+        "metadata": {
+            "description": (
+                "41-layer Xception model pre-trained on ImageNet 1k."
+            ),
+            "params": 20861480,
+            "path": "xception",
+        },
+        "kaggle_handle": "kaggle://keras/xception/keras/xception_41_imagenet/2",
+    },
+}

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/keras_hub/src/version_utils.py

@@ -1,7 +1,7 @@
 from keras_hub.src.api_export import keras_hub_export
 
 # Unique source of truth for the version number.
-__version__ = "0.21.0.dev202504040358"
+__version__ = "0.21.0.dev202504052235"
 
 
 @keras_hub_export("keras_hub.version")

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/keras_hub_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-hub-nightly
-Version: 0.21.0.dev202504040358
+Version: 0.21.0.dev202504052235
 Summary: Industry-strength Natural Language Processing extensions for Keras.
 Home-page: https://github.com/keras-team/keras-hub
 Author: Keras team

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/keras_hub_nightly.egg-info/SOURCES.txt

@@ -399,6 +399,12 @@ keras_hub/src/models/whisper/whisper_decoder.py
 keras_hub/src/models/whisper/whisper_encoder.py
 keras_hub/src/models/whisper/whisper_presets.py
 keras_hub/src/models/whisper/whisper_tokenizer.py
+keras_hub/src/models/xception/__init__.py
+keras_hub/src/models/xception/xception_backbone.py
+keras_hub/src/models/xception/xception_image_classifier.py
+keras_hub/src/models/xception/xception_image_classifier_preprocessor.py
+keras_hub/src/models/xception/xception_image_converter.py
+keras_hub/src/models/xception/xception_presets.py
 keras_hub/src/models/xlm_roberta/__init__.py
 keras_hub/src/models/xlm_roberta/xlm_roberta_backbone.py
 keras_hub/src/models/xlm_roberta/xlm_roberta_masked_lm.py

{keras_hub_nightly-0.21.0.dev202504040358 → keras_hub_nightly-0.21.0.dev202504052235}/setup.py

@@ -23,7 +23,7 @@ def get_version(rel_path):
 
 HERE = pathlib.Path(__file__).parent
 README = (HERE / "README.md").read_text()
-VERSION = "0.21.0.dev202504040358"  # get_version("keras_hub/src/version_utils.py")
+VERSION = "0.21.0.dev202504052235"  # get_version("keras_hub/src/version_utils.py")
 
 setup(
     name="keras-hub-nightly",  # "keras-hub",