keras-hub-nightly 0.16.1.dev202410180341__tar.gz → 0.16.1.dev202410200345__tar.gz

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: keras-hub-nightly
-Version: 0.16.1.dev202410180341
+Version: 0.16.1.dev202410200345
 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.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/api/layers/__init__.py

@@ -40,9 +40,7 @@ from keras_hub.src.models.deeplab_v3.deeplab_v3_image_converter import (
 from keras_hub.src.models.densenet.densenet_image_converter import (
     DenseNetImageConverter,
 )
-from keras_hub.src.models.mix_transformer.mix_transformer_image_converter import (
-    MiTImageConverter,
-)
+from keras_hub.src.models.mit.mit_image_converter import MiTImageConverter
 from keras_hub.src.models.pali_gemma.pali_gemma_image_converter import (
     PaliGemmaImageConverter,
 )
@@ -52,7 +50,7 @@ from keras_hub.src.models.resnet.resnet_image_converter import (
 from keras_hub.src.models.sam.sam_image_converter import SAMImageConverter
 from keras_hub.src.models.sam.sam_mask_decoder import SAMMaskDecoder
 from keras_hub.src.models.sam.sam_prompt_encoder import SAMPromptEncoder
-from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageConverter
+from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
 from keras_hub.src.models.whisper.whisper_audio_converter import (
     WhisperAudioConverter,
 )

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/api/models/__init__.py

@@ -202,13 +202,9 @@ from keras_hub.src.models.mistral.mistral_causal_lm_preprocessor import (
     MistralCausalLMPreprocessor,
 )
 from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
-from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier import (
-    MiTImageClassifier,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier_preprocessor import (
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_classifier import MiTImageClassifier
+from keras_hub.src.models.mit.mit_image_classifier_preprocessor import (
     MiTImageClassifierPreprocessor,
 )
 from keras_hub.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
@@ -299,7 +295,7 @@ from keras_hub.src.models.text_classifier_preprocessor import (
 from keras_hub.src.models.text_to_image import TextToImage
 from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
 from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageClassifier
-from keras_hub.src.models.vgg.vgg_image_classifier import (
+from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
     VGGImageClassifierPreprocessor,
 )
 from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone

keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/mit/__init__.py

@@ -0,0 +1,6 @@
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_classifier import MiTImageClassifier
+from keras_hub.src.models.mit.mit_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, MiTBackbone)

keras_hub_nightly-0.16.1.dev202410180341/keras_hub/src/models/mix_transformer/mix_transformer_backbone.py → keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/mit/mit_backbone.py

@@ -4,12 +4,8 @@ from keras import ops
 
 from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.models.feature_pyramid_backbone import FeaturePyramidBackbone
-from keras_hub.src.models.mix_transformer.mix_transformer_layers import (
-    HierarchicalTransformerEncoder,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_layers import (
-    OverlappingPatchingAndEmbedding,
-)
+from keras_hub.src.models.mit.mit_layers import HierarchicalTransformerEncoder
+from keras_hub.src.models.mit.mit_layers import OverlappingPatchingAndEmbedding
 
 
 @keras_hub_export("keras_hub.models.MiTBackbone")
@@ -61,7 +57,7 @@ class MiTBackbone(FeaturePyramidBackbone):
         ```python
         images = np.ones(shape=(1, 96, 96, 3))
         labels = np.zeros(shape=(1, 96, 96, 1))
-        backbone = keras_hub.models.MiTBackbone.from_preset("mit_b0_imagenet")
+        backbone = keras_hub.models.MiTBackbone.from_preset("mit_b0_ade20k_512")
 
         # Evaluate model
         model(images)

keras_hub_nightly-0.16.1.dev202410180341/keras_hub/src/models/mix_transformer/mix_transformer_classifier.py → keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/mit/mit_image_classifier.py

@@ -1,9 +1,7 @@
 from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.models.image_classifier import ImageClassifier
-from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier_preprocessor import (
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_classifier_preprocessor import (
     MiTImageClassifierPreprocessor,
 )
 

keras_hub_nightly-0.16.1.dev202410180341/keras_hub/src/models/mix_transformer/mix_transformer_classifier_preprocessor.py → keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/mit/mit_image_classifier_preprocessor.py

@@ -2,12 +2,8 @@ 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.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_image_converter import (
-    MiTImageConverter,
-)
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_converter import MiTImageConverter
 
 
 @keras_hub_export("keras_hub.models.MiTImageClassifierPreprocessor")

keras_hub_nightly-0.16.1.dev202410180341/keras_hub/src/models/mix_transformer/mix_transformer_image_converter.py → keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/mit/mit_image_converter.py

@@ -1,6 +1,6 @@
 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.mix_transformer import MiTBackbone
+from keras_hub.src.models.mit import MiTBackbone
 
 
 @keras_hub_export("keras_hub.layers.MiTImageConverter")

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/mobilenet/mobilenet_backbone.py

@@ -96,7 +96,7 @@ class MobileNetBackbone(Backbone):
         stackwise_activation,
         output_num_filters,
         inverted_res_block,
-        image_shape=(224, 224, 3),
+        image_shape=(None, None, 3),
         input_activation="hard_swish",
         output_activation="hard_swish",
         depth_multiplier=1.0,

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/resnet/resnet_backbone.py

@@ -68,7 +68,7 @@ class ResNetBackbone(FeaturePyramidBackbone):
     input_data = np.random.uniform(0, 1, size=(2, 224, 224, 3))
 
     # Pretrained ResNet backbone.
-    model = keras_hub.models.ResNetBackbone.from_preset("resnet50")
+    model = keras_hub.models.ResNetBackbone.from_preset("resnet_50_imagenet")
     model(input_data)
 
     # Randomly initialized ResNetV2 backbone with a custom config.
@@ -80,7 +80,6 @@ class ResNetBackbone(FeaturePyramidBackbone):
         stackwise_num_strides=[1, 2, 2],
         block_type="basic_block",
         use_pre_activation=True,
-        pooling="avg",
     )
     model(input_data)
     ```

keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/resnet/resnet_presets.py

@@ -0,0 +1,223 @@
+"""ResNet preset configurations."""
+
+backbone_presets = {
+    "resnet_18_imagenet": {
+        "metadata": {
+            "description": (
+                "18-layer ResNet model pre-trained on the ImageNet 1k dataset "
+                "at a 224x224 resolution."
+            ),
+            "params": 11186112,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/2110.00476",
+        },
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_18_imagenet/2",
+    },
+    "resnet_50_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNet model pre-trained on the ImageNet 1k dataset "
+                "at a 224x224 resolution."
+            ),
+            "params": 23561152,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/2110.00476",
+        },
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_50_imagenet/2",
+    },
+    "resnet_101_imagenet": {
+        "metadata": {
+            "description": (
+                "101-layer ResNet model pre-trained on the ImageNet 1k dataset "
+                "at a 224x224 resolution."
+            ),
+            "params": 42605504,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/2110.00476",
+        },
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_101_imagenet/2",
+    },
+    "resnet_152_imagenet": {
+        "metadata": {
+            "description": (
+                "152-layer ResNet model pre-trained on the ImageNet 1k dataset "
+                "at a 224x224 resolution."
+            ),
+            "params": 58295232,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/2110.00476",
+        },
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_152_imagenet/2",
+    },
+    "resnet_v2_50_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetV2 model pre-trained on the ImageNet 1k "
+                "dataset at a 224x224 resolution."
+            ),
+            "params": 23561152,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/2110.00476",
+        },
+        "kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_50_imagenet/2",
+    },
+    "resnet_v2_101_imagenet": {
+        "metadata": {
+            "description": (
+                "101-layer ResNetV2 model pre-trained on the ImageNet 1k "
+                "dataset at a 224x224 resolution."
+            ),
+            "params": 42605504,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/2110.00476",
+        },
+        "kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_101_imagenet/2",
+    },
+    "resnet_vd_18_imagenet": {
+        "metadata": {
+            "description": (
+                "18-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 11722824,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_18_imagenet",
+    },
+    "resnet_vd_34_imagenet": {
+        "metadata": {
+            "description": (
+                "34-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 21838408,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_34_imagenet",
+    },
+    "resnet_vd_50_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_imagenet",
+    },
+    "resnet_vd_50_ssld_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_imagenet",
+    },
+    "resnet_vd_50_ssld_v2_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation and AutoAugment."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_v2_imagenet",
+    },
+    "resnet_vd_50_ssld_v2_fix_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation, AutoAugment and "
+                "additional fine-tuning of the classification head."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_v2_fix_imagenet",
+    },
+    "resnet_vd_101_imagenet": {
+        "metadata": {
+            "description": (
+                "101-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 44673864,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_101_imagenet",
+    },
+    "resnet_vd_101_ssld_imagenet": {
+        "metadata": {
+            "description": (
+                "101-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation."
+            ),
+            "params": 44673864,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_101_ssld_imagenet",
+    },
+    "resnet_vd_152_imagenet": {
+        "metadata": {
+            "description": (
+                "152-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 60363592,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_152_imagenet",
+    },
+    "resnet_vd_200_imagenet": {
+        "metadata": {
+            "description": (
+                "200-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 74933064,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_200_imagenet",
+    },
+}

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/sam/sam_image_segmenter.py

@@ -31,7 +31,7 @@ class SAMImageSegmenter(ImageSegmenter):
 
 
     Args:
-      backbone: A `keras_hub.models.VGGBackbone` instance.
+      backbone: A `keras_hub.models.SAMBackbone` instance.
 
     Example:
     Load pretrained model using `from_preset`.

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/sam/sam_presets.py

@@ -9,7 +9,7 @@ backbone_presets = {
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/2",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/4",
     },
     "sam_large_sa1b": {
         "metadata": {
@@ -19,7 +19,7 @@ backbone_presets = {
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/3",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/4",
     },
     "sam_huge_sa1b": {
         "metadata": {
@@ -29,6 +29,6 @@ backbone_presets = {
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/3",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/4",
     },
 }

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py

@@ -10,7 +10,7 @@ backbone_presets = {
             ),
             "params": 2987080931,
             "official_name": "StableDiffusion3",
-            "path": "stablediffusion3",
+            "path": "stable_diffusion_3",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
         "kaggle_handle": "kaggle://keras/stablediffusion3/keras/stable_diffusion_3_medium/1",

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/vae/vae_backbone.py

@@ -10,7 +10,7 @@ from keras_hub.src.utils.keras_utils import standardize_data_format
 
 
 class VAEBackbone(Backbone):
-    """VAE backbone used in latent diffusion models.
+    """Variational Autoencoder(VAE) backbone used in latent diffusion models.
 
     When encoding, this model generates mean and log variance of the input
     images. When decoding, it reconstructs images from the latent space.
@@ -51,6 +51,18 @@ class VAEBackbone(Backbone):
             `"channels_last"`.
         dtype: `None` or str or `keras.mixed_precision.DTypePolicy`. The dtype
             to use for the model's computations and weights.
+
+    Example:
+    ```Python
+    backbone = VAEBackbone(
+        encoder_num_filters=[32, 32, 32, 32],
+        encoder_num_blocks=[1, 1, 1, 1],
+        decoder_num_filters=[32, 32, 32, 32],
+        decoder_num_blocks=[1, 1, 1, 1],
+    )
+    input_data = ops.ones((2, self.height, self.width, 3))
+    output = backbone(input_data)
+    ```
     """
 
     def __init__(

keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/vgg/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
+from keras_hub.src.models.vgg.vgg_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, VGGBackbone)

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/vgg/vgg_backbone.py

@@ -20,7 +20,7 @@ class VGGBackbone(Backbone):
       stackwise_num_filters: list of ints, filter size for convolutional
             blocks per VGG block. For both VGG16 and VGG19 this is [
             64, 128, 256, 512, 512].
-      image_shape: tuple, optional shape tuple, defaults to (224, 224, 3).
+      image_shape: tuple, optional shape tuple, defaults to (None, None, 3).
 
     Examples:
     ```python

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/vgg/vgg_image_classifier.py

@@ -1,24 +1,12 @@
 import keras
 
 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.image_classifier import ImageClassifier
-from keras_hub.src.models.image_classifier_preprocessor import (
-    ImageClassifierPreprocessor,
-)
 from keras_hub.src.models.task import Task
 from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
-
-
-@keras_hub_export("keras_hub.layers.VGGImageConverter")
-class VGGImageConverter(ImageConverter):
-    backbone_cls = VGGBackbone
-
-
-@keras_hub_export("keras_hub.models.VGGImageClassifierPreprocessor")
-class VGGImageClassifierPreprocessor(ImageClassifierPreprocessor):
-    backbone_cls = VGGBackbone
-    image_converter_cls = VGGImageConverter
+from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
+    VGGImageClassifierPreprocessor,
+)
 
 
 @keras_hub_export("keras_hub.models.VGGImageClassifier")
@@ -211,6 +199,7 @@ class VGGImageClassifier(ImageClassifier):
         self.pooling = pooling
         self.pooling_hidden_dim = pooling_hidden_dim
         self.dropout = dropout
+        self.preprocessor = preprocessor
 
     def get_config(self):
         # Backbone serialized in `super`

keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/vgg/vgg_image_classifier_preprocessor.py

@@ -0,0 +1,12 @@
+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.vgg.vgg_backbone import VGGBackbone
+from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
+
+
+@keras_hub_export("keras_hub.models.VGGImageClassifierPreprocessor")
+class VGGImageClassifierPreprocessor(ImageClassifierPreprocessor):
+    backbone_cls = VGGBackbone
+    image_converter_cls = VGGImageConverter

keras_hub_nightly-0.16.1.dev202410200345/keras_hub/src/models/vgg/vgg_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.vgg.vgg_backbone import VGGBackbone
+
+
+@keras_hub_export("keras_hub.layers.VGGImageConverter")
+class VGGImageConverter(ImageConverter):
+    backbone_cls = VGGBackbone

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/models/vit_det/vit_det_backbone.py

@@ -31,7 +31,7 @@ class ViTDetBackbone(Backbone):
         global_attention_layer_indices (list): Indexes for blocks using
             global attention.
         image_shape (tuple[int], optional): The size of the input image in
-            `(H, W, C)` format. Defaults to `(1024, 1024, 3)`.
+            `(H, W, C)` format. Defaults to `(None, None, 3)`.
         patch_size (int, optional): the patch size to be supplied to the
             Patching layer to turn input images into a flattened sequence of
             patches. Defaults to `16`.
@@ -79,7 +79,7 @@ class ViTDetBackbone(Backbone):
         intermediate_dim,
         num_heads,
         global_attention_layer_indices,
-        image_shape=(1024, 1024, 3),
+        image_shape=(None, None, 3),
         patch_size=16,
         num_output_channels=256,
         use_bias=True,

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub/src/utils/pipeline_model.py

@@ -232,7 +232,7 @@ class PipelineModel(keras.Model):
     ):
         data = self.preprocess_samples(x, y, sample_weight)
         x, y, sample_weight = keras.utils.unpack_x_y_sample_weight(data)
-        x = ops.convert_to_tensor(x)
+        x = tree.map_structure(ops.convert_to_tensor, x)
         if y is not None:
             y = ops.convert_to_tensor(y)
         if sample_weight is not None:
@@ -253,7 +253,7 @@ class PipelineModel(keras.Model):
     ):
         data = self.preprocess_samples(x, y, sample_weight)
         x, y, sample_weight = keras.utils.unpack_x_y_sample_weight(data)
-        x = ops.convert_to_tensor(x)
+        x = tree.map_structure(ops.convert_to_tensor, x)
         if y is not None:
             y = ops.convert_to_tensor(y)
         if sample_weight is not None:
@@ -272,7 +272,7 @@ class PipelineModel(keras.Model):
     ):
         data = self.preprocess_samples(x)
         x, _, _ = keras.utils.unpack_x_y_sample_weight(data)
-        x = ops.convert_to_tensor(x)
+        x = tree.map_structure(ops.convert_to_tensor, x)
         return super().predict_on_batch(
             x=x,
             **kwargs,

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/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.16.1.dev202410180341"
+__version__ = "0.16.1.dev202410200345"
 
 
 @keras_hub_export("keras_hub.version")

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: keras-hub-nightly
-Version: 0.16.1.dev202410180341
+Version: 0.16.1.dev202410200345
 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.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/keras_hub_nightly.egg-info/SOURCES.txt

@@ -212,13 +212,13 @@ keras_hub/src/models/mistral/mistral_layer_norm.py
 keras_hub/src/models/mistral/mistral_presets.py
 keras_hub/src/models/mistral/mistral_tokenizer.py
 keras_hub/src/models/mistral/mistral_transformer_decoder.py
-keras_hub/src/models/mix_transformer/__init__.py
-keras_hub/src/models/mix_transformer/mix_transformer_backbone.py
-keras_hub/src/models/mix_transformer/mix_transformer_classifier.py
-keras_hub/src/models/mix_transformer/mix_transformer_classifier_preprocessor.py
-keras_hub/src/models/mix_transformer/mix_transformer_image_converter.py
-keras_hub/src/models/mix_transformer/mix_transformer_layers.py
-keras_hub/src/models/mix_transformer/mix_transformer_presets.py
+keras_hub/src/models/mit/__init__.py
+keras_hub/src/models/mit/mit_backbone.py
+keras_hub/src/models/mit/mit_image_classifier.py
+keras_hub/src/models/mit/mit_image_classifier_preprocessor.py
+keras_hub/src/models/mit/mit_image_converter.py
+keras_hub/src/models/mit/mit_layers.py
+keras_hub/src/models/mit/mit_presets.py
 keras_hub/src/models/mobilenet/__init__.py
 keras_hub/src/models/mobilenet/mobilenet_backbone.py
 keras_hub/src/models/mobilenet/mobilenet_image_classifier.py
@@ -301,6 +301,8 @@ keras_hub/src/models/vae/vae_layers.py
 keras_hub/src/models/vgg/__init__.py
 keras_hub/src/models/vgg/vgg_backbone.py
 keras_hub/src/models/vgg/vgg_image_classifier.py
+keras_hub/src/models/vgg/vgg_image_classifier_preprocessor.py
+keras_hub/src/models/vgg/vgg_image_converter.py
 keras_hub/src/models/vgg/vgg_presets.py
 keras_hub/src/models/vit_det/__init__.py
 keras_hub/src/models/vit_det/vit_det_backbone.py

{keras_hub_nightly-0.16.1.dev202410180341 → keras_hub_nightly-0.16.1.dev202410200345}/setup.py

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