Skip to content

Commit c54f4a8

Browse files
Export video swin APIs (#2428)
* Export APIs * generate API
1 parent fcf9576 commit c54f4a8

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

keras_cv/api/models/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,15 @@
178178
ResNetV2Backbone,
179179
)
180180
from keras_cv.src.models.backbones.vgg16.vgg16_backbone import VGG16Backbone
181+
from keras_cv.src.models.backbones.video_swin.video_swin_aliases import (
182+
VideoSwinBBackbone,
183+
)
184+
from keras_cv.src.models.backbones.video_swin.video_swin_aliases import (
185+
VideoSwinSBackbone,
186+
)
187+
from keras_cv.src.models.backbones.video_swin.video_swin_aliases import (
188+
VideoSwinTBackbone,
189+
)
181190
from keras_cv.src.models.backbones.video_swin.video_swin_backbone import (
182191
VideoSwinBackbone,
183192
)

keras_cv/src/models/backbones/video_swin/video_swin_aliases.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
import copy
1616

17+
from keras_cv.src.api_export import keras_cv_export
1718
from keras_cv.src.models.backbones.video_swin.video_swin_backbone import (
1819
VideoSwinBackbone,
1920
)
@@ -42,6 +43,7 @@
4243
""" # noqa: E501
4344

4445

46+
@keras_cv_export("keras_cv.models.VideoSwinTBackbone")
4547
class VideoSwinTBackbone(VideoSwinBackbone):
4648
def __new__(
4749
cls,
@@ -79,6 +81,7 @@ def presets_with_weights(cls):
7981
return cls.presets
8082

8183

84+
@keras_cv_export("keras_cv.models.VideoSwinSBackbone")
8285
class VideoSwinSBackbone(VideoSwinBackbone):
8386
def __new__(
8487
cls,
@@ -116,6 +119,7 @@ def presets_with_weights(cls):
116119
return cls.presets
117120

118121

122+
@keras_cv_export("keras_cv.models.VideoSwinBBackbone")
119123
class VideoSwinBBackbone(VideoSwinBackbone):
120124
def __new__(
121125
cls,

0 commit comments

Comments
 (0)