Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Video swin #2319

Closed
wants to merge 32 commits into from
Closed

Video swin #2319

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
32 commits
Select commit Hold shift + click to select a range
c1eaeb5
add: video siwn mlp layer
innat Jan 20, 2024
14b1825
add patch embed 3d
innat Jan 20, 2024
5d33506
add patch merging layer
innat Jan 20, 2024
8dcf7e2
add window attention layer in 3d
innat Jan 20, 2024
9d0bca4
add utility methods for video swin
innat Jan 20, 2024
bfd54d7
add basic layer of video swin
innat Jan 20, 2024
03f3b11
add 3d swin transformer block
innat Jan 20, 2024
c8f6a90
add keras_cv_export on window attention and swin block
innat Jan 20, 2024
23a1d71
simplify the mlp layer's argument name
innat Jan 20, 2024
27de752
update mlp layer: instance to identifier
innat Jan 20, 2024
e493fa5
update mlp calls
innat Jan 20, 2024
cf13394
add docstring to mlp layer
innat Jan 20, 2024
5291e74
update patch embedding layer
innat Jan 20, 2024
ec6a005
add docstring to patch embedding layer
innat Jan 20, 2024
1e3ae30
update patch merging layer
innat Jan 20, 2024
512c71d
update window attention
innat Jan 20, 2024
610c6ba
update swin transformer block
innat Jan 20, 2024
a158711
update swin basic layer
innat Jan 20, 2024
dcf01bb
run formatters
innat Jan 20, 2024
5eedb9f
update docstring and restructure call method of SwinTransformerBlock3…
Jan 27, 2024
f13e697
remove return_attention_maps argument from call method
Jan 27, 2024
276f1b3
create video_swin_transformer_layers_test file
Jan 27, 2024
b31f78f
add TestPatchEmbedding3D
Jan 27, 2024
55a70ae
update window attention layer and add init test case
Jan 27, 2024
4298869
update window attention test case
Jan 27, 2024
d8e98b1
update siwn transformer block and add test file
Jan 27, 2024
e1f3d30
update siwn transformer test file
Jan 27, 2024
2636d9b
run formatter
Jan 27, 2024
3f26791
add layers api to init
Jan 27, 2024
e0645c4
run formatter
Jan 27, 2024
c3e1099
update mlp layer with build method
innat Mar 1, 2024
b5fa44b
Revert "run formatter"
innat Mar 1, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions keras_cv/layers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -141,3 +141,5 @@
from keras_cv.layers.vit_det_layers import WindowedTransformerEncoder
from keras_cv.layers.vit_det_layers import WindowPartitioning
from keras_cv.layers.vit_layers import PatchingAndEmbedding
from keras_cv.layers.video_swin_transformer_layers import WindowAttention3D
from keras_cv.layers.video_swin_transformer_layers import SwinTransformerBlock3D
Loading
Loading