Skip to content

Commit 9569150

Browse files
Add VideoPrism models. (keras-team#2592)
* Add VideoPrism models. * Address gemini-cli comments. * Add `data_format` to `VideoPrism`. * Add `compute_output_shape` and `build` to the layers.
1 parent 2264d42 commit 9569150

File tree

11 files changed

+2088
-0
lines changed

11 files changed

+2088
-0
lines changed

keras_hub/api/layers/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,9 @@
168168
from keras_hub.src.models.vgg.vgg_image_converter import (
169169
VGGImageConverter as VGGImageConverter,
170170
)
171+
from keras_hub.src.models.video_prism.video_prism_image_converter import (
172+
VideoPrismImageConverter as VideoPrismImageConverter,
173+
)
171174
from keras_hub.src.models.vit.vit_image_converter import (
172175
ViTImageConverter as ViTImageConverter,
173176
)

keras_hub/api/models/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -766,6 +766,12 @@
766766
from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
767767
VGGImageClassifierPreprocessor as VGGImageClassifierPreprocessor,
768768
)
769+
from keras_hub.src.models.video_prism.video_prism_backbone import (
770+
VideoPrismBackbone as VideoPrismBackbone,
771+
)
772+
from keras_hub.src.models.video_prism.video_prism_tokenizer import (
773+
VideoPrismTokenizer as VideoPrismTokenizer,
774+
)
769775
from keras_hub.src.models.vit.vit_backbone import ViTBackbone as ViTBackbone
770776
from keras_hub.src.models.vit.vit_image_classifier import (
771777
ViTImageClassifier as ViTImageClassifier,

keras_hub/api/tokenizers/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,9 @@
112112
from keras_hub.src.models.t5gemma.t5gemma_tokenizer import (
113113
T5GemmaTokenizer as T5GemmaTokenizer,
114114
)
115+
from keras_hub.src.models.video_prism.video_prism_tokenizer import (
116+
VideoPrismTokenizer as VideoPrismTokenizer,
117+
)
115118
from keras_hub.src.models.whisper.whisper_tokenizer import (
116119
WhisperTokenizer as WhisperTokenizer,
117120
)
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from keras_hub.src.models.video_prism.video_prism_backbone import (
2+
VideoPrismBackbone,
3+
)
4+
from keras_hub.src.models.video_prism.video_prism_presets import (
5+
backbone_presets,
6+
)
7+
from keras_hub.src.utils.preset_utils import register_presets
8+
9+
register_presets(backbone_presets, VideoPrismBackbone)

0 commit comments

Comments
 (0)