[PaddlePaddle Hackathon] add DenseNet

python/paddle/tests/test_pretrained_model.py

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

python/paddle/tests/test_vision_models.py

@@ -70,6 +70,21 @@ def test_resnet101(self):
  def test_resnet152(self):
  self.models_infer('resnet152')
 
+ def test_densenet121(self):
+ self.models_infer('densenet121')
+
+ def test_densenet161(self):
+ self.models_infer('densenet161')
+
+ def test_densenet169(self):
+ self.models_infer('densenet169')
+
+ def test_densenet201(self):
+ self.models_infer('densenet201')
+
+ def test_densenet264(self):
+ self.models_infer('densenet264')
+
  def test_alexnet(self):
  self.models_infer('alexnet')
 

python/paddle/vision/__init__.py

@@ -44,6 +44,12 @@
 from .models import vgg16 # noqa: F401
 from .models import vgg19 # noqa: F401
 from .models import LeNet # noqa: F401
+from .models import DenseNet # noqa: F401
+from .models import densenet121 # noqa: F401
+from .models import densenet161 # noqa: F401
+from .models import densenet169 # noqa: F401
+from .models import densenet201 # noqa: F401
+from .models import densenet264 # noqa: F401
 from .models import AlexNet # noqa: F401
 from .models import alexnet # noqa: F401
 from .models import ResNeXt # noqa: F401

python/paddle/vision/models/__init__.py

@@ -28,6 +28,12 @@
 from .vgg import vgg16 # noqa: F401
 from .vgg import vgg19 # noqa: F401
 from .lenet import LeNet # noqa: F401
+from .densenet import DenseNet # noqa: F401
+from .densenet import densenet121 # noqa: F401
+from .densenet import densenet161 # noqa: F401
+from .densenet import densenet169 # noqa: F401
+from .densenet import densenet201 # noqa: F401
+from .densenet import densenet264 # noqa: F401
 from .alexnet import AlexNet # noqa: F401
 from .alexnet import alexnet # noqa: F401
 from .resnext import ResNeXt # noqa: F401
@@ -57,6 +63,12 @@
  'MobileNetV2',
  'mobilenet_v2',
  'LeNet',
+ 'DenseNet',
+ 'densenet121',
+ 'densenet161',
+ 'densenet169',
+ 'densenet201',
+ 'densenet264',
  'AlexNet',
  'alexnet',
  'ResNeXt',