add mvitv2 backbone

Reviewed By: feichtenhofer, haooooooqi Differential Revision: D36089769 fbshipit-source-id: 2450123157fcba36b62d138938b9d7921d2566db
OBaruch · Jun 9, 2022 · b01e0e9 · b01e0e9
1 parent 667693d
commit b01e0e9
Show file tree

Hide file tree

Showing 3 changed files with 450 additions and 0 deletions.
diff --git a/detectron2/modeling/__init__.py b/detectron2/modeling/__init__.py
@@ -13,6 +13,8 @@
     make_stage,
     ViT,
     SimpleFeaturePyramid,
+    MViT,
+    SwinTransformer,
 )
 from .meta_arch import (
     META_ARCH_REGISTRY,

diff --git a/detectron2/modeling/backbone/__init__.py b/detectron2/modeling/backbone/__init__.py
@@ -13,6 +13,8 @@
     BottleneckBlock,
 )
 from .vit import ViT, SimpleFeaturePyramid
+from .mvit import MViT
+from .swin import SwinTransformer
 
 __all__ = [k for k in globals().keys() if not k.startswith("_")]
 # TODO can expose more resnet blocks after careful consideration