Add mspn backbone (#278)

* Add mspn backbone * change module name * standardize name style
open-mmlab · Nov 16, 2020 · fd8442f · fd8442f
1 parent 0ecede9
commit fd8442f
Show file tree

Hide file tree

Showing 4 changed files with 562 additions and 1 deletion.
diff --git a/mmpose/models/backbones/__init__.py b/mmpose/models/backbones/__init__.py
@@ -4,6 +4,7 @@
 from .hrnet import HRNet
 from .mobilenet_v2 import MobileNetV2
 from .mobilenet_v3 import MobileNetV3
+from .mspn import MSPN
 from .regnet import RegNet
 from .resnet import ResNet, ResNetV1d
 from .resnext import ResNeXt
@@ -17,5 +18,5 @@
 __all__ = [
     'AlexNet', 'HourglassNet', 'HRNet', 'MobileNetV2', 'MobileNetV3', 'RegNet',
     'ResNet', 'ResNetV1d', 'ResNeXt', 'SCNet', 'SEResNet', 'SEResNeXt',
-    'ShuffleNetV1', 'ShuffleNetV2', 'CPM', 'RSN'
+    'ShuffleNetV1', 'ShuffleNetV2', 'CPM', 'RSN', 'MSPN'
 ]