diff --git a/keras_cv/api/models/__init__.py b/keras_cv/api/models/__init__.py index e425b26899..97f9bc577b 100644 --- a/keras_cv/api/models/__init__.py +++ b/keras_cv/api/models/__init__.py @@ -178,6 +178,15 @@ ResNetV2Backbone, ) from keras_cv.src.models.backbones.vgg16.vgg16_backbone import VGG16Backbone +from keras_cv.src.models.backbones.video_swin.video_swin_aliases import ( + VideoSwinBBackbone, +) +from keras_cv.src.models.backbones.video_swin.video_swin_aliases import ( + VideoSwinSBackbone, +) +from keras_cv.src.models.backbones.video_swin.video_swin_aliases import ( + VideoSwinTBackbone, +) from keras_cv.src.models.backbones.video_swin.video_swin_backbone import ( VideoSwinBackbone, ) diff --git a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py index 781b718910..3bbc4a4f85 100644 --- a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py +++ b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py @@ -14,6 +14,7 @@ import copy +from keras_cv.src.api_export import keras_cv_export from keras_cv.src.models.backbones.video_swin.video_swin_backbone import ( VideoSwinBackbone, ) @@ -42,6 +43,7 @@ """ # noqa: E501 +@keras_cv_export("keras_cv.models.VideoSwinTBackbone") class VideoSwinTBackbone(VideoSwinBackbone): def __new__( cls, @@ -79,6 +81,7 @@ def presets_with_weights(cls): return cls.presets +@keras_cv_export("keras_cv.models.VideoSwinSBackbone") class VideoSwinSBackbone(VideoSwinBackbone): def __new__( cls, @@ -116,6 +119,7 @@ def presets_with_weights(cls): return cls.presets +@keras_cv_export("keras_cv.models.VideoSwinBBackbone") class VideoSwinBBackbone(VideoSwinBackbone): def __new__( cls,