[PaddlePaddle Hackathon] add ShuffleNetV2

python/paddle/tests/test_pretrained_model.py

@@ -54,7 +54,8 @@ def infer(self, arch):
     def test_models(self):
         arches = [
             'mobilenet_v1', 'mobilenet_v2', 'resnet18', 'vgg16', 'alexnet',
-            'resnext50_32x4d', 'inception_v3', 'densenet121', 'googlenet'
+            'resnext50_32x4d', 'inception_v3', 'densenet121', 'googlenet',
+            'shufflenet_v2_x0_25', 'shufflenet_v2_swish'
         ]
         for arch in arches:
             self.infer(arch)

python/paddle/tests/test_vision_models.py

@@ -88,6 +88,9 @@ def test_densenet264(self):
     def test_alexnet(self):
         self.models_infer('alexnet')
 
+    def test_shufflenetv2_swish(self):
+        self.models_infer('shufflenet_v2_swish')
+
     def test_resnext50_32x4d(self):
         self.models_infer('resnext50_32x4d')
 
@@ -112,6 +115,24 @@ def test_inception_v3(self):
     def test_googlenet(self):
         self.models_infer('googlenet')
 
+    def test_shufflenetv2_x0_25(self):
+        self.models_infer('shufflenet_v2_x0_25')
+
+    def test_shufflenetv2_x0_33(self):
+        self.models_infer('shufflenet_v2_x0_33')
+
+    def test_shufflenetv2_x0_5(self):
+        self.models_infer('shufflenet_v2_x0_5')
+
+    def test_shufflenetv2_x1_0(self):
+        self.models_infer('shufflenet_v2_x1_0')
+
+    def test_shufflenetv2_x1_5(self):
+        self.models_infer('shufflenet_v2_x1_5')
+
+    def test_shufflenetv2_x2_0(self):
+        self.models_infer('shufflenet_v2_x2_0')
+
     def test_vgg16_num_classes(self):
         vgg16 = models.__dict__['vgg16'](pretrained=False, num_classes=10)
 

python/paddle/vision/__init__.py

@@ -63,6 +63,14 @@
 from .models import inception_v3  # noqa: F401
 from .models import GoogLeNet  # noqa: F401
 from .models import googlenet  # noqa: F401
+from .models import ShuffleNetV2  # noqa: F401
+from .models import shufflenet_v2_x0_25  # noqa: F401
+from .models import shufflenet_v2_x0_33  # noqa: F401
+from .models import shufflenet_v2_x0_5  # noqa: F401
+from .models import shufflenet_v2_x1_0  # noqa: F401
+from .models import shufflenet_v2_x1_5  # noqa: F401
+from .models import shufflenet_v2_x2_0  # noqa: F401
+from .models import shufflenet_v2_swish  # noqa: F401
 from .transforms import BaseTransform  # noqa: F401
 from .transforms import Compose  # noqa: F401
 from .transforms import Resize  # noqa: F401

python/paddle/vision/models/__init__.py

@@ -47,6 +47,14 @@
 from .inceptionv3 import inception_v3  # noqa: F401
 from .googlenet import GoogLeNet  # noqa: F401
 from .googlenet import googlenet  # noqa: F401
+from .shufflenetv2 import ShuffleNetV2  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_x0_25  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_x0_33  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_x0_5  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_x1_0  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_x1_5  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_x2_0  # noqa: F401
+from .shufflenetv2 import shufflenet_v2_swish  # noqa: F401
 
 __all__ = [ #noqa
     'ResNet',
@@ -84,4 +92,12 @@
     'inception_v3',
     'GoogLeNet',
     'googlenet',
+    'ShuffleNetV2',
+    'shufflenet_v2_x0_25',
+    'shufflenet_v2_x0_33',
+    'shufflenet_v2_x0_5',
+    'shufflenet_v2_x1_0',
+    'shufflenet_v2_x1_5',
+    'shufflenet_v2_x2_0',
+    'shufflenet_v2_swish'
 ]