File tree Expand file tree Collapse file tree 2 files changed +13
-0
lines changed
src/models/backbones/video_swin Expand file tree Collapse file tree 2 files changed +13
-0
lines changed Original file line number Diff line number Diff line change 178178 ResNetV2Backbone ,
179179)
180180from 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+ )
181190from keras_cv .src .models .backbones .video_swin .video_swin_backbone import (
182191 VideoSwinBackbone ,
183192)
Original file line number Diff line number Diff line change 1414
1515import copy
1616
17+ from keras_cv .src .api_export import keras_cv_export
1718from keras_cv .src .models .backbones .video_swin .video_swin_backbone import (
1819 VideoSwinBackbone ,
1920)
4243""" # noqa: E501
4344
4445
46+ @keras_cv_export ("keras_cv.models.VideoSwinTBackbone" )
4547class 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" )
8285class 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" )
119123class VideoSwinBBackbone (VideoSwinBackbone ):
120124 def __new__ (
121125 cls ,
You can’t perform that action at this time.
0 commit comments